PagerMaid-Modify/some-plugins/sendat.py

""" Send Msg At A Specified Time. """

# By tg @fruitymelon
# extra requirements: dateparser

imported = True

import os, sys, time, traceback

try:
    import dateparser
except ImportError:
    imported = False

import asyncio
from pagermaid import log 
from pagermaid.listener import listener
from datetime import datetime
from dateutil import parser


def logsync(message):
    sys.stdout.writelines(f"{message}\n")

logsync("sendat: loading... If failed, please install dateparser first.")

# https://stackoverflow.com/questions/1111056/get-time-zone-information-of-the-system-in-python
def local_time_offset(t=None):
    """Return offset of local zone from GMT, either at present or at time t."""
    # python2.3 localtime() can't take None
    if t is None:
        t = time.time()

    if time.localtime(t).tm_isdst and time.daylight:
        return -time.altzone
    else:
        return -time.timezone

offset = local_time_offset() // 3600
sign = "+" if offset >= 0 else "-"
offset = abs(offset)
offset_str = str(offset)
offset_str = offset_str if len(offset_str) == 2 else f"0{offset_str}"

settings = {'TIMEZONE': f'{sign}{offset_str}00'}

logsync(f"sendat: local time zone offset is {sign}{abs(offset)}")

mem = []

helpmsg = """
定时发送消息。
-sendat 时间 | 消息内容

i.e.
-sendat 16:00:00 | 投票截止！
-sendat every 23:59:59 | 又是无所事事的一天呢。
-sendat every 1 minutes | 又过去了一分钟。
-sendat *3 1 minutes | 此消息将出现三次，间隔均为一分钟。
"""

@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):
    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
    await context.edit(f"tz data: {time.timezone} {time.tzname} {sign}{offset}")
    if len(args) != 2:
        await context.edit("Invalid argument. Expected: 2")
        return
    if offset is None:
        await context.edit("Failed to get server timezone.")
        return
    try:
        if args[0].find("every ") == 0:
            # at this point, let's assume args[0] contains relative time
            # i.e. -sendat every 3 minutes
            time_str = args[0][6:]
            if time_str.find(":") != -1:
                # then it should be absolute time
                sleep_times = [abs(dateparser.parse(time_str, settings=settings).timestamp() - time.time()), 24 * 60 * 60]
                index = 0
                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
            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])
            rest = " ".join(args[0][1:].split(" ")[1:])
            if rest.find(":") != -1:
                # then it should be absolute time
                sleep_times = [abs(dateparser.parse(rest, settings=settings).timestamp() - time.time()), 24 * 60 * 60]
                count = 0
                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
            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
        dt = parser.parse(args[0])
        delta = dt.timestamp() - time.time()
        if delta < 3:
            await context.edit("Target time before now.")
            return
        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


@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)