22 def upgrade(migrate_engine): 23 meta.bind = migrate_engine 24 dialect = migrate_engine.url.get_dialect().name 25 26 records = Table('records', meta, autoload=True) 27 domains = Table('domains', meta, autoload=True) 28 29 # add foreignkey if not sqlite 30 if not dialect.startswith('sqlite'): 31 ForeignKeyConstraint(columns=[records.c.domain_id], 32 refcolumns=[domains.c.id]).drop() 33 ForeignKeyConstraint(columns=[records.c.domain_id], 34 refcolumns=[domains.c.id], 35 ondelete='CASCADE').create() 36 37 38 def downgrade(migrate_engine): 39 meta.bind = migrate_engine 40 dialect = migrate_engine.url.get_dialect().name 41 42 records = Table('records', meta, autoload=True) 43 domains = Table('domains', meta, autoload=True) 44 45 # add foreignkey if not sqlite 46 if not dialect.startswith('sqlite'): 47 ForeignKeyConstraint(columns=[records.c.domain_id], 48 refcolumns=[domains.c.id]).drop() 49 ForeignKeyConstraint(columns=[records.c.domain_id], 50 refcolumns=[domains.c.id]).create() ~