From e60a0570b4bbe12270d4aacf3833b58c9771a313 Mon Sep 17 00:00:00 2001
From: xtaodada <1344089750@qq.com>
Date: Mon, 6 Apr 2020 14:16:01 +0800
Subject: [PATCH] =?UTF-8?q?=E2=8F=B2=20sendat=20=E6=94=AF=E6=8C=81?=
 =?UTF-8?q?=E5=AF=BC=E5=85=A5/=E5=AF=BC=E5=87=BA=E8=AE=B0=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 some-plugins/sendat.py | 54 +++++++++++++++++++++++++++++++++++-------
 1 file changed, 46 insertions(+), 8 deletions(-)

diff --git a/some-plugins/sendat.py b/some-plugins/sendat.py
index 731922a..2d8ef42 100644
--- a/some-plugins/sendat.py
+++ b/some-plugins/sendat.py
@@ -46,6 +46,7 @@ settings = {'TIMEZONE': f'{sign}{offset_str}00'}
 
 logsync(f"sendat: local time zone offset is {sign}{abs(offset)}")
 
+mem = []
 
 helpmsg = """
 定时发送消息。
@@ -61,9 +62,13 @@ i.e.
 @listener(outgoing=True, command="sendat", diagnostics=True, ignore_edited=False,
           description=helpmsg,
           parameters="<atmsg>")
+async def sendatwrap(context):
+    await sendat(context)
+
 async def sendat(context):
-    args = " ".join(context.parameter).split("|")
+    mem_id = len(mem)
     chat = await context.get_chat()
+    args = " ".join(context.parameter).split("|")
     if not imported:
         await context.edit("Please install dateparser first: python3 -m pip install dateparser")
         return
@@ -83,24 +88,28 @@ async def sendat(context):
                 # then it should be absolute time
                 sleep_times = [abs(dateparser.parse(time_str, settings=settings).timestamp() - time.time()), 24 * 60 * 60]
                 index = 0
-                await context.edit(f"Registered.")
+                mem.append("|".join(args))
+                await context.edit(f"Registered: id {mem_id}")
                 while True:
                     last_time = time.time()
                     while time.time() < last_time + sleep_times[index]:
                         await asyncio.sleep(2)
                     await sendmsg(context, chat, args[1])
                     index = 1
+                mem[mem_id] = ""
                 return
             sleep_time = abs(dateparser.parse(time_str, settings=settings).timestamp() - time.time())
             if sleep_time < 5:
                 await context.edit(f"Sleep time too short. Should be longer than 5 seconds. Got {sleep_time}") 
                 return
-            await context.edit(f"Registered.")
+            mem.append("|".join(args))
+            await context.edit(f"Registered: id {mem_id}")
             while True:
                 last_time = time.time()
                 while time.time() < last_time + sleep_time:
                     await asyncio.sleep(2)
                 await sendmsg(context, chat, args[1])
+            mem[mem_id] = ""
             return
         elif args[0].find("*") == 0:
             times = int(args[0][1:].split(" ")[0])
@@ -109,26 +118,30 @@ async def sendat(context):
                 # then it should be absolute time
                 sleep_times = [abs(dateparser.parse(rest, settings=settings).timestamp() - time.time()), 24 * 60 * 60]
                 count = 0
-                await context.edit("Registered.")
+                mem.append("|".join(args))
+                await context.edit(f"Registered: id {mem_id}")
                 while count <= times:
                     last_time = time.time()
                     while time.time() < last_time + sleep_times[0 if count == 0 else 1]:
                         await asyncio.sleep(2)
                     await sendmsg(context, chat, args[1])
                     count += 1
+                mem[mem_id] = ""
                 return
             sleep_time = abs(dateparser.parse(rest, settings=settings).timestamp() - time.time())
             if sleep_time < 5:
                 await context.edit("Sleep time too short. Should be longer than 5 seconds.")
                 return
             count = 0
-            await context.edit("Registered.")
+            mem.append("|".join(args))
+            await context.edit(f"Registered: id {mem_id}")
             while count <= times:
                 last_time = time.time()
                 while time.time() < last_time + sleep_time:
                     await asyncio.sleep(2)
                 await sendmsg(context, chat, args[1])
                 count += 1
+            mem[mem_id] = ""
             return
         
         # absolute time
@@ -137,17 +150,42 @@ async def sendat(context):
         if delta < 3:
             await context.edit("Target time before now.")
             return
-        await context.edit("Registered.")
+        mem.append("|".join(args))
+        await context.edit(f"Registered: id {mem_id}")
         while delta > 0:
             delta = dt.timestamp() - time.time()
             await asyncio.sleep(2)
         await sendmsg(context, chat, args[1])
+        mem[mem_id] = ""
     except Exception as e:
         await log(str(e))
         await log(str(traceback.format_stack()))
         return
 
 
-async def sendmsg(context, chat, text):
-    await context.client.send_message(chat, text)
+@listener(outgoing=True, command="sendatdump", diagnostics=True, ignore_edited=False,
+          description="导出 sendat 消息")
+async def sendatdump(context):
+    if mem.count("") != 0:
+        await context.edit(".\n-sendat " + "\n-sendat ".join(mem[:].remove("")))
+    else:
+        await context.edit(".\n-sendat " + "\n-sendat ".join(mem))
 
+@listener(outgoing=True, command="sendatparse", diagnostics=True, ignore_edited=True,
+          description="导入已导出的 sendat 消息")
+async def sendatparse(context):
+    chat = await context.get_chat()
+    text = "\n".join(context.message.text.split("\n")[1:])
+    if text == "":
+        return
+    if text.find(".\n") == 0:
+        text = "\n".join(text.split("\n")[1:])
+    lines = text.split("\n")
+    for i in range(len(lines)):
+        line = lines[i]
+        sent = await sendmsg(context, chat, line)
+        sent.parameter = line.replace("-sendat ", "").split(" ")
+        await sendat(sent)
+
+async def sendmsg(context, chat, text):
+    return await context.client.send_message(chat, text)
\ No newline at end of file