Large refactor

plugins/balance.py

@@ -0,0 +1,63 @@
+# -*- coding: utf-8 -*-
+
+import logging
+from traceback import format_exc
+
+from libbot import config_get, json_read
+from pyrogram import filters
+from pyrogram.client import Client
+from pyrogram.enums.chat_action import ChatAction
+from pyrogram.types import Message
+
+from modules.utils import string, userGet
+from modules.bwt import getWaterLeft
+
+logger = logging.getLogger(__name__)
+
+
+@Client.on_message(
+    ~filters.scheduled & filters.command(["balance", "баланс"], prefixes=["/", ""])  # type: ignore
+)
+async def command_balance(app: Client, msg: Message):
+    try:
+        if "card" in (await json_read("data/database.json"))[str(msg.from_user.id)]:
+            await app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
+            water_left = await getWaterLeft(
+                userGet(msg.from_user.id, "card"), msg.from_user.id, app
+            )
+            if water_left == "":
+                await msg.reply_text(
+                    (await string("error_new")).format(
+                        f'https://bwtaqua.com.ua/card-topup/?id={userGet(msg.from_user.id, "card")}'
+                    )
+                )
+                # raise EmptyCardException("Card information is empty")
+            elif water_left == "Failure":
+                await msg.reply_text(
+                    (await string("error_occured")).format(await string("get_number"))
+                )
+                logger.warning(
+                    f"User {str(msg.from_user.id)} could not get left water amount"
+                )
+            else:
+                await msg.reply_text((await string("card_balance")).format(water_left))
+                logger.info(
+                    f"User {str(msg.from_user.id)} has {water_left} liters remaining"
+                )
+        else:
+            await msg.reply_text(
+                (await string("card_not_linked")).format(await string("get_number"))
+            )
+            logger.info(
+                f"User {str(msg.from_user.id)} tried to get balance without card set"
+            )
+    except Exception as exp:
+        if msg.from_user.id != await config_get("owner_id"):
+            await msg.reply_text(
+                (await string("error_occured")).format(await string("get_number"))
+            )
+        await app.send_message(
+            await config_get("owner_id"),
+            f"Error occured by {str(msg.from_user.id)}:\nException: `{exp}`\nTraceback: `{format_exc()}`",
+        )
+        logger.warning(f"User {str(msg.from_user.id)} could not get left water amount")

plugins/help.py

@@ -0,0 +1,24 @@
+# -*- coding: utf-8 -*-
+
+import logging
+
+from libbot import json_read
+from pyrogram import filters
+from pyrogram.client import Client
+from pyrogram.types import Message
+
+from modules.utils import string, userSet
+
+logger = logging.getLogger(__name__)
+
+
+@Client.on_message(
+    ~filters.scheduled
+    & filters.command(["start", "help", "допомога"], prefixes=["/", ""])  # type: ignore
+)
+async def command_help(app: Client, msg: Message):
+    await msg.reply_text((await string("welcome")).format(await string("get_number")))
+    if msg.from_user.language_code in await json_read("strings.json"):
+        userSet(msg.from_user.id, "locale", msg.from_user.language_code)
+    else:
+        userSet(msg.from_user.id, "locale", "en")

plugins/register_commands.py

@@ -0,0 +1,60 @@
+# -*- coding: utf-8 -*-
+
+import logging
+
+from libbot import config_get
+from pyrogram import filters
+from pyrogram.client import Client
+from pyrogram.types import BotCommand, BotCommandScopeChat, Message
+
+logger = logging.getLogger(__name__)
+
+
+@Client.on_message(
+    ~filters.scheduled & filters.command(["register_commands"], prefixes="/")  # type: ignore
+)
+async def command_register_commands(app: Client, msg: Message):
+    if msg.from_user.id == await config_get("owner_id"):
+        app.set_bot_commands(
+            [
+                BotCommand("help", "Меню допомоги"),
+                BotCommand("balance", "Баланс картки"),
+                BotCommand("topup", "Поповнити картку"),
+                BotCommand("setcard", "Прив'язати картку"),
+                BotCommand("resetcard", "Відв'язати картку"),
+            ],
+            language_code="uk",
+        )  # type: ignore
+
+        app.set_bot_commands(
+            [
+                BotCommand("help", "Меню допомоги"),
+                BotCommand("balance", "Баланс картки"),
+                BotCommand("topup", "Поповнити картку"),
+                BotCommand("setcard", "Прив'язати картку"),
+                BotCommand("resetcard", "Відв'язати картку"),
+            ],
+            language_code="ru",
+        )  # type: ignore
+
+        app.set_bot_commands(
+            [
+                BotCommand("help", "Help menu"),
+                BotCommand("balance", "Card's balance"),
+                BotCommand("topup", "Refill card"),
+                BotCommand("setcard", "Link card"),
+                BotCommand("resetcard", "Unlink card"),
+            ]
+        )  # type: ignore
+
+        app.set_bot_commands(
+            [
+                BotCommand("help", "Help menu"),
+                BotCommand("balance", "Card's balance"),
+                BotCommand("topup", "Refill card"),
+                BotCommand("setcard", "Link card"),
+                BotCommand("resetcard", "Unlink card"),
+                BotCommand("shutdown", "Turn off the bot"),
+            ],
+            scope=BotCommandScopeChat(chat_id=await config_get("owner_id")),
+        )  # type: ignore

