generated from nhcarrigan/template
naomi
4fdb5d06f1
- 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
107 lines
2.9 KiB
Python
107 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
"""Remove cohort and team roles from inactive Discord members."""
|
|
|
|
import asyncio
|
|
import os
|
|
|
|
import aiohttp
|
|
|
|
DISCORD_BOT_TOKEN = os.environ["DISCORD_BOT_TOKEN"]
|
|
GUILD_ID = "1354624415861833870"
|
|
BASE_URL = "https://discord.com/api/v10"
|
|
|
|
INACTIVE_MEMBERS = [
|
|
"1177481351490981889",
|
|
"899092786802987069",
|
|
"1318882254365397032",
|
|
"680429718511943770",
|
|
"1195902319976521748",
|
|
"1424001797072359576",
|
|
"1333378962289590365",
|
|
"799293680799711273",
|
|
"1183395404293869662",
|
|
"1325958873831440566",
|
|
"717028253633871965",
|
|
"847789364217905182",
|
|
"746703502369095700",
|
|
"192541018908786690",
|
|
"1017761694514163712",
|
|
]
|
|
|
|
COHORT_ROLE_ID = "1464314780935258112"
|
|
TEAM_ROLE_IDS = {
|
|
1: "1464314923780931677",
|
|
2: "1464315093402784015",
|
|
3: "1464315098452726106",
|
|
4: "1464315105264275600",
|
|
5: "1464315109873684593",
|
|
6: "1464315114378498152",
|
|
7: "1464315118904152107",
|
|
8: "1464315124251754559",
|
|
9: "1464315128437801177",
|
|
10: "1464315132896088168",
|
|
11: "1464315138428633241",
|
|
12: "1464315142710890520",
|
|
13: "1464315149203804405",
|
|
14: "1464315153599299803",
|
|
}
|
|
|
|
MEMBER_TO_TEAM = {
|
|
"1177481351490981889": 1,
|
|
"899092786802987069": 1,
|
|
"1318882254365397032": 2,
|
|
"680429718511943770": 2,
|
|
"1195902319976521748": 3,
|
|
"1424001797072359576": 4,
|
|
"1333378962289590365": 4,
|
|
"799293680799711273": 5,
|
|
"1183395404293869662": 7,
|
|
"1325958873831440566": 8,
|
|
"717028253633871965": 8,
|
|
"847789364217905182": 10,
|
|
"746703502369095700": 10,
|
|
"192541018908786690": 11,
|
|
"1017761694514163712": 13,
|
|
}
|
|
|
|
|
|
async def remove_role(
|
|
session: aiohttp.ClientSession, user_id: str, role_id: str
|
|
) -> bool:
|
|
"""Remove a role from a user."""
|
|
url = f"{BASE_URL}/guilds/{GUILD_ID}/members/{user_id}/roles/{role_id}"
|
|
headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"}
|
|
|
|
async with session.delete(url, headers=headers) as resp:
|
|
if resp.status == 204:
|
|
return True
|
|
text = await resp.text()
|
|
print(f" Error removing role {role_id} from {user_id}: {resp.status} - {text}")
|
|
return False
|
|
|
|
|
|
async def main() -> None:
|
|
"""Remove roles from all inactive members."""
|
|
async with aiohttp.ClientSession() as session:
|
|
print("Removing roles from inactive members...\n")
|
|
|
|
for member_id in INACTIVE_MEMBERS:
|
|
print(f"Processing {member_id}...")
|
|
|
|
if await remove_role(session, member_id, COHORT_ROLE_ID):
|
|
print(" ✓ Removed cohort role")
|
|
|
|
if member_id in MEMBER_TO_TEAM:
|
|
team_id = MEMBER_TO_TEAM[member_id]
|
|
team_role_id = TEAM_ROLE_IDS[team_id]
|
|
if await remove_role(session, member_id, team_role_id):
|
|
print(f" ✓ Removed Team {team_id} role")
|
|
|
|
await asyncio.sleep(1.5)
|
|
|
|
print("\nDone!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|