fix: add db schema to migration

Maze21127 · Feb 29, 2024 · 7a87ca8 · 7a87ca8
1 parent 056f40b
commit 7a87ca8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...if use_sqlalchemy %}migration{% endif %}/{% if use_sqlalchemy %}script.py.mako{% endif %} b/...if use_sqlalchemy %}migration{% endif %}/{% if use_sqlalchemy %}script.py.mako{% endif %}
@@ -9,6 +9,7 @@ from typing import Sequence, Union
 
 from alembic import op
 import sqlalchemy as sa
+import os
 ${imports if imports else ""}
 
 # revision identifiers, used by Alembic.
@@ -17,6 +18,7 @@ down_revision: Union[str, None] = ${repr(down_revision)}
 branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
 depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
 
+DB_SCHEMA = os.environ.get("DB_SCHEMA")
 
 def upgrade() -> None:
     ${upgrades if upgrades else "pass"}