peewee migrator

migrate.py

# generic migration template

from playhouse.migrate import *
from models import *

migrator = SchemaMigrator.from_database(database)

def add_column(field):
    return migrator.add_column(field.model._meta.table_name, field.column_name, field)

migrations = (
    add_column(Foo.field),
    )

def main():
    with database.transaction():
        migrate(
            *migrations
            )

magic = 'yes please do the migration'
if __name__ == '__main__':
    import sys
    args = dict(enumerate(sys.argv))
    print('Connected to {}'.format(database.connect_params['host']))

    for m in migrations:
        if m.method in ('add_column',):
            table, column, field = m.args
            print('{:<15} {:<20} {:<20} {:<10}'.format(m.method, table, column, str(Context().sql(field.ddl(None)).query())))
        else:
            print('{:<10} {}'.format(m.method, m.args))

    if input('Enter "{}" to execute: '.format(magic)) != magic:
        sys.exit(1)

    main()