leads-db/drop_tables.py

21 lines
534 B
Python

import os
import django
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "leadsdb.settings")
django.setup()
from django.db import connection
with connection.cursor() as cursor:
cursor.execute("""
DO $$ DECLARE
r RECORD;
BEGIN
FOR r IN (SELECT tablename FROM pg_tables WHERE schemaname = current_schema()) LOOP
EXECUTE 'DROP TABLE IF EXISTS ' || quote_ident(r.tablename) || ' CASCADE';
END LOOP;
END $$;
""")
print("All tables have been dropped.")