Clean code

api/clean_api.py

@@ -1,9 +1,38 @@
-from .database import ChangeSet
-from .batch_api_cs import clean_scada_device_cs, clean_scada_device_data_cs, clean_scada_element_cs
+from .database import ChangeSet, read_all
 from .batch_exe import execute_batch_command
 
 # TODO: merge to batch_api
 
+def clean_scada_device_cs(name: str) -> ChangeSet:
+    cs = ChangeSet()
+
+    rows = read_all(name, 'select id from scada_device acs')
+    for row in rows:
+        cs.delete({ 'type': 'scada_device', 'id': row['id'] })
+
+    return cs
+
+
+def clean_scada_device_data_cs(name: str) -> ChangeSet:
+    cs = ChangeSet()
+
+    rows = read_all(name, 'select distinct device_id from scada_device_data acs')
+    for row in rows:
+        cs.update({ 'type': 'scada_device_data', 'device_id': row['device_id'], 'data': [] })
+
+    return cs
+
+
+def clean_scada_element_cs(name: str) -> ChangeSet:
+    cs = ChangeSet()
+
+    rows = read_all(name, 'select id from scada_element acs')
+    for row in rows:
+        cs.delete({ 'type': 'scada_element', 'id': row['id'] })
+
+    return cs
+
+
 def clean_scada_device(name: str) -> ChangeSet:
     return execute_batch_command(name, clean_scada_device_cs(name))