TeamPGM/PagerMaid_Plugins_Pyro
2
0
Fork 0
mirror of https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git synced 2024-11-16 07:39:26 +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", "name": "sendat",
"version": "1.001", "version": "1.002",
"section": "chat", "section": "chat",
"maintainer": "xtaodada", "maintainer": "xtaodada",
"size": "12.2 kb", "size": "12.2 kb",

12
sendat/main.py
View File

@ -139,7 +139,7 @@ class SendTasks:
def add(self, task: SendTask): def add(self, task: SendTask):
for i in self.tasks: for i in self.tasks:
if i.cid == task.cid: if i.task_id == task.task_id:
return return
self.tasks.append(task) 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) 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): def save_to_file(self):
data = sqlite.get("sendat_tasks", []) data = [task.export() for task in self.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())
sqlite["sendat_tasks"] = data sqlite["sendat_tasks"] = data
def load_from_file(self): def load_from_file(self):
@ -310,6 +304,7 @@ async def send_at(message: Message):
if uid := await from_msg_get_task_id(message): if uid := await from_msg_get_task_id(message):
send_tasks.remove(uid) send_tasks.remove(uid)
send_tasks.save_to_file() send_tasks.save_to_file()
send_tasks.load_from_file()
return await message.edit(f"已删除任务 {uid}") return await message.edit(f"已删除任务 {uid}")
elif message.parameter[0] == "pause": elif message.parameter[0] == "pause":
if uid := await from_msg_get_task_id(message): 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.add(task)
send_tasks.register_single_task(task) send_tasks.register_single_task(task)
send_tasks.save_to_file() send_tasks.save_to_file()
send_tasks.load_from_file()
await message.edit(f"已添加任务 {task.task_id}") await message.edit(f"已添加任务 {task.task_id}")