plugins/resetcard.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+
+import logging
+
+from libbot import json_read
+from pyrogram import filters
+from pyrogram.client import Client
+from pyrogram.types import Message
+
+from modules.utils import string, userReset
+
+logger = logging.getLogger(__name__)
+
+
+@Client.on_message(
+    ~filters.scheduled
+    & filters.command(["resetcard", "забути картку"], prefixes=["/", ""])  # type: ignore
+)
+async def command_resetcard(app: Client, msg: Message):
+    if "card" in (await json_read("data/database.json"))[str(msg.from_user.id)]:
+        userReset(msg.from_user.id, "card")
+        await msg.reply_text(await string("card_unlinked"))
+        logger.info(f"User {str(msg.from_user.id)} reseted his card")
+    else:
+        await msg.reply_text(
+            (await string("card_not_linked")).format(await string("get_number"))
+        )
+        logger.info(f"User {str(msg.from_user.id)} tried to reset non-existent card")

plugins/setcard.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+
+import logging
+
+from convopyro import listen_message
+from pyrogram import filters
+from pyrogram.client import Client
+from pyrogram.types import ForceReply, Message, ReplyKeyboardRemove
+
+from modules.utils import string, userSet
+
+logger = logging.getLogger(__name__)
+
+
+@Client.on_message(
+    ~filters.scheduled
+    & filters.command(["setcard", "задать карту"], prefixes=["/", ""])  # type: ignore
+)
+async def command_setcard(app: Client, msg: Message):
+    await msg.reply_text(
+        await string("send_number"),
+        reply_markup=ForceReply(placeholder=await string("enter_number")),
+    )
+    answer = await listen_message(app, msg.chat.id, timeout=None)
+    if answer is None:
+        return
+    elif answer.text.strip() in ["/cancel", "cancel", "/відміна", "відміна"]:
+        await msg.reply_text(await string("cancel"), reply_markup=ReplyKeyboardRemove())
+        return
+    userSet(answer.from_user.id, "card", answer.text)
+    logger.info(f"User {str(msg.from_user.id)} set card id to {answer.text}")
+    await msg.reply_text(
+        (await string("card_linked")).format(answer.text),
+        reply_markup=ReplyKeyboardRemove(),
+    )

plugins/shutdown.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+
+import logging
+from os import getpid
+
+from libbot import config_get
+from pyrogram import filters
+from pyrogram.client import Client
+from pyrogram.types import Message
+
+logger = logging.getLogger(__name__)
+
+
+@Client.on_message(
+    ~filters.scheduled & filters.command(["kill", "die", "shutdown"], prefixes="/")  # type: ignore
+)
+async def kill(app: Client, msg: Message):
+    if msg.from_user.id == await config_get("owner_id"):
+        await msg.reply_text(f"Shutting down bot with pid **{getpid()}**")
+        logger.info(f"Shutting down as requested by {msg.from_user.id}")
+        exit()

plugins/topup.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+
+import logging
+
+from libbot import json_read
+from pyrogram import filters
+from pyrogram.client import Client
+from pyrogram.enums.chat_action import ChatAction
+from pyrogram.types import Message
+
+from modules.utils import string, userGet
+
+logger = logging.getLogger(__name__)
+
+
+@Client.on_message(
+    ~filters.scheduled
+    & filters.command(["topup", "refill", "поповнити"], prefixes=["/", ""])  # type: ignore
+)
+async def command_topup(app: Client, msg: Message):
+    try:
+        if "card" in (await json_read("data/database.json"))[str(msg.from_user.id)]:
+            await app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
+            await msg.reply_text(
+                (await string("top_up")).format(str(userGet(msg.from_user.id, "card")))
+            )
+            logger.info(f"User {str(msg.from_user.id)} requested top up")
+        else:
+            await msg.reply_text(
+                (await string("card_not_linked")).format(await string("get_number"))
+            )
+            logger.info(
+                f"User {str(msg.from_user.id)} tried to request top up without card set"
+            )
+    except Exception as exp:
+        await msg.reply_text(str(exp))