fix: add schema in migration

Maze21127 · Mar 13, 2024 · 4f324e3 · 4f324e3
1 parent f615b5b
commit 4f324e3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...if %}/{% if use_sqlalchemy %}2024_01_07_2001-fe58b071a045_create_user_table.py{% endif %} b/...if %}/{% if use_sqlalchemy %}2024_01_07_2001-fe58b071a045_create_user_table.py{% endif %}
@@ -9,6 +9,7 @@ from typing import Sequence, Union
 
 from alembic import op
 import sqlalchemy as sa
+import os
 
 
 # revision identifiers, used by Alembic.
@@ -17,6 +18,7 @@ down_revision: Union[str, None] = None
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None
 
+DB_SCHEMA = os.environ.get("DB_SCHEMA")
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###