implement host key generation

Author: akhmerov

common.py

@@ -0,0 +1,31 @@
+from functools import lru_cache
+import os
+from time import time
+
+import jwt
+import requests
+
+ZOOM_API = "https://api.zoom.us/v2/"
+SPEAKERS_CORNER_USER_ID = "D0n5UNEHQiajWtgdWLlNSA"
+
+@lru_cache()
+def zoom_headers(duration: int=10) -> dict:
+    zoom_api_key = os.getenv("ZOOM_API_KEY")
+    zoom_api_secret = os.getenv("ZOOM_API_SECRET")
+    token = jwt.encode(
+        # Create a payload of the token containing API Key & expiration time
+        {"iss": zoom_api_key, "exp": time() + duration},
+        zoom_api_secret,
+        algorithm='HS256'
+    ).decode('utf-8')
+
+    return {'authorization': f'Bearer {token}', 'content-type': 'application/json'}
+
+
+def speakers_corner_user_id() -> str:
+    users = requests.get(ZOOM_API + "users", headers=zoom_headers()).json()["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

host_key_rotation.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+
+import os
+import hashlib
+import datetime
+import json
+
+import requests
+import pytz
+
+import common
+
+
+def host_key(timeslot: datetime.datetime) -> int:
+    """Generate a host key for a specified time."""
+    key_salt = os.getenv("HOST_KEY_SALT").encode()
+    timestamp = timeslot.replace(second=0, microsecond=0, minute=0).timestamp()
+    hashed = hashlib.sha512(int(timestamp.to_bytes(5, "big")) + key_salt)
+    return f"{int(hashed.hexdigest(), 16) % int(1e7):06}"
+
+
+def update_host_key():
+    """Update the host key of the speakers' corner user for the upcoming hour."""
+    response = 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())