parent
fce15c10a1
commit
cfe4cf076a
4 changed files with 166 additions and 0 deletions
@ -0,0 +1 @@ |
|||||||
|
Generic single-database configuration. |
@ -0,0 +1,45 @@ |
|||||||
|
# A generic, single database configuration. |
||||||
|
|
||||||
|
[alembic] |
||||||
|
# template used to generate migration files |
||||||
|
# file_template = %%(rev)s_%%(slug)s |
||||||
|
|
||||||
|
# set to 'true' to run the environment during |
||||||
|
# the 'revision' command, regardless of autogenerate |
||||||
|
# revision_environment = false |
||||||
|
|
||||||
|
|
||||||
|
# Logging configuration |
||||||
|
[loggers] |
||||||
|
keys = root,sqlalchemy,alembic |
||||||
|
|
||||||
|
[handlers] |
||||||
|
keys = console |
||||||
|
|
||||||
|
[formatters] |
||||||
|
keys = generic |
||||||
|
|
||||||
|
[logger_root] |
||||||
|
level = WARN |
||||||
|
handlers = console |
||||||
|
qualname = |
||||||
|
|
||||||
|
[logger_sqlalchemy] |
||||||
|
level = WARN |
||||||
|
handlers = |
||||||
|
qualname = sqlalchemy.engine |
||||||
|
|
||||||
|
[logger_alembic] |
||||||
|
level = INFO |
||||||
|
handlers = |
||||||
|
qualname = alembic |
||||||
|
|
||||||
|
[handler_console] |
||||||
|
class = StreamHandler |
||||||
|
args = (sys.stderr,) |
||||||
|
level = NOTSET |
||||||
|
formatter = generic |
||||||
|
|
||||||
|
[formatter_generic] |
||||||
|
format = %(levelname)-5.5s [%(name)s] %(message)s |
||||||
|
datefmt = %H:%M:%S |
@ -0,0 +1,96 @@ |
|||||||
|
from __future__ import with_statement |
||||||
|
|
||||||
|
import logging |
||||||
|
from logging.config import fileConfig |
||||||
|
|
||||||
|
from sqlalchemy import engine_from_config |
||||||
|
from sqlalchemy import pool |
||||||
|
|
||||||
|
from alembic import context |
||||||
|
|
||||||
|
# this is the Alembic Config object, which provides |
||||||
|
# access to the values within the .ini file in use. |
||||||
|
config = context.config |
||||||
|
|
||||||
|
# Interpret the config file for Python logging. |
||||||
|
# This line sets up loggers basically. |
||||||
|
fileConfig(config.config_file_name) |
||||||
|
logger = logging.getLogger('alembic.env') |
||||||
|
|
||||||
|
# add your model's MetaData object here |
||||||
|
# for 'autogenerate' support |
||||||
|
# from myapp import mymodel |
||||||
|
# target_metadata = mymodel.Base.metadata |
||||||
|
from flask import current_app |
||||||
|
config.set_main_option( |
||||||
|
'sqlalchemy.url', |
||||||
|
str(current_app.extensions['migrate'].db.engine.url).replace('%', '%%')) |
||||||
|
target_metadata = current_app.extensions['migrate'].db.metadata |
||||||
|
|
||||||
|
# other values from the config, defined by the needs of env.py, |
||||||
|
# can be acquired: |
||||||
|
# my_important_option = config.get_main_option("my_important_option") |
||||||
|
# ... etc. |
||||||
|
|
||||||
|
|
||||||
|
def run_migrations_offline(): |
||||||
|
"""Run migrations in 'offline' mode. |
||||||
|
|
||||||
|
This configures the context with just a URL |
||||||
|
and not an Engine, though an Engine is acceptable |
||||||
|
here as well. By skipping the Engine creation |
||||||
|
we don't even need a DBAPI to be available. |
||||||
|
|
||||||
|
Calls to context.execute() here emit the given string to the |
||||||
|
script output. |
||||||
|
|
||||||
|
""" |
||||||
|
url = config.get_main_option("sqlalchemy.url") |
||||||
|
context.configure( |
||||||
|
url=url, target_metadata=target_metadata, literal_binds=True |
||||||
|
) |
||||||
|
|
||||||
|
with context.begin_transaction(): |
||||||
|
context.run_migrations() |
||||||
|
|
||||||
|
|
||||||
|
def run_migrations_online(): |
||||||
|
"""Run migrations in 'online' mode. |
||||||
|
|
||||||
|
In this scenario we need to create an Engine |
||||||
|
and associate a connection with the context. |
||||||
|
|
||||||
|
""" |
||||||
|
|
||||||
|
# this callback is used to prevent an auto-migration from being generated |
||||||
|
# when there are no changes to the schema |
||||||
|
# reference: http://alembic.zzzcomputing.com/en/latest/cookbook.html |
||||||
|
def process_revision_directives(context, revision, directives): |
||||||
|
if getattr(config.cmd_opts, 'autogenerate', False): |
||||||
|
script = directives[0] |
||||||
|
if script.upgrade_ops.is_empty(): |
||||||
|
directives[:] = [] |
||||||
|
logger.info('No changes in schema detected.') |
||||||
|
|
||||||
|
connectable = engine_from_config( |
||||||
|
config.get_section(config.config_ini_section), |
||||||
|
prefix='sqlalchemy.', |
||||||
|
poolclass=pool.NullPool, |
||||||
|
) |
||||||
|
|
||||||
|
with connectable.connect() as connection: |
||||||
|
context.configure( |
||||||
|
connection=connection, |
||||||
|
target_metadata=target_metadata, |
||||||
|
process_revision_directives=process_revision_directives, |
||||||
|
**current_app.extensions['migrate'].configure_args |
||||||
|
) |
||||||
|
|
||||||
|
with context.begin_transaction(): |
||||||
|
context.run_migrations() |
||||||
|
|
||||||
|
|
||||||
|
if context.is_offline_mode(): |
||||||
|
run_migrations_offline() |
||||||
|
else: |
||||||
|
run_migrations_online() |
@ -0,0 +1,24 @@ |
|||||||
|
"""${message} |
||||||
|
|
||||||
|
Revision ID: ${up_revision} |
||||||
|
Revises: ${down_revision | comma,n} |
||||||
|
Create Date: ${create_date} |
||||||
|
|
||||||
|
""" |
||||||
|
from alembic import op |
||||||
|
import sqlalchemy as sa |
||||||
|
${imports if imports else ""} |
||||||
|
|
||||||
|
# revision identifiers, used by Alembic. |
||||||
|
revision = ${repr(up_revision)} |
||||||
|
down_revision = ${repr(down_revision)} |
||||||
|
branch_labels = ${repr(branch_labels)} |
||||||
|
depends_on = ${repr(depends_on)} |
||||||
|
|
||||||
|
|
||||||
|
def upgrade(): |
||||||
|
${upgrades if upgrades else "pass"} |
||||||
|
|
||||||
|
|
||||||
|
def downgrade(): |
||||||
|
${downgrades if downgrades else "pass"} |
Loading…
Reference in new issue