summaryrefslogtreecommitdiff
path: root/ATRI/plugins/setu/__init__.py
diff options
context:
space:
mode:
authorLint Action <[email protected]>2021-07-08 14:09:51 +0000
committerLint Action <[email protected]>2021-07-08 14:09:51 +0000
commit951dcdf4ec224b31433c5ab5c170b76fd3087910 (patch)
treefe0ee2b5c219bd8f9796917ac3ee6286d4181146 /ATRI/plugins/setu/__init__.py
parentbe2747e4d4b820ca0f1f988d3b77a628da26fe7b (diff)
downloadATRI-951dcdf4ec224b31433c5ab5c170b76fd3087910.tar.gz
ATRI-951dcdf4ec224b31433c5ab5c170b76fd3087910.tar.bz2
ATRI-951dcdf4ec224b31433c5ab5c170b76fd3087910.zip
:rotating_light: 自动进行代码格式化
Diffstat (limited to 'ATRI/plugins/setu/__init__.py')
-rw-r--r--ATRI/plugins/setu/__init__.py24
1 files changed, 11 insertions, 13 deletions
diff --git a/ATRI/plugins/setu/__init__.py b/ATRI/plugins/setu/__init__.py
index 7eb2c61..d67d75b 100644
--- a/ATRI/plugins/setu/__init__.py
+++ b/ATRI/plugins/setu/__init__.py
@@ -12,7 +12,10 @@ _setu_flmt = FreqLimiter(120)
_setu_dlmt = DailyLimiter(5)
-random_setu = Setu().on_command("来张涩图", "来张随机涩图,冷却2分钟,每天限5张", aliases={"涩图来", "来点涩图", "来份涩图"})
+random_setu = Setu().on_command(
+ "来张涩图", "来张随机涩图,冷却2分钟,每天限5张", aliases={"涩图来", "来点涩图", "来份涩图"}
+)
+
@random_setu.handle()
async def _random_setu(bot: Bot, event: MessageEvent):
@@ -21,12 +24,9 @@ async def _random_setu(bot: Bot, event: MessageEvent):
await random_setu.finish()
if not _setu_dlmt.check(user_id):
await random_setu.finish()
-
+
setu, title, p_id = await Setu().random_setu()
- repo = (
- f"Title: {title}\n"
- f"Pid: {p_id}"
- )
+ repo = f"Title: {title}\n" f"Pid: {p_id}"
await bot.send(event, repo)
msg_1 = await bot.send(event, Message(setu))
event_id = msg_1["message_id"]
@@ -38,6 +38,7 @@ async def _random_setu(bot: Bot, event: MessageEvent):
tag_setu = Setu().on_regex(r"来[张点丶份](.*?)的[涩色🐍]图", "根据提供的tag查找涩图")
+
@tag_setu.handle()
async def _tag_setu(bot: Bot, event: MessageEvent):
user_id = event.get_user_id()
@@ -45,18 +46,15 @@ async def _tag_setu(bot: Bot, event: MessageEvent):
await random_setu.finish()
if not _setu_dlmt.check(user_id):
await random_setu.finish()
-
+
msg = str(event.message).strip()
pattern = r"来[张点丶份](.*?)的[涩色🐍]图"
tag = re.findall(pattern, msg)[0]
setu, title, p_id, is_ok = await Setu().tag_setu(tag)
if not is_ok:
await tag_setu.finish(f"没有 {tag} 的涩图呢...")
- repo_0 = (
- f"Title: {title}\n"
- f"Pid: {p_id}"
- )
-
+ repo_0 = f"Title: {title}\n" f"Pid: {p_id}"
+
await bot.send(event, repo_0)
msg_1 = await bot.send(event, Message(setu))
event_id = msg_1["message_id"]
@@ -77,6 +75,6 @@ async def _scheduler_setu(bot):
message_id = msg_0["message_id"]
await asyncio.sleep(60)
await bot.delete_msg(message_id=message_id)
-
+
except BaseException:
pass