# █ █ █ █▄▀ ▄▀█ █▀▄▀█ █▀█ █▀█ █ █ # █▀█ █ █ █ █▀█ █ ▀ █ █▄█ █▀▄ █▄█ # 🔒 Licensed under the GNU GPLv3 # 🌐 https://www.gnu.org/licenses/agpl-3.0.html # 👤 https://t.me/hikamoru # meta developer: @hikamorumods # meta banner: https://raw.githubusercontent.com/AmoreForever/assets/master/Alarm.jpg import re import pytz import random import logging import asyncio from datetime import datetime from .. import utils, loader logger = logging.getLogger(__name__) day_to_weekday = { "mon": 0, "tue": 1, "wed": 2, "thu": 3, "fri": 4, "sat": 5, "sun": 6, "пн": 0, "вт": 1, "ср": 2, "чт": 3, "пт": 4, "сб": 5, "вс": 6, } @loader.tds class AlarmMod(loader.Module): """Alarm module for remind you about something""" strings = { "name": "Alarm", "set": "⏰ Alarm set for {}!", "unset": "⏰ Alarm for {} unset!", "unset_all": "⏰ All alarms unset!", "list_item": ( "⏰ Alarm for {}! #{}" "\n🕔 Time: {}" "\n🔊 Message: {}" ), "no_alarms": "🙅‍♂️ No alarms!", "off_button": "✋ Off", "notification": "⏰ Alarm!\n\n{}", "turned_off": "✔️ Alarm turned off!", "incorrect_time": "🖕 Incorrect time!", "where_args": "🖕 Where arguments?", "incorrect_args": "🖕 Incorrect arguments! Write like this: .setalarm mon 12:00 text", "interval_doc": "Interval of sending notifications in seconds", "time_zone_doc": "Time zone for alarms (for example, Europe/Moscow)", } strings_ru = { "set": "⏰ Напоминание установлено на {}!", "unset": "⏰ Напоминание для {} отменено!", "unset_all": "⏰ Все напоминания отменены!", "list_item": ( "⏰ Напоминание для {}! #{}" "\n🕔 Время: {}" "\n🔊 Сообщение: {}" ), "no_alarms": "🙅‍♂️ Нет напоминаний!", "off_button": "✋ Выключить", "notification": "⏰ Напоминание!\n\n{}", "turned_off": "✔️ Напоминание выключено!", "incorrect_time": "🖕 Неправильное время!", "where_args": "🖕 Где аргументы?", "incorrect_args": "🖕 Неправильные аргументы! Пиши так: .setalarm пн 12:00 текст", "interval_doc": "Интервал отправления напоминаний в секундах", "time_zone_doc": "Часовой пояс для напоминаний (например, Europe/Moscow)", } def __init__(self): self.config = loader.ModuleConfig( loader.ConfigValue( "interval", 5, lambda: self.strings("interval_doc"), validator=loader.validators.Integer(minimum=1, maximum=60), ), loader.ConfigValue( "time_zone", "Europe/Moscow", lambda: self.strings("time_zone_doc"), validator=loader.validators.RegExp( r"^[\w/]+$", ) ), ) @loader.command(ru_doc="<день недели> <время> <сообщение> - установить напоминание") async def setalarm(self, message): """ - set alarm""" args = utils.get_args_raw(message) if not args: return await utils.answer(message, self.strings("where_args")) try: re_args = re.match(r"^(.+) (\d{1,2}):(\d{1,2}) (.*)$", args) day = re_args.group(1).lower() hour = int(re_args.group(2)) minute = int(re_args.group(3)) text = re_args.group(4) except AttributeError: return await utils.answer(message, self.strings("incorrect_args")) if not (0 <= hour <= 23 and 0 <= minute <= 59): return await utils.answer(message, self.strings("incorrect_time")) if day not in day_to_weekday.keys(): text = f"Wrong day of the week!\nAvailable days: {', '.join(day_to_weekday.keys())}" return await utils.answer(message, text) id_ = random.randint(100, 999) self.set( "alarms", { **self.get("alarms", {}), day: { "hour": hour, "minute": minute, "text": text, "id": id_, "status": "on", }, }, ) await utils.answer(message, self.strings("set").format(day)) @loader.command(ru_doc="получить список напоминаний") async def alarms(self, message): """get alarms list""" alarms = self.get("alarms", {}) if not alarms: return await utils.answer(message, self.strings("no_alarms")) text = "" for day, alarm in alarms.items(): text += self.strings("list_item").format( day, f"{alarm['id']}", f"{alarm['hour']}:{alarm['minute']}", alarm["text"], ) text += "\n\n" await utils.answer(message, text) @loader.command(ru_doc=" - отменить напоминание") async def unsetalarm(self, message): """ - unset alarm""" args = utils.get_args_raw(message) if not args: return await utils.answer(message, self.strings("where_time")) if args.startswith("#"): args = args[1:] alarms = self.get("alarms", {}) for day, alarm in alarms.items(): if str(alarm["id"]) == args: alarms.pop(day) self.set("alarms", alarms) return await utils.answer( message, self.strings("unset").format(day) ) await utils.answer(message, self.strings("unset").format(args)) @loader.command(ru_doc="отменить все напоминания") async def unsetallalarms(self, message): """unset all alarms""" self.set("alarms", {}) await utils.answer(message, self.strings("unset_all")) @loader.loop(interval=2, autostart=True) async def check_alarms(self): alarms = self.get("alarms", {}) if not alarms: return now = datetime.now(tz=pytz.timezone(self.config["time_zone"])) day = now.weekday() hour = now.hour minute = now.minute for day_, alarm in alarms.items(): if ( day_to_weekday[day_] == day and alarm["hour"] == hour and alarm["minute"] == minute ): while alarm["status"] == "on": self._markup = self.inline.generate_markup( { "text": self.strings("off_button"), "callback": self.off_alarm, "args": (alarm["id"],), } ) await self.inline.bot.send_message( self.tg_id, self.strings("notification").format(alarm["text"]), reply_markup=self._markup, ) await asyncio.sleep(self.config["interval"]) break async def off_alarm(self, call, id_): alarms = self.get("alarms", {}) for day, alarm in alarms.items(): if alarm["id"] == id_: alarm["status"] = "off" self.set("alarms", alarms) await call.edit(self.strings("turned_off")) return False await call.answer("Не найдено!")