| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- <%!
- import re
- %>"""${message}
- Revision ID: ${up_revision}
- Revises: ${down_revision | comma,n}
- Create Date: ${create_date}
- """
- from typing import Sequence, Union
- from alembic import op
- import sqlalchemy as sa
- ${imports if imports else ""}
- # revision identifiers, used by Alembic.
- revision: str = ${repr(up_revision)}
- 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)}
- def upgrade(engine_name: str) -> None:
- """Upgrade schema."""
- globals()["upgrade_%s" % engine_name]()
- def downgrade(engine_name: str) -> None:
- """Downgrade schema."""
- globals()["downgrade_%s" % engine_name]()
- <%
- db_names = config.get_main_option("databases")
- %>
- ## generate an "upgrade_<xyz>() / downgrade_<xyz>()" function
- ## for each database name in the ini file.
- % for db_name in re.split(r',\s*', db_names):
- def upgrade_${db_name}() -> None:
- """Upgrade ${db_name} schema."""
- ${context.get("%s_upgrades" % db_name, "pass")}
- def downgrade_${db_name}() -> None:
- """Downgrade ${db_name} schema."""
- ${context.get("%s_downgrades" % db_name, "pass")}
- % endfor
|
