generated from nhcarrigan/template
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
This commit is contained in:
@@ -0,0 +1,106 @@
|
||||
#!/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())
|
||||
Reference in New Issue
Block a user