TeamPGM/PagerMaid_Plugins_Pyro
2
0
Fork 0
mirror of https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git synced 2024-09-28 06:45:05 +00:00
Code Issues Actions Projects Releases Wiki Activity

sendat 修复错误

Browse Source
This commit is contained in:
xtaodada 2022-07-12 15:36:13 +08:00
parent de96ea7093
commit 15e58060c3
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
2 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
list.json
View File

@ -292,7 +292,7 @@
},
{
"name": "sendat",
"version": "1.001",
"version": "1.002",
"section": "chat",
"maintainer": "xtaodada",
"size": "12.2 kb",

12
sendat/main.py
View File

@ -139,7 +139,7 @@ class SendTasks:
def add(self, task: SendTask):
for i in self.tasks:
if i.cid == task.cid:
if i.task_id == task.task_id:
return
self.tasks.append(task)
@ -164,13 +164,7 @@ class SendTasks:
return "\n".join(task.export_str(show_all) for task in self.tasks if task.cid == cid or show_all)
def save_to_file(self):
data = sqlite.get("sendat_tasks", [])
for task in self.tasks:
for i in data:
if i["task_id"] == task.task_id:
data.remove(i)
break
data.append(task.export())
data = [task.export() for task in self.tasks]
sqlite["sendat_tasks"] = data
def load_from_file(self):
@ -310,6 +304,7 @@ async def send_at(message: Message):
if uid := await from_msg_get_task_id(message):
send_tasks.remove(uid)
send_tasks.save_to_file()
send_tasks.load_from_file()
return await message.edit(f"已删除任务 {uid}")
elif message.parameter[0] == "pause":
if uid := await from_msg_get_task_id(message):
@ -335,4 +330,5 @@ async def send_at(message: Message):
send_tasks.add(task)
send_tasks.register_single_task(task)
send_tasks.save_to_file()
send_tasks.load_from_file()
await message.edit(f"已添加任务 {task.task_id}")