From e6040d1b25b87877ace873b18f4a0bda7045b744 Mon Sep 17 00:00:00 2001
From: Pieter-Jan Briers <pieterjan.briers+git@gmail.com>
Date: Sun, 4 May 2025 16:11:28 +0200
Subject: [PATCH] Update GDPR erase script to latest DB schema (#37162)

---
 Tools/erase_user_data.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

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