#290 Fix silk_clear_request_log errors on Postgres (#291)

jazzband · Jun 13, 2018 · 639da91 · 639da91
1 parent 33f2d28
commit 639da91
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/silk/management/commands/silk_clear_request_log.py b/silk/management/commands/silk_clear_request_log.py
@@ -14,16 +14,15 @@ def delete_model(model):
         table = model._meta.db_table
         if 'mysql' in engine or 'postgresql' in engine:
             # Use "TRUNCATE" on the table
-            cursor = connection.cursor()
-            if 'mysql' in engine:
-                cursor.execute("SET FOREIGN_KEY_CHECKS=0;")
-            if 'postgres' in engine:
-                cursor.execute("ALTER TABLE %s DISABLE TRIGGER ALL;", [table])
-            cursor.execute("TRUNCATE TABLE %s", [table])
-            if 'mysql' in engine:
-                cursor.execute("SET FOREIGN_KEY_CHECKS=1;")
-            if 'postgres' in engine:
-                cursor.execute("ALTER TABLE %s ENABLE TRIGGER ALL;", [table])
+            with connection.cursor() as cursor:
+                if 'mysql' in engine:
+                    cursor.execute("SET FOREIGN_KEY_CHECKS=0;")
+                    cursor.execute("TRUNCATE TABLE {0}".format(table))
+                    cursor.execute("SET FOREIGN_KEY_CHECKS=1;")
+                elif 'postgres' in engine:
+                    cursor.execute("ALTER TABLE {0} DISABLE TRIGGER ALL;".format(table))
+                    cursor.execute("TRUNCATE TABLE {0} CASCADE".format(table))
+                    cursor.execute("ALTER TABLE {0} ENABLE TRIGGER ALL;".format(table))
             return
 
         # Manually delete rows because sqlite does not support TRUNCATE and