diff --git a/Tools/erase_user_data.py b/Tools/erase_user_data.py index 198ba2070b..0cc1a31d93 100644 --- a/Tools/erase_user_data.py +++ b/Tools/erase_user_data.py @@ -12,7 +12,7 @@ import os import psycopg2 from uuid import UUID -LATEST_DB_MIGRATION = "20230725193102_AdminNotesImprovementsForeignKeys" +LATEST_DB_MIGRATION = "20250211131539_LoadoutNames" def main(): parser = argparse.ArgumentParser() @@ -43,6 +43,7 @@ def main(): clear_server_role_ban(cur, user_id) clear_uploaded_resource_log(cur, user_id) clear_whitelist(cur, user_id) + clear_blacklist(cur, user_id) print("Committing...") conn.commit() @@ -197,6 +198,16 @@ WHERE user_id = %s """, (user_id,)) +def clear_blacklist(cur: "psycopg2.cursor", user_id: str): + print("Clearing blacklist...") + + cur.execute(""" +DELETE FROM + blacklist +WHERE + user_id = %s +""", (user_id,)) + main()