Changeset 9be02d9 in python-cinema-club-bot for commands/movie.py

Timestamp:

Oct 10, 2024, 4:12:00 AM (5 weeks ago)

Author:

Mikhail Kirillov <w96k@…>

Branches:

master

Children:

8d2d5ac

Parents:

6623428

Message:

Fix #9. Add events commands

File:

• commands/movie.py (modified) (5 diffs)

commands/movie.py

@@ -13 +13 @@
 from telegram.ext import ContextTypes
 from imdb import Cinemagoer
-from pprint import pformat
 
-from utils import context_init
-
+from utils import context_init, choose_next_user
+from predicates import has_finished_event
 from strings import MOVIE_NOT_PROVIDED, EXPECTED_ONE_MOVIE, \
     MOVIE_ANOTHER_USER, FETCHING_MOVIE, FETCHING_ERROR, \
-    MOVIE_REMOVE, MOVIE_SET
+    MOVIE_REMOVE, MOVIE_SET, MOVIES_LIST, NO_MOVIES, ADD_MORE_USERS
 
 
@@ -31 +30 @@
     context_init(context)
 
-    chooser = context.chat_data["users"][0] or None
+    users = context.chat_data["users"]
+
+    if users == []:
+        raise error.TelegramError(ADD_MORE_USERS)
+
+    chooser = users[0]
     username = update.message.from_user.username
+
+    if has_finished_event(context, chooser):
+        users = context.chat_data["users"] = choose_next_user(users)
+        raise error.TelegramError(MOVIE_ANOTHER_USER.format(users[0]))
 
     if "@"+username != chooser:
@@ -51 +59 @@
     except:
         raise error.TelegramError(FETCHING_ERROR)
-    
+
     movie_dict = dict(
         title=movie.data.get("title"),
         id=movie.getID(),
         user=update.effective_user.username,
-        poster=movie.data.get("cover url")
+        poster=movie.data.get("cover url"),
+        rating=movie.data.get("rating"),
+        genres=movie.data.get("genres"),
+        runtime=movie.data.get("runtimes"),
     )
 
@@ -76 +87 @@
     context_init(context)
 
-    movies = context.chat_data["movies"].copy()
+    movies = context.chat_data["movies"]
+
+    movies_formatted: str = ""
 
     for movie in movies:
-        if movie["poster"] or None:
-            del movie["poster"]
+        movies_formatted += MOVIES_LIST.format(
+            title=movie["title"],
+            id=movie["id"],
+            user=movie["user"]
+        )
 
-    await update.message.reply_text(pformat(movies))
+    if movies_formatted == "":
+        movies_formatted = NO_MOVIES
+
+    await update.message.reply_text(movies_formatted)
 
 
@@ -101 +120 @@
                 context.chat_data["movies"].remove(movie)
                 await update.message.reply_text(
-                    MOVIE_REMOVE.format(title=movie["title"], id=["movie.id"])
+                    MOVIE_REMOVE.format(title=movie["title"], id=movie["id"])
                 )
-