ephemere/python/cohort/remove_discord_roles.py

#!/usr/bin/env python3
"""Remove the Cohort and team-specific Discord roles from a list of members.

Update INACTIVE_MEMBERS and MEMBER_TO_TEAM before running to target the correct
members. Removes both the cohort-wide role and the member's team role.

Data files (place in data/):
  - None (member IDs and team mappings are defined as constants in the script)

Env vars:
  - DISCORD_BOT_TOKEN  Bot token for the Discord API
"""

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())