feat: port remaining cohort scripts and make reusable

- Port 19 cohort scripts from /home/naomi/docs/cohort/ - Replace all hardcoded tokens and dotenv usage with os.environ - Add pandas==3.0.1 dependency - Add E501 to ruff ignore list for Discord message string content - Make remove_resigned_members.py reusable (empty RESIGNED_IDS constant) - Make update_roster_messages.py reusable (iterates all teams from JSON) - Exclude 12 one-off/event-specific scripts as non-reusable
2026-02-23 15:23:10 -08:00
parent e481823e06
commit 4fdb5d06f1
20 changed files with 2108 additions and 1 deletions
@@ -0,0 +1,100 @@
+"""Update team roster messages in Discord from team_assignments.json."""
+
+import asyncio
+import json
+import os
+
+import aiohttp
+
+DISCORD_BOT_TOKEN = os.environ["DISCORD_BOT_TOKEN"]
+GUILD_ID = "692816967895220344"
+
+API_BASE = "https://discord.com/api/v10"
+
+
+async def edit_message(
+    session: aiohttp.ClientSession,
+    channel_id: str,
+    message_id: str,
+    content: str,
+) -> bool:
+    """Edit a Discord message."""
+    url = f"{API_BASE}/channels/{channel_id}/messages/{message_id}"
+    headers = {
+        "Authorization": f"Bot {DISCORD_BOT_TOKEN}",
+        "Content-Type": "application/json",
+    }
+    payload = {"content": content}
+
+    async with session.patch(url, headers=headers, json=payload) as resp:
+        if resp.status == 200:
+            return True
+        text = await resp.text()
+        print(f"    ❌ Failed to edit message: {resp.status} - {text}")
+        return False
+
+
+def generate_roster(team: dict, discord_to_github: dict) -> str:
+    """Generate roster message for a team."""
+    team_name = team["name"]
+
+    leader_lines = [
+        f"- <@{discord_id}> ({discord_to_github.get(discord_id, 'Unknown')})"
+        for discord_id in team["leaders"]
+    ]
+
+    participant_lines = [
+        f"- <@{discord_id}> ({discord_to_github.get(discord_id, 'Unknown')})"
+        for discord_id in team["participants"]
+    ]
+
+    leaders_text = "\n".join(leader_lines) if leader_lines else "None"
+    participants_text = "\n".join(participant_lines) if participant_lines else "None"
+
+    return f"""# Team {team_name}
+
+**Leaders:**
+{leaders_text}
+
+**Participants:**
+{participants_text}"""
+
+
+async def main() -> None:
+    """Update roster messages for all teams."""
+    with open("team_message_ids.json") as f:
+        team_data = json.load(f)
+
+    with open("team_assignments.json") as f:
+        teams = json.load(f)
+
+    with open("discord_to_github.json") as f:
+        discord_to_github = json.load(f)
+
+    async with aiohttp.ClientSession() as session:
+        for team in teams:
+            team_name = team["name"]
+            print(f"Updating roster for {team_name}...")
+
+            if team_name not in team_data:
+                print("  ⚠️  Team not found in team_message_ids.json")
+                continue
+
+            channel_id = team_data[team_name]["channel_id"]
+            message_id = team_data[team_name]["message_id"]
+
+            roster_content = generate_roster(team, discord_to_github)
+
+            success = await edit_message(
+                session, channel_id, message_id, roster_content
+            )
+            if success:
+                print(f"  ✅ Updated (Message ID: {message_id})")
+
+            await asyncio.sleep(0.5)
+
+        print("\n✅ All roster messages updated!")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())