generated from nhcarrigan/template
naomi
a40188413a
All Python cohort scripts now use DATA_DIR = Path(__file__).parent.parent.parent / "data" to correctly resolve the repo-root data/ directory regardless of the working directory set by run.sh. All TypeScript scripts have expanded JSDoc headers documenting data file requirements and environment variables.
105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Assign team-specific roles to all 155 participants.
|
|
Respects Discord rate limits with proper backoff and retry logic.
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
import time
|
|
from pathlib import Path
|
|
|
|
import requests
|
|
|
|
DATA_DIR = Path(__file__).parent.parent.parent / "data"
|
|
|
|
BOT_TOKEN = os.environ["DISCORD_BOT_TOKEN"]
|
|
GUILD_ID = "692816967895220344"
|
|
|
|
BASE_URL = "https://discord.com/api/v10"
|
|
HEADERS = {"Authorization": f"Bot {BOT_TOKEN}", "Content-Length": "0"}
|
|
|
|
TEAM_ROLE_IDS = {
|
|
"Jade Jasmine": "1464314923780931677",
|
|
"Crimson Dahlia": "1464315093402784015",
|
|
"Rose Camellia": "1464315098452726106",
|
|
"Amber Wisteria": "1464315105264275600",
|
|
"Ivory Orchid": "1464315109873684593",
|
|
"Teal Iris": "1464315114378498152",
|
|
"Peach Gardenia": "1464315118904152107",
|
|
"Violet Carnation": "1464315124251754559",
|
|
"Azure Lotus": "1464315128437801177",
|
|
"Coral Sunflower": "1464315132896088168",
|
|
"Indigo Tulip": "1464315138428633241",
|
|
"Scarlet Hydrangea": "1464315142710890520",
|
|
"Mint Narcissus": "1464315149203804405",
|
|
"Sage Marigold": "1464315153599299803",
|
|
}
|
|
|
|
|
|
def assign_role_with_retry(user_id: str, role_id: str, max_retries: int = 5) -> bool:
|
|
url = f"{BASE_URL}/guilds/{GUILD_ID}/members/{user_id}/roles/{role_id}"
|
|
|
|
for attempt in range(max_retries):
|
|
response = requests.put(url, headers=HEADERS)
|
|
|
|
if response.status_code == 204:
|
|
return True
|
|
elif response.status_code == 429:
|
|
# Check headers first, fall back to JSON body
|
|
retry_after = response.headers.get("Retry-After")
|
|
if retry_after is None:
|
|
retry_after = response.headers.get("X-RateLimit-Reset-After")
|
|
if retry_after is None:
|
|
try:
|
|
retry_after = response.json().get("retry_after", 1)
|
|
except Exception:
|
|
retry_after = 1
|
|
retry_after = float(retry_after)
|
|
print(f" Rate limited! Waiting {retry_after:.2f}s before retry...")
|
|
time.sleep(retry_after)
|
|
else:
|
|
print(f" Error {response.status_code}: {response.text}")
|
|
backoff_time = (2**attempt) * 0.5
|
|
print(f" Retrying in {backoff_time:.2f}s...")
|
|
time.sleep(backoff_time)
|
|
|
|
return False
|
|
|
|
|
|
def main():
|
|
with open(DATA_DIR / "team_assignments.json") as f:
|
|
teams = json.load(f)
|
|
|
|
print(f"Assigning team roles to {len(teams)} teams...")
|
|
print("-" * 50)
|
|
|
|
total_success = 0
|
|
total_fail = 0
|
|
|
|
for team in teams:
|
|
team_name = team["name"]
|
|
role_id = TEAM_ROLE_IDS[team_name]
|
|
all_members = team["leaders"] + team["participants"]
|
|
|
|
print(f"\n[{team_name}] Assigning role to {len(all_members)} members...")
|
|
|
|
for user_id in all_members:
|
|
print(f" {user_id}...", end=" ")
|
|
|
|
if assign_role_with_retry(user_id, role_id):
|
|
print("✓")
|
|
total_success += 1
|
|
else:
|
|
print("✗ FAILED")
|
|
total_fail += 1
|
|
|
|
# Small delay between requests to be nice to the API
|
|
time.sleep(0.1)
|
|
|
|
print("-" * 50)
|
|
print(f"Complete! Success: {total_success}, Failed: {total_fail}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|