Changeset cc34991 in python-cinema-club-bot for main.py

Timestamp:

Oct 9, 2024, 3:09:50 AM (5 weeks ago)

Author:

Mikhail Kirillov <w96k@…>

Branches:

master

Children:

0d48ea2

Parents:

0d67456

git-author:

Mikhail Kirillov <w96k@…> (10/09/24 03:05:41)

git-committer:

Mikhail Kirillov <w96k@…> (10/09/24 03:09:50)

Message:

Add commands modules; Added /chooser command

File:

• main.py (modified) (4 diffs)

main.py

@@ -10 +10 @@
 # <http://creativecommons.org/publicdomain/zero/1.0/>
 
-from telegram import Update, error
+from telegram import Update
 from telegram.ext import ApplicationBuilder, CommandHandler, ContextTypes, \
     MessageHandler, filters
@@ -16 +16 @@
 import os
 import logging
-from rich import inspect
 
-from strings import INVALID_COMMAND, USER_NOT_PROVIDED, USER_ADDED, \
-    USER_REMOVED
-from utils import context_init
+from strings import INVALID_COMMAND
 from persistence import Persistence
+import commands
 
 
@@ -32 +30 @@
 
 
-async def about(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
-    await update.message.reply_text('''
-Version 0.0
-
-python-cinema-club-bot is a bot for Telegram specialized for managing cinema club activities. This software is released as Public Domain using CC0 license.
-
-Project information: http://57.129.46.169/trac/wiki/python-cinema-club-bot
-Source code: http://57.129.46.169/cgit/python-cinema-club-bot/
-
-Contributed in 2024 by Mikhail Kirillov (~w96k) <w96k@runbox.com>
-
-To the extent possible under law, the author(s) have dedicated all copyright and related and neighboring rights to this software to the public domain worldwide. This software is distributed without any warranty.
-
-You should have received a copy of the CC0 Public Domain Dedication along with this software. If not, see:
-<http://creativecommons.org/publicdomain/zero/1.0/>
-
-    ''')
-
-
-async def add(
-        update: Update,
-        context: ContextTypes.DEFAULT_TYPE
-) -> None:
-    context_init(context)
-
-    if context.args == []:
-        await update.message.reply_text(USER_NOT_PROVIDED)
-        raise error.TelegramError(USER_NOT_PROVIDED)
-
-    for user in context.args:
-        context.chat_data["users"].append(user)
-        await update.message.reply_text(USER_ADDED.format(user=user))
-
-    await update.message.reply_text(context.chat_data)
-
-
-async def list(
-        update: Update,
-        context: ContextTypes.DEFAULT_TYPE
-) -> None:
-    context_init(context)
-
-    await update.message.reply_text(context.chat_data)
-
-
-async def remove(
-        update: Update,
-        context: ContextTypes.DEFAULT_TYPE
-) -> None:
-    context_init(context)
-
-    if context.args == []:
-        await update.message.reply_text(USER_NOT_PROVIDED)
-        raise error.TelegramError(USER_NOT_PROVIDED)
-
-    for user in context.args:
-        if user == "*":
-            context.chat_data["users"] = []
-            break
-        
-        context.chat_data["users"].remove(user)
-        await update.message.reply_text(USER_REMOVED.format(user=user))
-
-    await update.message.reply_text(context.chat_data)
-
-
 async def unknown(update: Update, context: ContextTypes.DEFAULT_TYPE):
     await context.bot.send_message(
@@ -105 +37 @@
 
 
-if __name__ == "__main__":    
+if __name__ == "__main__":
     app = ApplicationBuilder().token(
         os.environ.get("TELEGRAM_TOKEN")
     ).persistence(Persistence).build()
 
-    app.add_handler(CommandHandler("about", about))
-    app.add_handler(CommandHandler("add", add))
-    app.add_handler(CommandHandler("list", list))
-    app.add_handler(CommandHandler("remove", remove))
+    app.add_handler(CommandHandler("about", commands.about))
+    app.add_handler(CommandHandler("add", commands.add_users))
+    app.add_handler(CommandHandler("list", commands.list_users))
+    app.add_handler(CommandHandler("remove", commands.remove_users))
+    app.add_handler(CommandHandler("chooser", commands.chooser_user))