Changeset 0d48ea2 in python-cinema-club-bot for commands/users.py

Timestamp:

Oct 9, 2024, 3:40:27 AM (5 weeks ago)

Author:

Mikhail Kirillov <w96k@…>

Branches:

master

Children:

6623428

Parents:

cc34991

Message:

Add /set command; Add error_handler;

File:

• commands/users.py (modified) (6 diffs)

commands/users.py

@@ -14 +14 @@
 from collections import deque
 
-from strings import USER_NOT_PROVIDED, USER_ADDED, USER_REMOVED, EXPECTED_ONE_USER
+from strings import USER_NOT_PROVIDED, USERS_ADDED, USERS_REMOVED, \
+    EXPECTED_ONE_USER, USER_SET, USER_ADD, USER_REMOVE
 from utils import context_init
 
-from rich import inspect
 
+async def set_users(
+        update: Update,
+        context: ContextTypes.DEFAULT_TYPE
+) -> None:
+    context_init(context)
+
+    if context.args == []:
+        raise error.TelegramError(USER_NOT_PROVIDED)
+
+    context.chat_data["users"] = context.args
+
+    await update.message.reply_text(USER_SET)
 
 async def add_users(
@@ -27 +39 @@
 
     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(USER_ADD.format(user=user))
 
-    await update.message.reply_text(context.chat_data["users"])
+    await update.message.reply_text(USERS_ADDED)
 
 
@@ -53 +64 @@
 
     if context.args == []:
-        await update.message.reply_text(USER_NOT_PROVIDED)
         raise error.TelegramError(USER_NOT_PROVIDED)
 
@@ -62 +72 @@
 
         context.chat_data["users"].remove(user)
-        await update.message.reply_text(USER_REMOVED.format(user=user))
+        await update.message.reply_text(USER_REMOVE.format(user=user))
 
-    await update.message.reply_text(context.chat_data["users"])
+    await update.message.reply_text(USERS_REMOVED)
 
 
@@ -74 +84 @@
 
     if context.args == []:
-        await update.message.reply_text(USER_NOT_PROVIDED)
         raise error.TelegramError(USER_NOT_PROVIDED)
 
     if len(context.args) > 1:
-        await update.message.reply_text(EXPECTED_ONE_USER)
         raise error.TelegramError(EXPECTED_ONE_USER)
 
@@ -89 +97 @@
     context.chat_data["users"] = list(users)
 
-    await update.message.reply_text(context.chat_data)
+    await update.message.reply_text(context.chat_data["users"])