mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
89 lines
3.7 KiB
Python
89 lines
3.7 KiB
Python
from typing import *
|
|
import dateparser
|
|
import datetime
|
|
import pickle
|
|
import telegram
|
|
import discord
|
|
from sqlalchemy import and_
|
|
import royalnet.commands as rc
|
|
import royalnet.utils as ru
|
|
import royalnet.serf.telegram as rst
|
|
import royalnet.serf.discord as rsd
|
|
|
|
from ..tables import Reminder
|
|
|
|
|
|
class ReminderCommand(rc.Command):
|
|
name: str = "reminder"
|
|
|
|
aliases = ["calendar"]
|
|
|
|
description: str = "Ti ricorda di fare qualcosa dopo un po' di tempo."
|
|
|
|
syntax: str = "[ {data} ] {messaggio}"
|
|
|
|
def __init__(self, serf, config):
|
|
super().__init__(serf, config)
|
|
|
|
session = self.alchemy.Session()
|
|
reminders = (
|
|
session.query(self.alchemy.get(Reminder))
|
|
.filter(and_(
|
|
self.alchemy.get(Reminder).datetime >= datetime.datetime.now(),
|
|
self.alchemy.get(Reminder).interface_name == self.serf.interface_name))
|
|
.all()
|
|
)
|
|
for reminder in reminders:
|
|
self.serf.tasks.add(self._remind(reminder))
|
|
|
|
async def _remind(self, reminder):
|
|
await ru.sleep_until(reminder.datetime)
|
|
if isinstance(self.serf, rst.TelegramSerf):
|
|
chat_id: int = pickle.loads(reminder.interface_data)
|
|
client: telegram.Bot = self.serf.client
|
|
await self.serf.api_call(client.send_message,
|
|
chat_id=chat_id,
|
|
text=rst.escape(f"❗️ {reminder.message}"),
|
|
parse_mode="HTML",
|
|
disable_web_page_preview=True)
|
|
elif isinstance(self.serf, rsd.DiscordSerf):
|
|
channel_id: int = pickle.loads(reminder.interface_data)
|
|
client: discord.Client = self.serf.client
|
|
channel = client.get_channel(channel_id)
|
|
await channel.send(rsd.escape(f"❗️ {reminder.message}"))
|
|
|
|
async def run(self, args: rc.CommandArgs, data: rc.CommandData) -> None:
|
|
try:
|
|
date_str, reminder_text = args.match(r"\[\s*([^]]+)\s*]\s*([^\n]+)\s*")
|
|
except rc.InvalidInputError:
|
|
date_str, reminder_text = args.match(r"\s*(.+?)\s*\n\s*([^\n]+)\s*")
|
|
|
|
try:
|
|
date: Optional[datetime.datetime] = dateparser.parse(date_str, settings={
|
|
"PREFER_DATES_FROM": "future"
|
|
})
|
|
except OverflowError:
|
|
date = None
|
|
if date is None:
|
|
await data.reply("⚠️ La data che hai inserito non è valida.")
|
|
return
|
|
if date <= datetime.datetime.now():
|
|
await data.reply("⚠️ La data che hai specificato è nel passato.")
|
|
return
|
|
await data.reply(f"✅ Promemoria impostato per [b]{date.strftime('%Y-%m-%d %H:%M:%S')}[/b]")
|
|
if isinstance(self.serf, rst.TelegramSerf):
|
|
interface_data = pickle.dumps(data.message.chat.id)
|
|
elif isinstance(self.serf, rsd.DiscordSerf):
|
|
interface_data = pickle.dumps(data.message.channel.id)
|
|
else:
|
|
raise rc.UnsupportedError("This command does not support the current interface.")
|
|
async with data.session_acm() as session:
|
|
creator = await data.find_author(session=session)
|
|
reminder = self.alchemy.get(Reminder)(creator=creator,
|
|
interface_name=self.serf.interface_name,
|
|
interface_data=interface_data,
|
|
datetime=date,
|
|
message=reminder_text)
|
|
self.serf.tasks.add(self._remind(reminder))
|
|
session.add(reminder)
|
|
await ru.asyncify(session.commit)
|