mirror of
https://github.com/TeamPGM/PagerMaid_Plugins.git
synced 2024-11-25 21:16:28 +00:00
334 lines
13 KiB
Python
334 lines
13 KiB
Python
|
""" 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 | 此消息将出现三次，间隔均为一分钟。
|
|||
|
|
|
|||
|
|
根据 Registered id，删除已注册的 timer：
|
|||
|
|
-sendat rm 2
|
|||
|
|
|
|||
|
|
此命令可与 autorespond（也是我写的插件）, ghost, deny 命令串联：
|
|||
|
|
|
|||
|
|
-sendat 10 minutes | -autorespond 我暂时有事离开一下。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
@listener(is_plugin=True, outgoing=True, command="sendat", diagnostics=True, ignore_edited=True,
|
|||
|
|
description=helpmsg,
|
|||
|
|
parameters="<atmsg>")
|
|||
|
|
async def sendatwrap(context):
|
|||
|
|
await sendat(context)
|
|||
|
|
|
|||
|
|
async def sendat(context):
|
|||
|
|
if not context.parameter:
|
|||
|
|
await context.edit(helpmsg)
|
|||
|
|
return
|
|||
|
|
mem_id = len(mem)
|
|||
|
|
mem.append("[NULL]")
|
|||
|
|
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\nAnd restart your pagermaid process")
|
|||
|
|
return
|
|||
|
|
await context.edit(f"tz data: {time.timezone} {time.tzname} {sign}{offset}")
|
|||
|
|
if len(args) != 2:
|
|||
|
|
if args[0].find("rm ") == 0:
|
|||
|
|
# clear timer
|
|||
|
|
target_ids = args[0][3:].strip().split(" ")
|
|||
|
|
for target_id in target_ids:
|
|||
|
|
if target_id.isnumeric():
|
|||
|
|
if len(mem) > int(target_id):
|
|||
|
|
mem[int(target_id)] = ""
|
|||
|
|
await context.edit(f"id {target_id} removed.")
|
|||
|
|
else:
|
|||
|
|
await context.edit("id out of range.")
|
|||
|
|
return
|
|||
|
|
else:
|
|||
|
|
await context.edit("id should be a integer.")
|
|||
|
|
return
|
|||
|
|
else:
|
|||
|
|
await context.edit("Invalid argument counts. Expecting: 2")
|
|||
|
|
return
|
|||
|
|
if offset is None:
|
|||
|
|
await context.edit("Failed to get your server's 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[mem_id] = "|".join(args)
|
|||
|
|
await context.edit(f"Registered: id {mem_id}. You can use this id to cancel the timer.")
|
|||
|
|
while True and mem[mem_id] != "":
|
|||
|
|
last_time = time.time()
|
|||
|
|
while time.time() < last_time + sleep_times[index] and mem[mem_id] != "":
|
|||
|
|
await asyncio.sleep(2)
|
|||
|
|
if mem[mem_id] != "":
|
|||
|
|
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[mem_id] = "|".join(args)
|
|||
|
|
await context.edit(f"Registered: id {mem_id}. You can use this id to cancel the timer.")
|
|||
|
|
while True and mem[mem_id] != "":
|
|||
|
|
last_time = time.time()
|
|||
|
|
while time.time() < last_time + sleep_time and mem[mem_id] != "":
|
|||
|
|
await asyncio.sleep(2)
|
|||
|
|
if mem[mem_id] != "":
|
|||
|
|
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[mem_id] = "|".join(args)
|
|||
|
|
await context.edit(f"Registered: id {mem_id}. You can use this id to cancel the timer.")
|
|||
|
|
while count <= times and mem[mem_id] != "":
|
|||
|
|
last_time = time.time()
|
|||
|
|
while time.time() < last_time + sleep_times[0 if count == 0 else 1] and mem[mem_id] != "":
|
|||
|
|
await asyncio.sleep(2)
|
|||
|
|
if mem[mem_id] != "":
|
|||
|
|
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[mem_id] = "|".join(args)
|
|||
|
|
await context.edit(f"Registered: id {mem_id}. You can use this id to cancel the timer.")
|
|||
|
|
while count <= times and mem[mem_id] != "":
|
|||
|
|
last_time = time.time()
|
|||
|
|
while time.time() < last_time + sleep_time and mem[mem_id] != "":
|
|||
|
|
await asyncio.sleep(2)
|
|||
|
|
if mem[mem_id] != "":
|
|||
|
|
await sendmsg(context, chat, args[1])
|
|||
|
|
count += 1
|
|||
|
|
mem[mem_id] = ""
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if args[0].find(":") == -1:
|
|||
|
|
# relative time
|
|||
|
|
dt = dateparser.parse(args[0])
|
|||
|
|
delta = time.time() - dt.timestamp()
|
|||
|
|
if delta < 3:
|
|||
|
|
await context.edit("Target time before now.")
|
|||
|
|
return
|
|||
|
|
mem[mem_id] = "|".join(args)
|
|||
|
|
await context.edit(f"Registered: id {mem_id}. You can use this id to cancel the timer.")
|
|||
|
|
while dt.timestamp() + 2*delta > time.time() and mem[mem_id] != "":
|
|||
|
|
await asyncio.sleep(2)
|
|||
|
|
if mem[mem_id] != "":
|
|||
|
|
await sendmsg(context, chat, args[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[mem_id] = "|".join(args)
|
|||
|
|
await context.edit(f"Registered: id {mem_id}. You can use this id to cancel the timer.")
|
|||
|
|
while delta > 0 and mem[mem_id] != "":
|
|||
|
|
delta = dt.timestamp() - time.time()
|
|||
|
|
await asyncio.sleep(2)
|
|||
|
|
if mem[mem_id] != "":
|
|||
|
|
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=True,
|
|||
|
|
description="导出并转储内存中的 sendat 配置")
|
|||
|
|
async def sendatdump(context):
|
|||
|
|
clean_mem = mem[:]
|
|||
|
|
if clean_mem.count("[NULL]") != 0:
|
|||
|
|
clean_mem.remove("[NULL]")
|
|||
|
|
if clean_mem.count("") != 0:
|
|||
|
|
clean_mem.remove("")
|
|||
|
|
await context.edit(".\n-sendat " + "\n-sendat ".join(clean_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)
|
|||
|
|
|
|||
|
|
""" Modified pagermaid autorespond plugin. """
|
|||
|
|
|
|||
|
|
from telethon.events import StopPropagation
|
|||
|
|
from pagermaid import persistent_vars
|
|||
|
|
|
|||
|
|
async def autorespond(context):
|
|||
|
|
""" Enables the auto responder. """
|
|||
|
|
message = "我还在睡觉... ZzZzZzZzZZz"
|
|||
|
|
if context.arguments:
|
|||
|
|
message = context.arguments
|
|||
|
|
await context.edit("成功启用自动响应器。")
|
|||
|
|
await log(f"启用自动响应器，将自动回复 `{message}`.")
|
|||
|
|
persistent_vars.update({'autorespond': {'enabled': True, 'message': message, 'amount': 0}})
|
|||
|
|
|
|||
|
|
from pagermaid import redis, redis_status
|
|||
|
|
|
|||
|
|
async def ghost(context):
|
|||
|
|
""" Toggles ghosting of a user. """
|
|||
|
|
if not redis_status():
|
|||
|
|
await context.edit("出错了呜呜呜 ~ Redis 好像离线了，无法执行命令。")
|
|||
|
|
return
|
|||
|
|
if len(context.parameter) != 1:
|
|||
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|||
|
|
return
|
|||
|
|
myself = await context.client.get_me()
|
|||
|
|
self_user_id = myself.id
|
|||
|
|
if context.parameter[0] == "true":
|
|||
|
|
if context.chat_id == self_user_id:
|
|||
|
|
await context.edit("在？为什么要在收藏夹里面用？")
|
|||
|
|
return
|
|||
|
|
redis.set("ghosted.chat_id." + str(context.chat_id), "true")
|
|||
|
|
await context.delete()
|
|||
|
|
await log(f"已成功将 ChatID {str(context.chat_id)} 添加到自动已读对话列表中。")
|
|||
|
|
elif context.parameter[0] == "false":
|
|||
|
|
if context.chat_id == self_user_id:
|
|||
|
|
await context.edit("在？为什么要在收藏夹里面用？")
|
|||
|
|
return
|
|||
|
|
redis.delete("ghosted.chat_id." + str(context.chat_id))
|
|||
|
|
await context.delete()
|
|||
|
|
await log(f"已成功将 ChatID {str(context.chat_id)} 从自动已读对话列表中移除。")
|
|||
|
|
elif context.parameter[0] == "status":
|
|||
|
|
if redis.get("ghosted.chat_id." + str(context.chat_id)):
|
|||
|
|
await context.edit("emm...当前对话已被加入自动已读对话列表中。")
|
|||
|
|
else:
|
|||
|
|
await context.edit("emm...当前对话已从自动已读对话列表中移除。")
|
|||
|
|
else:
|
|||
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def deny(context):
|
|||
|
|
""" Toggles denying of a user. """
|
|||
|
|
if not redis_status():
|
|||
|
|
await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行。")
|
|||
|
|
return
|
|||
|
|
if len(context.parameter) != 1:
|
|||
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|||
|
|
return
|
|||
|
|
myself = await context.client.get_me()
|
|||
|
|
self_user_id = myself.id
|
|||
|
|
if context.parameter[0] == "true":
|
|||
|
|
if context.chat_id == self_user_id:
|
|||
|
|
await context.edit("在？为什么要在收藏夹里面用？")
|
|||
|
|
return
|
|||
|
|
redis.set("denied.chat_id." + str(context.chat_id), "true")
|
|||
|
|
await context.delete()
|
|||
|
|
await log(f"ChatID {str(context.chat_id)} 已被添加到自动拒绝对话列表中。")
|
|||
|
|
elif context.parameter[0] == "false":
|
|||
|
|
if context.chat_id == self_user_id:
|
|||
|
|
await context.edit("在？为什么要在收藏夹里面用？")
|
|||
|
|
return
|
|||
|
|
redis.delete("denied.chat_id." + str(context.chat_id))
|
|||
|
|
await context.delete()
|
|||
|
|
await log(f"ChatID {str(context.chat_id)} 已从自动拒绝对话列表中移除。")
|
|||
|
|
elif context.parameter[0] == "status":
|
|||
|
|
if redis.get("denied.chat_id." + str(context.chat_id)):
|
|||
|
|
await context.edit("emm...当前对话已被加入自动拒绝对话列表中。")
|
|||
|
|
else:
|
|||
|
|
await context.edit("emm...当前对话已从自动拒绝对话列表移除。")
|
|||
|
|
else:
|
|||
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def sendmsg(context, chat, origin_text):
|
|||
|
|
text = origin_text.strip()
|
|||
|
|
msg = await context.client.send_message(chat, text)
|
|||
|
|
if text.find("-autorespond") == 0:
|
|||
|
|
msg.arguments = text.replace("-autorespond", "").lstrip()
|
|||
|
|
await autorespond(msg)
|
|||
|
|
elif text.find("-ghost ") == 0:
|
|||
|
|
msg.parameter = [text[7:]]
|
|||
|
|
await ghost(msg)
|
|||
|
|
elif text.find("-deny ") == 0:
|
|||
|
|
msg.parameter = [text[6:]]
|
|||
|
|
await deny(msg)
|
|||
|
|
return msg
|