Update GDPR erase script to latest DB schema (#37162)

This commit is contained in:
Pieter-Jan Briers
2025-05-04 16:11:28 +02:00
committed by GitHub
parent 20da7fcd25
commit e6040d1b25

View File

@@ -12,7 +12,7 @@ import os
import psycopg2 import psycopg2
from uuid import UUID from uuid import UUID
LATEST_DB_MIGRATION = "20230725193102_AdminNotesImprovementsForeignKeys" LATEST_DB_MIGRATION = "20250211131539_LoadoutNames"
def main(): def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
@@ -43,6 +43,7 @@ def main():
clear_server_role_ban(cur, user_id) clear_server_role_ban(cur, user_id)
clear_uploaded_resource_log(cur, user_id) clear_uploaded_resource_log(cur, user_id)
clear_whitelist(cur, user_id) clear_whitelist(cur, user_id)
clear_blacklist(cur, user_id)
print("Committing...") print("Committing...")
conn.commit() conn.commit()
@@ -197,6 +198,16 @@ WHERE
user_id = %s user_id = %s
""", (user_id,)) """, (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() main()