Commited backup

hikariatama/ftg/youdbettersleep.py

@@ -0,0 +1,111 @@
+#             █ █ ▀ █▄▀ ▄▀█ █▀█ ▀
+#             █▀█ █ █ █ █▀█ █▀▄ █
+#              © Copyright 2022
+#           https://t.me/hikariatama
+#
+# 🔒      Licensed under the GNU AGPLv3
+# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
+
+# meta pic: https://static.dan.tatar/youdbettersleep_icon.png
+# meta banner: https://mods.hikariatama.ru/badges/youdbettersleep.jpg
+# meta developer: @hikarimods
+# scope: hikka_only
+# scope: hikka_min 1.3.0
+
+import re
+import time
+
+from telethon.tl.types import Message
+
+from .. import loader, utils
+
+
+def s2time(temp_time: str) -> int:
+    seconds, minutes, hours = 0, 0, 0
+
+    try:
+        seconds = int(str(re.search("([0-9]+)s", temp_time).group(1)))
+    except Exception:
+        pass
+
+    try:
+        minutes = int(str(re.search("([0-9]+)m", temp_time).group(1))) * 60
+    except Exception:
+        pass
+
+    try:
+        hours = int(str(re.search("([0-9]+)h", temp_time).group(1))) * 60 * 60
+    except Exception:
+        pass
+
+    return round(seconds + minutes + hours)
+
+
+@loader.tds
+class YouDBetterSleepMod(loader.Module):
+    """Restricts user from sending messages while sleeping"""
+
+    strings = {
+        "name": "YouDBetterSleep",
+        "no_time": "🚫 <b>You can't sleep forever, specify <time> argument</b>",
+        "awake": "🥱 <b>Good morning</b>",
+        "asleep": "😴 <b>Good night. Now I can't write messages for {}</b>",
+        "disabled": (
+            "😴 <b>I can't write messages, because my userbot wants me to sleep</b>"
+        ),
+    }
+
+    strings_ru = {
+        "no_time": "👾 <b>Ты не можешь спать вечно, укажи аргумент <время></b>",
+        "awake": "🥱 <b>Доброе утро</b>",
+        "asleep": (
+            "😴 <b>Спокойной ночи. Я не могу писать сообщения на протяжении {}</b>"
+        ),
+        "disabled": (
+            "😴 <b>Я не могу писать сообщения, так как мой юзербот хочет, чтобы я"
+            " поспал</b>"
+        ),
+        "_cls_doc": "Запрещает писать во время сна",
+    }
+
+    @loader.command(ru_doc="<время> - Поспать")
+    async def sleepcmd(self, message: Message):
+        """<time> - Sleep for time"""
+        args = utils.get_args_raw(message)
+
+        t = s2time(args)
+
+        if not args or t == 0:
+            self.set("asleep", False)
+            self.get("until", 0)
+            await utils.answer(message, self.strings("awake"))
+        else:
+            self.set("asleep", True)
+            self.set("until", t + time.time())
+            await utils.answer(message, self.strings("asleep").format(args))
+
+    @loader.tag("only_messages", "no_commands")
+    async def watcher(self, message: Message):
+        if not self.get("asleep", False) or not self.get("until", False):
+            return
+
+        if self.get("until", 0) <= time.time():
+            self.set("until", 0)
+            self.set("asleep", False)
+            await self.inline.bot.send_message(
+                self._tg_id,
+                self.strings("awake"),
+                parse_mode="HTML",
+            )
+            return
+
+        if message.mentioned:
+            await self._client.send_read_acknowledge(
+                message.peer_id,
+                message,
+                clear_mentions=True,
+            )
+            await utils.answer(message, self.strings("disabled"))
+
+        if message.out:
+            await utils.answer(message, self.strings("disabled"))