script.py.mako 1011 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. <%!
  2. import re
  3. %>"""${message}
  4. Revision ID: ${up_revision}
  5. Revises: ${down_revision | comma,n}
  6. Create Date: ${create_date}
  7. """
  8. # revision identifiers, used by Alembic.
  9. revision = ${repr(up_revision)}
  10. down_revision = ${repr(down_revision)}
  11. branch_labels = ${repr(branch_labels)}
  12. depends_on = ${repr(depends_on)}
  13. from alembic import op
  14. import sqlalchemy as sa
  15. ${imports if imports else ""}
  16. def upgrade(engine_name):
  17. globals()["upgrade_%s" % engine_name]()
  18. def downgrade(engine_name):
  19. globals()["downgrade_%s" % engine_name]()
  20. <%
  21. from flask import current_app
  22. db_names = [''] + list(current_app.config.get("SQLALCHEMY_BINDS").keys())
  23. %>
  24. ## generate an "upgrade_<xyz>() / downgrade_<xyz>()" function
  25. ## for each database name in the ini file.
  26. % for db_name in db_names:
  27. def upgrade_${db_name}():
  28. ${context.get("%s_upgrades" % db_name, "pass")}
  29. def downgrade_${db_name}():
  30. ${context.get("%s_downgrades" % db_name, "pass")}
  31. % endfor