_wipe_tables that works for PostgreSQL. For #31

django_heroku/core.py

@@ -25,18 +25,8 @@ def setup_databases(self, **kwargs):
 
     def _wipe_tables(self, connection):
         with connection.cursor() as cursor:
-            cursor.execute(
-                """
-                    DROP TABLE (
-                        SELECT
-                            table_name
-                        FROM
-                            information_schema.tables
-                        WHERE
-                            table_schema = 'public'
-                    ) CASCADE;
-                """
-            )
+            for table_name in connection.introspection.table_names():
+                cursor.execute(f"DROP TABLE IF EXISTS {table_name} CASCADE")
 
     def teardown_databases(self, old_config, **kwargs):
         self.keepdb = True