first_commit .gitignore

plugins/sign_in/__init__.py

@@ -0,0 +1,53 @@
+from .group_user_checkin import group_user_check_in, group_user_check, group_impression_rank
+from nonebot.typing import T_State
+from nonebot.adapters.cqhttp import Bot, Event, GroupMessageEvent
+from nonebot.adapters.cqhttp.permission import GROUP
+from util.utils import get_message_text
+from nonebot.plugin import MatcherGroup
+
+
+__plugin_name__ = '签到'
+__plugin_usage__ = (
+    '用法：\n'
+    '对我说 “签到” 来签到\n'
+    '“我的签到” 来获取历史签到信息\n'
+    '“好感度排行” 来查看当前好感度前十的伙伴\n'
+    '/ 签到时有 3% 概率 * 2 /'
+
+)
+
+
+sign_match_group = MatcherGroup(priority=5, permission=GROUP, block=True)
+
+sign = sign_match_group.on_command("签到")
+
+
+@sign.handle()
+async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
+    if get_message_text(event.json()) in ['帮助']:
+        await sign.finish(__plugin_usage__)
+    await sign.send(
+        await group_user_check_in(event.user_id, event.group_id),
+        at_sender=True,
+    )
+
+
+my_sign = sign_match_group.on_command(cmd="我的签到", aliases={'好感度'})
+
+
+@my_sign.handle()
+async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
+    await my_sign.send(
+        await group_user_check(event.user_id, event.group_id),
+        at_sender=True,
+    )
+
+sign_ranking = sign_match_group.on_command(cmd="积分排行", aliases={'好感度排行', '签到排行', '积分排行', '好感排行',
+                                                               '好感度排名，签到排名，积分排名'})
+
+
+@sign_ranking.handle()
+async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
+    await sign_ranking.send(
+        await group_impression_rank(event.group_id)
+    )

plugins/sign_in/group_user_checkin.py

@@ -0,0 +1,119 @@
+import random
+from datetime import datetime, timedelta
+
+from services.log import logger
+from services.db_context import db
+from models.sigin_group_user import SignGroupUser
+from models.group_member_info import GroupInfoUser
+from models.bag_user import UserBag
+from configs.config import MAX_SIGN_GOLD
+
+
+async def group_user_check_in(user_qq: int, group: int) -> str:
+    'Returns string describing the result of checking in'
+    present = datetime.now()
+    async with db.transaction():
+        # 取得相应用户
+        user = await SignGroupUser.ensure(user_qq, group, for_update=True)
+        # 如果同一天签到过，特殊处理
+        if user.checkin_time_last.date() == present.date():
+            return _handle_already_checked_in(user)
+        return await _handle_check_in(user_qq, group, present)    # ok
+
+
+def _handle_already_checked_in(user: SignGroupUser) -> str:
+    return f'已经签到过啦~ 好感度：{user.impression:.2f}'
+
+
+async def _handle_check_in(user_qq: int, group: int, present: datetime) -> str:
+    user = await SignGroupUser.ensure(user_qq, group, for_update=True)
+    impression_added = random.random()
+    present = present + timedelta(hours=8)
+    critx2 = random.random()
+    add_probability = user.add_probability
+    specify_probability = user.specify_probability
+    if critx2 + add_probability > 0.97:
+        impression_added *= 2
+    elif critx2 < specify_probability:
+        impression_added *= 2
+    new_impression = user.impression + impression_added
+    message = random.choice((
+        '谢谢，你是个好人！',
+        '对了，来喝杯茶吗？',
+    ))
+    await user.update(
+        checkin_count=user.checkin_count + 1,
+        checkin_time_last=present,
+        impression=new_impression,
+        add_probability=0,
+        specify_probability=0,
+    ).apply()
+    # glod = await random_glod(user_qq, group, specify_probability)
+    if user.impression < 1:
+        impression = 1
+    else:
+        impression = user.impression
+    gold = random.randint(1, 100)
+    imgold = random.randint(1, int(impression))
+    if imgold > MAX_SIGN_GOLD:
+        imgold = MAX_SIGN_GOLD
+    await UserBag.add_glod(user_qq, group, gold + imgold)
+    if critx2 + add_probability > 0.97 or critx2 < specify_probability:
+        logger.info(f'(USER {user.user_qq}, GROUP {user.belonging_group})'
+                    f' CHECKED IN successfully. score: {new_impression:.2f} (+{impression_added * 2:.2f}).获取金币：{gold+imgold}')
+        return f'{message} 好感度：{new_impression:.2f} (+{impression_added/2:.2f}×2)！！！\n获取金币：{gold} \n好感度额外获得金币：{imgold}'
+    else:
+        logger.info(f'(USER {user.user_qq}, GROUP {user.belonging_group})'
+                    f' CHECKED IN successfully. score: {new_impression:.2f} (+{impression_added:.2f}).获取金币：{gold+imgold}')
+        return f'{message} 好感度：{new_impression:.2f} (+{impression_added:.2f})\n获取金币：{gold} \n好感度额外获得金币：{imgold}'
+
+
+async def group_user_check(user_qq: int, group: int) -> str:
+    # heuristic: if users find they have never checked in they are probable to check in
+    user = await SignGroupUser.ensure(user_qq, group)
+    glod = await UserBag.get_gold(user_qq, group)
+    return '好感度：{:.2f}\n金币：{}\n历史签到数：{}\n上次签到日期：{}'.format(
+        user.impression,
+        glod,
+        user.checkin_count,
+        user.checkin_time_last.date() if user.checkin_time_last != datetime.min else '从未',
+    )
+
+
+async def group_impression_rank(group: int) -> str:
+    result = "\t好感度排行榜\t\n"
+    user_qq_list, impression_list = await SignGroupUser.query_impression_all(group)
+    _count = 11
+    if user_qq_list and impression_list:
+        for i in range(1, 100):
+            if len(user_qq_list) == 0 or len(impression_list) == 0 or i == _count:
+                break
+            impression = max(impression_list)
+            index = impression_list.index(impression)
+            user_qq = user_qq_list[index]
+            print(user_qq, group)
+            try:
+                user_name = (await GroupInfoUser.select_member_info(user_qq, group)).user_name
+            except Exception as e:
+                logger.info(f"USER {user_qq}, GROUP {group} 不在群内")
+                _count += 1
+                impression_list.remove(impression)
+                user_qq_list.remove(user_qq)
+                continue
+            result += f"{i - _count + 11}. {user_name}: {impression:.2f}\n"
+            impression_list.remove(impression)
+            user_qq_list.remove(user_qq)
+    return result[:-1]
+
+
+async def random_glod(user_id, group_id, impression):
+    if impression < 1:
+        impression = 1
+    glod = random.randint(1, 100) + random.randint(1, int(impression))
+    if await UserBag.add_glod(user_id, group_id, glod):
+        return glod
+    else:
+        return 0
+
+
+