Cleanups and bugfixes for (#2 and #8)

modules/utils/autocomplete_utils.py

@@ -6,7 +6,7 @@ from bson import ObjectId
 from discord import AutocompleteContext, OptionChoice
 from pymongo import ASCENDING
 
-from modules.database import col_events, col_stages
+from modules.database import col_events, col_stages, col_users
 
 
 async def autocomplete_timezones(ctx: AutocompleteContext) -> List[str]:
@@ -29,7 +29,7 @@ async def autocomplete_active_events(ctx: AutocompleteContext) -> List[OptionCho
     query: Dict[str, Any] = {
         "ended": None,
         "ends": {"$gt": datetime.now(tz=ZoneInfo("UTC"))},
-        "cancelled": {"$ne": True},
+        "is_cancelled": {"$ne": True},
     }
 
     event_names: List[OptionChoice] = []
@@ -40,6 +40,43 @@ async def autocomplete_active_events(ctx: AutocompleteContext) -> List[OptionCho
     return event_names
 
 
+async def autocomplete_user_available_events(ctx: AutocompleteContext) -> List[OptionChoice]:
+    """Return list of active events user can register in"""
+
+    return await autocomplete_active_events(ctx)
+
+
+async def autocomplete_user_registered_events(ctx: AutocompleteContext) -> List[OptionChoice]:
+    """Return list of active events user is registered in"""
+
+    pipeline: List[Dict[str, Any]] = [
+        {
+            "$lookup": {
+                "from": "events",
+                "localField": "registered_event_ids",
+                "foreignField": "_id",
+                "as": "registered_events",
+            }
+        },
+        {
+            "$match": {
+                "registered_events.ended": None,
+                "registered_events.ends": {"$gt": datetime.now(tz=ZoneInfo("UTC"))},
+                "registered_events.starts": {"$gt": datetime.now(tz=ZoneInfo("UTC"))},
+                "registered_events.is_cancelled": {"$ne": True},
+            }
+        },
+    ]
+
+    event_names: List[OptionChoice] = []
+
+    async for result in col_users.aggregate(pipeline):
+        for registered_event in result["registered_events"]:
+            event_names.append(OptionChoice(registered_event["name"], str(registered_event["_id"])))
+
+    return event_names
+
+
 async def autocomplete_event_stages(ctx: AutocompleteContext) -> List[OptionChoice]:
     """Return list of stages of the event"""