x/sqlalchemy-migrate
Code Issues Proposed changes

fixed bug in create column where foreign keys were being left out.

Browse Source
This commit is contained in:
percious17 2008-12-16 23:23:54 +00:00
parent a5abd977a3
commit 1fa26920d9
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
migrate/changeset/ansisql.py
View File

@ -59,6 +59,8 @@ class AlterTableVisitor(SchemaIterator,RawAlterTableVisitor):
"""Common operations for 'alter table' statements""" """Common operations for 'alter table' statements"""
from sqlalchemy.schema import ForeignKeyConstraint
class ANSIColumnGenerator(AlterTableVisitor,SchemaGenerator): class ANSIColumnGenerator(AlterTableVisitor,SchemaGenerator):
"""Extends ansisql generator for column creation (alter table add col)""" """Extends ansisql generator for column creation (alter table add col)"""
#def __init__(self, *args, **kwargs): #def __init__(self, *args, **kwargs):
@ -79,11 +81,13 @@ class ANSIColumnGenerator(AlterTableVisitor,SchemaGenerator):
pks = table.primary_key pks = table.primary_key
colspec = self.get_column_specification(column) colspec = self.get_column_specification(column)
self.append(colspec) self.append(colspec)
self.execute() if column.foreign_keys:
#if column.primary_key: for fk in column.foreign_keys:
# cons = self._pk_constraint(table,column,True) self.append(";\n\t ")
# cons.drop() self.add_foreignkey(fk.constraint)
# cons.create() else:
self.execute()
def visit_table(self,table): def visit_table(self,table):
pass pass