Changeset - b9fe529d5752
[Not reviewed]
default
0 1 0
Søren Løvborg - 9 years ago 2016-07-05 16:57:01
sorenl@unity3d.com
db: enable Alembic autogeneration of migration scripts
1 file changed with 18 insertions and 0 deletions:
0 comments (0 inline, 0 general)
kallithea/alembic/env.py
Show inline comments
 
@@ -17,12 +17,14 @@
 
import logging
 
from logging.config import fileConfig
 

	
 
from alembic import context
 
from sqlalchemy import engine_from_config, pool
 

	
 
from kallithea.model import db
 

	
 

	
 
# The alembic.config.Config object, which wraps the current .ini file.
 
config = context.config
 

	
 
# Default to use the main Kallithea database string in [app:main].
 
# For advanced uses, this can be overridden by specifying an explicit
 
@@ -38,12 +40,22 @@ logging.getLogger('alembic').setLevel(lo
 

	
 
# Setup Python loggers based on the config file provided to the alembic
 
# command.
 
fileConfig(config.config_file_name, disable_existing_loggers=False)
 

	
 

	
 
def include_in_autogeneration(object, name, type, reflected, compare_to):
 
    """Filter changes subject to autogeneration of migrations. """
 

	
 
    # Don't include changes to sqlite_sequence.
 
    if type == 'table' and name == 'sqlite_sequence':
 
        return False
 

	
 
    return True
 

	
 

	
 
def run_migrations_offline():
 
    """Run migrations in 'offline' (--sql) mode.
 

	
 
    This produces an SQL script instead of directly applying the changes.
 
    Some migrations may not run in offline mode.
 
    """
 
@@ -69,12 +81,18 @@ def run_migrations_online():
 
        prefix='sqlalchemy.',
 
        poolclass=pool.NullPool)
 

	
 
    with connectable.connect() as connection:
 
        context.configure(
 
            connection=connection,
 

	
 
            # Support autogeneration of migration scripts based on "diff" between
 
            # current database schema and kallithea.model.db schema.
 
            target_metadata=db.Base.metadata,
 
            include_object=include_in_autogeneration,
 
            render_as_batch=True, # batch mode is needed for SQLite support
 
        )
 

	
 
        with context.begin_transaction():
 
            context.run_migrations()
 

	
 

	
0 comments (0 inline, 0 general)