opensource/galaxy
3
0
Fork 0
mirror of https://github.com/galaxyproject/galaxy.git synced 2026-10-29 17:42:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set dburl in alembic config

Browse Source
This commit is contained in:
John Davis
2022-09-16 00:39:03 -04:00
parent 0a798ea536
commit e247afd8d7
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/galaxy/model/migrations/dbscript.py
+13 -6
View File
@@ -44,14 +44,18 @@ TSI_CONFIG_PREFIX = "GALAXY_INSTALL_CONFIG_"
class DbScript:
"""
Used to manage the gxy db.
The upgrade command is called on both: gxy and tsi. Reason: if this is the first alembic command on this branch,
the upgrade command will stamp the alembic_version table: we need that for both branches.
Facade for common database schema migration operations on the gxy branch.
When the gxy and tsi branches are persisted in the same database, some
alembic commands will display output on the state on both branches (e.g.
history, version, dbversion). The upgrade command is executed on both
branches: gxy and tsi (the upgrade command ensures the branch has been
initialized by stamping its version in the alembic_version table).
"""
def __init__(self, config_file: Optional[str] = None) -> None:
self.alembic_config = self._get_alembic_cfg()
self._set_dburl(config_file)
self.alembic_config.set_main_option("sqlalchemy.url", self.gxy_url)
def upgrade(self, args: argparse.Namespace) -> None:
def upgrade_to_revision(rev):
@@ -60,11 +64,14 @@ class DbScript:
if args.revision:
revision = self._parse_revision(args.revision)
upgrade_to_revision(revision)
else: # Run for each model
else:
self.alembic_config.set_main_option("sqlalchemy.url", self.gxy_url)
upgrade_to_revision("gxy@head")
self.alembic_config.set_main_option("sqlalchemy.url", self.tsi_url)
upgrade_to_revision("tsi@head")
try:
self.alembic_config.set_main_option("sqlalchemy.url", self.tsi_url)
upgrade_to_revision("tsi@head")
finally:
self.alembic_config.set_main_option("sqlalchemy.url", self.gxy_url)
def downgrade(self, args: argparse.Namespace) -> None:
revision = self._parse_revision(args.revision)