<s> from migrate import ForeignKeyConstraint <EOL> from sqlalchemy import MetaData , Table <EOL> DB2_FKEYS = [ <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } , <EOL> { '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT>' : [ '<STR_LIT>' ] , <EOL> '<STR_LIT:name>' : '<STR_LIT>' } <EOL> ] <EOL> def _get_refcolumns ( metadata , refcolumns ) : <EOL> refcolumn_objects = [ ] <EOL> for refcol in refcolumns : <EOL> table , column = refcol . split ( '<STR_LIT:.>' ) <EOL> table = Table ( table , metadata , autoload = True ) <EOL> refcolumn_objects . append ( table . c [ column ] ) <EOL> return refcolumn_objects <EOL> def upgrade ( migrate_engine ) : <EOL> if migrate_engine . name == '<STR_LIT>' : <EOL> metadata = MetaData ( bind = migrate_engine ) <EOL> for values in DB2_FKEYS : <EOL> params = dict ( name = values [ '<STR_LIT:name>' ] ) <EOL> table = Table ( values [ '<STR_LIT>' ] , metadata , autoload = True ) <EOL> params [ '<STR_LIT>' ] = table <EOL> params [ '<STR_LIT>' ] = [ table . c [ col ] for col in values [ '<STR_LIT>' ] ] <EOL> params [ '<STR_LIT>' ] = _get_refcolumns ( metadata , <EOL> values [ '<STR_LIT>' ] ) <EOL> fkey = ForeignKeyConstraint ( ** params ) <EOL> fkey . create ( ) </s>
