From 993943c8eea6f69f3d8173d3bf2469d6f1ed5d1a Mon Sep 17 00:00:00 2001 From: Jun-te Kim Date: Tue, 1 Apr 2025 12:21:53 +0000 Subject: [PATCH] load db first --- .github/workflows/db_migration.yml | 3 +-- etl/load_metadata.py | 7 +++++++ 2 files changed, 8 insertions(+), 2 deletions(-) create mode 100644 etl/load_metadata.py diff --git a/.github/workflows/db_migration.yml b/.github/workflows/db_migration.yml index b3c4137..bc42132 100644 --- a/.github/workflows/db_migration.yml +++ b/.github/workflows/db_migration.yml @@ -21,8 +21,7 @@ jobs: poetry install --no-root - name: run script run: | - ls -la - ls -la alembic/ + poetry run python etl/load_metadata.py poetry run alembic upgrade head env: PYTHONPATH: ${{ github.workspace }} diff --git a/etl/load_metadata.py b/etl/load_metadata.py new file mode 100644 index 0000000..a155bcc --- /dev/null +++ b/etl/load_metadata.py @@ -0,0 +1,7 @@ +# A very simple script to load metadata to whatever db we connected to +from etl.db.db import get_db_session, init_db + +init_db() + +with get_db_session() as session: + print("Metadata loaded to db") \ No newline at end of file