remove make_db_session_factory

Signed-off-by: Huong Nguyen <[email protected]>

package/kedro_viz/database.py

@@ -21,21 +21,3 @@ def configure_wal_for_azure(engine):
     if is_azure_ml:
         with engine.connect() as conn:
             conn.execute(text("PRAGMA journal_mode=WAL;"))
-
-
-def make_db_session_factory(session_store_location: str) -> sessionmaker:
-    """SQLAlchemy connection to a SQLite DB"""
-    database_url = f"sqlite:///{session_store_location}"
-    engine = create_engine(database_url, connect_args={"check_same_thread": False})
-    # TODO: making db session factory shouldn't depend on models.
-    # So want to move the table creation elsewhere ideally.
-    # But this means returning engine as well as session class.
-
-    # Check if we are running in an Azure ML environment if so enable WAL mode.
-    configure_wal_for_azure(engine)
-
-    # Create the database tables if they do not exist.
-    Base.metadata.create_all(bind=engine)
-
-    # Return a session factory bound to the engine.
-    return sessionmaker(bind=engine)

package/kedro_viz/integrations/kedro/data_loader.py

@@ -133,8 +133,7 @@ def load_data(
             configuration.
         is_lite: A flag to run Kedro-Viz in lite mode.
     Returns:
-        A tuple containing the data catalog, pipeline dictionary, session store
-        and dataset stats dictionary.
+        A tuple containing the data catalog, pipeline dictionary,and dataset stats dictionary.
     """
     if package_name:
         configure_project(package_name)

package/kedro_viz/server.py

@@ -10,7 +10,6 @@
 from kedro_viz.autoreload_file_filter import AutoreloadFileFilter
 from kedro_viz.constants import DEFAULT_HOST, DEFAULT_PORT
 from kedro_viz.data_access import DataAccessManager, data_access_manager
-from kedro_viz.database import make_db_session_factory
 from kedro_viz.integrations.kedro import data_loader as kedro_data_loader
 from kedro_viz.launchers.utils import _check_viz_up, _wait_for, display_cli_message
 

package/tests/test_api/test_graphql/conftest.py

@@ -7,59 +7,9 @@
 from kedro_datasets import matplotlib, pandas, plotly, tracking
 
 from kedro_viz.api.graphql.types import Run
-from kedro_viz.database import make_db_session_factory
 from kedro_viz.models.experiment_tracking import RunModel, UserRunDetailsModel
 
 
-@pytest.fixture
-def example_run_ids():
-    yield ["2021-11-03T18.24.24.379Z", "2021-11-02T18.24.24.379Z"]
-
-
-@pytest.fixture
-def example_db_session(tmp_path):
-    session_store_location = Path(tmp_path / "session_store.db")
-    session_class = make_db_session_factory(session_store_location)
-    yield session_class
-
-
-@pytest.fixture
-def example_db_session_with_runs(example_db_session, example_run_ids):
-    with example_db_session.begin() as session:
-        for run_id in example_run_ids:
-            session_data = {
-                "package_name": "testsql",
-                "project_path": "/Users/Projects/testsql",
-                "session_id": run_id,
-                "cli": {
-                    "args": [],
-                    "params": {
-                        "from_inputs": [],
-                        "to_outputs": [],
-                        "from_nodes": [],
-                        "to_nodes": [],
-                        "node_names": (),
-                        "runner": None,
-                        "parallel": False,
-                        "is_async": False,
-                        "env": None,
-                        "tag": (),
-                        "load_version": {},
-                        "pipeline": None,
-                        "config": None,
-                        "params": {},
-                    },
-                    "command_name": "run",
-                    "command_path": "kedro run",
-                },
-            }
-            run = RunModel(id=run_id, blob=json.dumps(session_data))
-            user_run_details = UserRunDetailsModel(run_id=run.id, bookmark=True)
-            session.add(run)
-            session.add(user_run_details)
-    yield example_db_session
-
-
 @pytest.fixture
 def data_access_manager_with_no_run(data_access_manager, example_db_session, mocker):
     data_access_manager.set_db_session(example_db_session)