process_tracker_python-23 Data store initialization thru CLI

✨ CLI can now initialize data store (or drop and recreate)

Found the issue to why drop_all and create_all were not working.  Needed
to include the schema for objects.

Working on #23

Author: OpenDataAlex

process_tracker/data_store.py

@@ -61,6 +61,23 @@ def __init__(self, config_location=None):
         self.data_store_port = data_store["data_store_port"]
         self.data_store_name = data_store["data_store_name"]
 
+    def delete_data_store(self):
+        """
+        Initializes data store deletion, including wiping of all data within.
+        :return:
+        """
+
+        self.logger.warn("ALERT - DATA STORE TO BE OVERWRITTEN - ALL DATA WILL BE LOST")
+
+        for table in reversed(Base.metadata.sorted_tables):
+            try:
+                self.logger.info("Table will be deleted: %s" % table)
+                table.drop(self.engine)
+            except Exception:
+                self.logger.error(
+                    "Table %s unable to be deleted.  Does it exist?" % table
+                )
+
     def get_or_create_item(self, model, create=True, **kwargs):
         """
         Testing if an entity instance exists or not.  If does, return entity key.  If not, create entity instance
@@ -112,16 +129,12 @@ def initialize_data_store(self, overwrite=False):
         self.logger.info("Attempting to initialize Process Tracker data store.")
 
         if overwrite:
-            self.logger.warn(
-                "ALERT - DATA STORE TO BE OVERWRITTEN - ALL DATA WILL BE LOST"
-            )
-
-            for table in reversed(Base.metadata.sorted_tables):
-                self.logger.info("Table will be deleted: %s" % table)
-                table.drop(self.engine)
+            self.delete_data_store()
 
         self.logger.info("Data store initialization beginning.  Creating data store.")
-        Base.metadata.create_all(self.engine)
+        for table in Base.metadata.sorted_tables:
+            self.logger.info("Table will be created: %s" % table)
+            table.create(self.engine)
 
         self.logger.info("Setting up application defaults.")
 

tests/test_cli.py

@@ -3,6 +3,7 @@
 import unittest
 
 from click.testing import CliRunner
+import sqlalchemy
 
 from process_tracker.cli import main
 from process_tracker.data_store import DataStore
@@ -32,20 +33,27 @@ def setUp(self):
         self.session = self.data_store.session
         self.runner = CliRunner()
 
+    def test_setup_delete(self):
+        """
+        Testing that data store is deleted if delete is triggered.
+        :return:
+        """
+        self.data_store.delete_data_store()
+        table_names = sqlalchemy.inspect(self.data_store.engine).get_table_names()
+        is_empty = table_names == []
+
+        self.assertTrue(True, is_empty)
+
     def test_setup_overwrite(self):
         """
         Testing that if data store is already set up and overwrite is set to True, wipe and recreate the data store.
         :return:
         """
-        self.data_store.get_or_create_item(model=Actor, actor_name="Testing Testor")
-        pre_delete = self.session.query(Actor).count()
-        time.sleep(1)
         self.runner.invoke(main, "setup -o True")
 
         instance = self.session.query(Actor).count()
 
         self.assertEqual(0, instance)
-        self.assertNotEqual(0, pre_delete)
 
     def test_setup_initialize(self):
         """