feat(#605): remove tables when removing history

hpi-dhc · May 5, 2023 · 3dac952 · 3dac952
1 parent 6ccaa77
commit 3dac952
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/scripts/common/remove_history.py b/scripts/common/remove_history.py
@@ -1,10 +1,11 @@
 def remove_history(data):
-    if 'AppData' in data:
-        data['AppData'] = []
+    tables_to_be_deleted = []
     for table_name in data.keys():
-        if table_name.endswith('_History'):
-            data[table_name] = []
+        if table_name.startswith('AppData') or table_name.endswith('_History'):
+            tables_to_be_deleted.append(table_name)
         else:
             for item in data[table_name]:
                 item['_v'] = 1
+    for table_name in tables_to_be_deleted:
+        del data[table_name]
     return data