diff options
| -rw-r--r-- | fietsboek/updater/script.py | 12 | ||||
| -rw-r--r-- | fietsboek/updater/scripts/upd_20260930_7fa8kflikvnlvuqe.py | 4 |
2 files changed, 14 insertions, 2 deletions
diff --git a/fietsboek/updater/script.py b/fietsboek/updater/script.py index e54f435..0ac7874 100644 --- a/fietsboek/updater/script.py +++ b/fietsboek/updater/script.py @@ -2,6 +2,10 @@ # Placed in a separate file to avoid cyclic dependencies +from sqlalchemy import Engine + +from ..models import get_engine + class UpdateScript: """Base class for update scripts. @@ -10,6 +14,14 @@ class UpdateScript: methods for user interaction. """ + def dbengine(self, config: dict) -> Engine: + """Return a database engine. + + :param config: The app configuration. + :return: The correctly configured database engine. + """ + return get_engine(config) + def tell(self, text): """Output a message to the user. diff --git a/fietsboek/updater/scripts/upd_20260930_7fa8kflikvnlvuqe.py b/fietsboek/updater/scripts/upd_20260930_7fa8kflikvnlvuqe.py index 84c57a9..2ef6015 100644 --- a/fietsboek/updater/scripts/upd_20260930_7fa8kflikvnlvuqe.py +++ b/fietsboek/updater/scripts/upd_20260930_7fa8kflikvnlvuqe.py @@ -7,7 +7,7 @@ Date created: 2026-09-30 19:54:52.669074 import time import redis -from sqlalchemy import create_engine, select +from sqlalchemy import select from sqlalchemy.orm import sessionmaker from fietsboek import actions @@ -29,7 +29,7 @@ class Up(UpdateScript): def post_alembic(self, config): fietsconfig = mod_config.parse(config) - engine = create_engine(fietsconfig.sqlalchemy_url) + engine = self.dbengine(config) Session = sessionmaker(engine) redis_con = redis.from_url(fietsconfig.redis_url) |