implement timed meeting status updates

Author: akhmerov

common.py

@@ -9,7 +9,7 @@
 SPEAKERS_CORNER_USER_ID = "D0n5UNEHQiajWtgdWLlNSA"
 
 @lru_cache()
-def zoom_headers(duration: int=10) -> dict:
+def zoom_headers(duration: int=100) -> dict:
     zoom_api_key = os.getenv("ZOOM_API_KEY")
     zoom_api_secret = os.getenv("ZOOM_API_SECRET")
     token = jwt.encode(
@@ -22,10 +22,52 @@ def zoom_headers(duration: int=10) -> dict:
     return {'authorization': f'Bearer {token}', 'content-type': 'application/json'}
 
 
+def zoom_request(method: callable, *args, **kwargs):
+    """A minimal wrapper around requests for querying zoom API with error handling"""
+    response = method(*args, **kwargs, headers=zoom_headers())
+    if response.status_code != 204:
+        raise RuntimeError(response.content.decode())
+
+    if response.content:
+        return response.json()
+
+
 def speakers_corner_user_id() -> str:
-    users = requests.get(ZOOM_API + "users", headers=zoom_headers()).json()["users"]
+    users = zoom_request(requests.get, ZOOM_API + "users")["users"]
     sc_user_id = next(
         u["id"] for u in users
         if u["first_name"] == "Speakers'" and u["last_name"] == "Corner"
         )
-    return sc_user_id
+    return sc_user_id
+
+
+def all_meetings(user_id) -> list:
+    """Return all meetings by a user.
+    
+    Handles pagination, and adds ``live: True`` to a meeting that is running (if any).
+    """
+    meetings = []
+    next_page_token = ""
+    while True:
+        meetings_page = zoom_request(
+            requests.get,
+            f"{ZOOM_API}users/{user_id}/meetings",
+            params={"type": "scheduled", "page_size": 300, "next_page_token": next_page_token}
+        )
+        meetings += meetings_page["meetings"]
+        next_page_token = meetings_page["next_page_token"]
+        if not next_page_token:
+            break
+    
+        live_meetings = zoom_request(
+            requests.get,
+            f"{ZOOM_API}users/{user_id}/meetings",
+            params={"type": "scheduled", "page_size": 300, "next_page_token": next_page_token}
+        )["meetings"]
+
+        if live_meetings:
+            for meeting in meetings:
+                if meeting["id"] == live_meetings[0]["id"]:
+                    meeting["live"] = True
+        
+        return meetings

host_key_rotation.py

@@ -7,8 +7,10 @@
 
 import requests
 import pytz
+from dateutil.parser import parse
 
 import common
+from common import zoom_request
 
 
 def host_key(timeslot: datetime.datetime) -> int:
@@ -21,12 +23,76 @@ def host_key(timeslot: datetime.datetime) -> int:
 
 def update_host_key():
     """Update the host key of the speakers' corner user for the upcoming hour."""
-    response = requests.patch(
+    zoom_request(
+        requests.patch,
         common.ZOOM_API + "users/" + common.SPEAKERS_CORNER_USER_ID,
-        headers=common.zoom_headers(),
         data=json.dumps({
             "host_key": host_key(datetime.datetime.now() + datetime.timedelta(hours=1))
         })
     )
-    if response.status_code != 204:
-        raise RuntimeError(response.content.decode())
+
+
+def rotate_meetings():
+    """Update the Speakers' corner meeting settings and statuses.
+    
+    1. If there is an upcoming meeting in less than an hour, allow joining
+       before host.
+    2. Stop the running meeting if there is an upcoming one or if it runs for too long.
+    3. Disable joining before host on recent meetings to prevent restarting.
+    """
+    now = datetime.datetime.now(tz=pytz.UTC)
+    sc_meetings = common.all_meetings(common.SPEAKERS_CORNER_USER_ID)
+    for m in sc_meetings:
+        m["start_time"] = parse(m["start_time"])
+
+    live = [m for m in sc_meetings if m["live"]]
+
+    try:
+        upcoming = min(
+            (m for m in sc_meetings if m["start_time"] > now),
+            key=(lambda meeting: meeting["start_time"])
+        )
+        upcoming_start = upcoming["start_time"]
+    except ValueError:
+        upcoming = None
+        upcoming_start = now + datetime.timedelta(weeks=1)
+    
+    recent = [
+        m for m in sc_meetings
+        if (now > m["start_time"] > now - datetime.timedelta(hours=2))
+        and not m["live"]
+    ]
+
+    starting_soon = upcoming_start - now < datetime.timedelta(hours=1)
+    if starting_soon:
+        common.zoom_request(
+            requests.patch,
+            f"{common.ZOOM_API}meetings/{upcoming['id']}",
+            data=json.dumps({"settings": {"join_before_host": True}}),
+        )
+
+    if (
+        live
+        and (
+            starting_soon 
+            or live[0]["start_time"] < now - datetime.timedelta(minutes=90)
+        )
+    ):
+        live_id = live[0]["id"]
+        common.zoom_request(
+            requests.put,
+            f"{common.ZOOM_API}meetings/{live_id}/status",
+            data=json.dumps({"action": "end"}),
+        )
+        common.zoom_request(
+            requests.patch,
+            f"{common.ZOOM_API}meetings/{live_id}",
+            data=json.dumps({"settings": {"join_before_host": False}}),
+        )
+
+    for meeting in recent:
+        common.zoom_request(
+            requests.patch,
+            f"{common.ZOOM_API}meetings/{meeting['id']}",
+            data=json.dumps({"settings": {"join_before_host": False}}),
+        )