Changeset 69dd60c in python-cinema-club-bot for utils.py

Timestamp:

Oct 11, 2024, 1:53:24 AM (5 weeks ago)

Author:

Mikhail Kirillov <w96k@…>

Branches:

master

Children:

ac7b16a

Parents:

694d823

Message:

Task #30. Allow anyone create or edit upcoming event

File:

• utils.py (modified) (2 diffs)

utils.py

@@ -12 +12 @@
 from telegram.ext import ContextTypes
 from collections import deque
+from typing import Optional
 
 def context_init(context: ContextTypes.DEFAULT_TYPE):
@@ -38 +39 @@
 
 
-def choose_next_user(users: list[dict]) -> list[dict]:
-    users = deque(users)
+def choose_next_user(context: ContextTypes.DEFAULT_TYPE) -> list[dict]:
+    users = deque(context.chat_data["users"])
     users.rotate(-1)  # -1 moves list to left by 1 element
 
+    init_new_event(context, users[0])
+
     return list(users)
+
+
+def init_new_event(context: ContextTypes.DEFAULT_TYPE, user: dict, movie: Optional[str] = None):
+    events = context.chat_data["events"]
+
+    last_event = events[-1] if events != [] else None
+
+    init_event = dict(
+        when=None,
+        where=None,
+        movie=movie,
+        user=user
+    )
+
+    if last_event and last_event["movie"] is None:
+        events[-1] = init_event
+    else:
+        events.append(init_event)