Fix scrub_user_tables

django-daiquiri · Dec 7, 2018 · c61cd82 · c61cd82
1 parent bf72ffd
commit c61cd82
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/daiquiri/query/management/commands/scrub_user_tables.py b/daiquiri/query/management/commands/scrub_user_tables.py
@@ -36,17 +36,17 @@ def handle(self, *args, **options):
                     table_name=table['name']
                 ).first()
 
-                if job and job.phase != QueryJob.PHASE_ARCHIVED:
-                    stale_tables.append((schema_name, table['name']))
+                if job and job.phase not in [QueryJob.PHASE_EXECUTING, QueryJob.PHASE_COMPLETED]:
+                    stale_tables.append((schema_name, table['name'], job.phase if job else None))
 
         if stale_tables:
             print('The following database tables have no associated QueryJob:')
 
             for stale_table in stale_tables:
-                print('* %s.%s' % stale_table)
+                print('%s.%s -> %s' % stale_table)
 
             if options['delete']:
-                for schema_name, table_name in stale_tables:
+                for schema_name, table_name, phase in stale_tables:
                     adapter.drop_table(schema_name, table_name)
 
                 print('The tables have been deleted.')