1 files changed, 219 insertions, 0 deletions
diff --git a/ATRI/plugins/character.py b/ATRI/plugins/character.py
new file mode 100644
index 0000000..93dbc02
--- /dev/null
+++ b/ATRI/plugins/character.py
@@ -0,0 +1,219 @@
+from random import choice
+from typing import Optional
+
+from nonebot.plugin import on_command, on_regex
+from nonebot.adapters.cqhttp import Bot, Event
+
+from ATRI.rule import is_in_ban_list, is_in_dormant, to_bot
+from ATRI.utils import count_list, del_list_aim, now_time
+from ATRI.config import BOT_CONFIG, RUNTIME_CONFIG
+from ATRI.service.plugin import Plugin
+
+# ===========================[Begin Command Processing]===========================
+
+__plugin_name__ = 'call_robot'
+__doc__ = """使用正则触发"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        command=BOT_CONFIG['callRobot']['command'])
+
+call_robot = on_regex('|'.join(BOT_CONFIG['callRobot']['command']),
+                        rule=is_in_ban_list() & is_in_dormant(),
+                        priority=4)
+
+@call_robot.handle()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().call_robot(int(event.get_user_id())))
+
+
+__plugin_name__ = 'call_me'
+__doc__ = """正确地呼叫咱"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        RUNTIME_CONFIG['nickname'])
+
+call_me = on_command(list(RUNTIME_CONFIG['nickname'])[0],
+                    aliases=RUNTIME_CONFIG['nickname'],
+                    rule=is_in_ban_list() & is_in_dormant())
+
+@call_me.handle()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().call_me())
+
+
+__plugin_name__ = 'tee_tee'
+__doc__ = """一般人员不许贴！
+需at"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        BOT_CONFIG['teetee']['command'])
+
+teeTee = on_command(BOT_CONFIG['teetee']['command'][0],
+                    aliases=set(BOT_CONFIG['teetee']['command']),
+                    rule=is_in_ban_list() & is_in_dormant() & to_bot())
+
+[email protected]()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().tee_tee(int(event.get_user_id())))
+
+
+
+__plugin_name__ = 'kani'
+__doc__ = """カニ！カニ！！！
+使用正则匹配"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        BOT_CONFIG['kani']['command'])
+
+kani = on_regex('|'.join(BOT_CONFIG['kani']['command']),
+                rule=is_in_ban_list() & is_in_dormant())
+
+[email protected]()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().kani())
+
+
+__plugin_name__ = 'waste'
+__doc__ = """不准骂咱废物
+使用正则匹配，需at"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        BOT_CONFIG['waste']['command'])
+
+waste = on_regex('|'.join(BOT_CONFIG['waste']['command']),
+                    rule=is_in_ban_list() & is_in_dormant() & to_bot(),
+                    priority=5)
+
+[email protected]()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().waste())
+
+
+__plugin_name__ = 'good_morning'
+__doc__ = """略带涩气的早安
+需at"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        BOT_CONFIG['morning']['command'])
+
+morning = on_command(BOT_CONFIG['morning']['command'][0],
+                        aliases=set(BOT_CONFIG['morning']['command']),
+                        rule=is_in_ban_list() & is_in_dormant() & to_bot())
+
+[email protected]()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().morning())
+
+
+__plugin_name__ = 'good_noon'
+__doc__ = """做白日梦
+需at"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        BOT_CONFIG['noon']['command'])
+
+noon = on_command(BOT_CONFIG['noon']['command'][0],
+                    aliases=set(BOT_CONFIG['noon']['command']),
+                    rule=is_in_ban_list() & is_in_dormant() & to_bot())
+
+[email protected]()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().noon())
+
+
+__plugin_name__ = 'good_night'
+__doc__ = """晚安~！
+需at"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        BOT_CONFIG['night']['command'])
+
+night = on_command(BOT_CONFIG['night']['command'][0],
+                    aliases=set(BOT_CONFIG['night']['command']),
+                    rule=is_in_ban_list() & is_in_dormant() & to_bot())
+
+[email protected]()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().night())
+
+
+__plugin_name__ = 'cant_do_anything'
+__doc__ = """吃饭第一名！好吃就是高兴！！
+使用正则匹配"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        BOT_CONFIG['cantdo']['command'])
+
+cantdo = on_regex('|'.join(BOT_CONFIG['cantdo']['command']),
+                    rule=is_in_ban_list() & is_in_dormant())
+
+[email protected]()
+async def _(bot: Bot, event: Event) -> None:
+    await bot.send(event, ATRI().cantdo())
+
+
+# ===========================[End Command Processing]=============================
+
+callrobot_list = []
+
+
+class ATRI:
+    def call_robot(self, user: Optional[int]) -> str:
+        global callrobot_list
+        result = ''
+        for i in range(0, 5):
+            if count_list(callrobot_list, user) == i:
+                result = choice(BOT_CONFIG['callRobot']['repo'][i])
+                callrobot_list.append(user)
+                if count_list(callrobot_list, user) == 5:
+                    callrobot_list = del_list_aim(callrobot_list, user)
+                break
+            else:
+                continue
+        return result
+
+    def call_me(self) -> str:
+        return choice(BOT_CONFIG['atri']['repo'])
+
+    def tee_tee(self, user: Optional[int]) -> str:
+        if user in RUNTIME_CONFIG['superusers']:
+            return choice(BOT_CONFIG['teetee']['repo']['superusers'])
+        else:
+            return choice(BOT_CONFIG['teetee']['repo']['user'])
+
+    def kani(self) -> str:
+        return choice(BOT_CONFIG['kani']['repo'])
+
+    def waste(self) -> str:
+        return choice(BOT_CONFIG['waste']['repo'])
+
+    def morning(self) -> str:
+        period = BOT_CONFIG['morning']['repo']
+        if period[0]['period'][0] <= now_time() < period[0]['period'][1]:
+            return choice(period[0]['repo'])
+        elif period[1]['period'][0] <= now_time() < period[1]['period'][1]:
+            return choice(period[1]['repo'])
+        elif period[2]['period'][0] <= now_time() < period[2]['period'][1]:
+            return choice(period[2]['repo'])
+        elif period[3]['period'][0] <= now_time() < period[3]['period'][1]:
+            return choice(period[3]['repo'])
+        elif period[4]['period'][0] <= now_time() < period[4]['period'][1]:
+            return choice(period[4]['repo'])
+        else:
+            return choice(period['error'])
+
+    def noon(self) -> str:
+        if BOT_CONFIG['noon']['period'][0] <= now_time(
+        ) < BOT_CONFIG['noon']['period'][1]:
+            return choice(BOT_CONFIG['noon']['repo'])
+        else:
+            return choice(BOT_CONFIG['noon']['error'])
+
+    def night(self) -> str:
+        period = BOT_CONFIG['night']['repo']
+        if period[0]['period'][0] <= now_time() < period[0]['period'][1]:
+            return choice(period[0]['repo'])
+        elif period[1]['period'][0] <= now_time() < period[1]['period'][1]:
+            return choice(period[1]['repo'])
+        elif period[2]['period'][0] <= now_time() < period[2]['period'][1]:
+            return choice(period[2]['repo'])
+        elif period[3]['period'][0] <= now_time() < period[3]['period'][1]:
+            return choice(period[3]['repo'])
+        elif period[4]['period'][0] <= now_time() < period[4]['period'][1]:
+            return choice(period[4]['repo'])
+        else:
+            return choice(period['error'])
+
+    def cantdo(self) -> str:
+        return choice(BOT_CONFIG['cantdo']['repo'])