From 5ecec7c6800d62cc60f98e574cf161d3855c7243 Mon Sep 17 00:00:00 2001
From: "jan.dittberner" <unknown>
Date: Wed, 18 Feb 2009 20:08:37 +0000
Subject: mark ALTER TABLE ADD FOREIGN KEY as unsupported by SQLite update
 corresponding test case

---
 migrate/changeset/databases/sqlite.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'migrate/changeset/databases/sqlite.py')

diff --git a/migrate/changeset/databases/sqlite.py b/migrate/changeset/databases/sqlite.py
index e1a3bee..3748659 100644
--- a/migrate/changeset/databases/sqlite.py
+++ b/migrate/changeset/databases/sqlite.py
@@ -77,6 +77,17 @@ class SQLiteConstraintGenerator(ansisql.ANSIConstraintGenerator):
         self.execute()
 
 
+class SQLiteFKGenerator(SQLiteSchemaChanger, ansisql.ANSIFKGenerator):
+    def visit_column(self, column):
+        """Create foreign keys for a column (table already exists); #32"""
+
+        if self.fk:
+            self._not_supported("ALTER TABLE ADD FOREIGN KEY")
+
+        if self.buffer.getvalue() !='':
+            self.execute()
+
+
 class SQLiteConstraintDropper(ansisql.ANSIColumnDropper):
 
     def visit_migrate_primary_key_constraint(self, constraint):
@@ -93,3 +104,4 @@ class SQLiteDialect(ansisql.ANSIDialect):
     schemachanger = SQLiteSchemaChanger
     constraintgenerator = SQLiteConstraintGenerator
     constraintdropper = SQLiteConstraintDropper
+    columnfkgenerator = SQLiteFKGenerator
-- 
cgit v1.2.1