Literals

Pipfile

@@ -14,6 +14,7 @@ faker = "*"
 streamlit-player = "*"
 
 [dev-packages]
+mock-generators = {editable = true, path = "."}
 
 [requires]
 python_version = "3.11"

mock_generators/__init__.py

@@ -1,2 +1,3 @@
-
+import sys
+sys.path.append('.')
 __version__ = "0.1.0"

mock_generators/app.py

@@ -4,12 +4,13 @@
 from tabs.design_tab import design_tab
 from tabs.data_importer import data_importer_tab
 from tabs.tutorial import tutorial_tab
-from config import preload_state, load_generators
+from config import setup_logging, preload_state, load_generators_to_streamlit
 
 # SETUP
 st.set_page_config(layout="wide")
+setup_logging()
 preload_state()
-load_generators()
+load_generators_to_streamlit()
 
 # UI
 st.title("Mock Graph Data Generator")

mock_generators/config.py

@@ -2,7 +2,14 @@
 from constants import *
 from file_utils import load_json
 from models.generator import generators_from_json
+import logging
 
+def setup_logging():
+    logger = logging.getLogger(__name__)
+    FORMAT = "[%(filename)s:%(lineno)s - %(funcName)20s() ] %(message)s"
+    logging.basicConfig(format=FORMAT)
+    logger.setLevel(logging.DEBUG)
+
 def preload_state():
     if ZIPS_PATH not in st.session_state:
         st.session_state[ZIPS_PATH] = DEFAULT_ZIPS_PATH
@@ -18,7 +25,6 @@ def preload_state():
         st.session_state[IMPORTED_FILENAME] = ""
     if IMPORTS_PATH not in st.session_state:
         st.session_state[IMPORTS_PATH] = DEFAULT_IMPORTS_PATH
-    # TODO: Replace with reference to selected import file
     if IMPORTED_FILE not in st.session_state:
         st.session_state[IMPORTED_FILE] = None
     if IMPORTED_NODES not in st.session_state:
@@ -32,7 +38,19 @@ def preload_state():
     if MAPPINGS not in st.session_state:
         st.session_state[MAPPINGS] = None
 
-def load_generators():
+def load_generators(
+        folderpath = str
+):
+    try:
+        with open(folderpath) as input:
+            generators_json = load_json(folderpath)
+            new_generators = generators_from_json(generators_json)
+            return new_generators
+    except FileNotFoundError:
+        raise Exception('Generator JSONs not found.')
+
+
+def load_generators_to_streamlit():
     spec_filepath = st.session_state[SPEC_FILE]
     generators = st.session_state[GENERATORS]
     try:

mock_generators/constants.py

@@ -24,6 +24,4 @@
 CODE_TEMPLATE_FILE = "templates_file"
 DEFAULT_DATA_IMPORTER_FILENAME = "neo4j_importer_model"
 NEW_GENERATOR_ARGS = "new_generator_args"
-
-# TODO: Can Streamlit's st.session hold all the data we'll be generating?
 MAPPINGS = "mappings"