# HG changeset patch
# User Marcin Kuzminski <marcin@python-works.com>
# Date 2011-01-02 19:52:24
# Node ID de560c47dd03fbad60eeb1e3c0fc49d47e242575
# Parent  2f83756f3041faa65ebf7604856b51f5bbe083f3

Added missing FK to migration

diff --git a/rhodecode/lib/dbmigrate/versions/003_version_1_2_0.py b/rhodecode/lib/dbmigrate/versions/003_version_1_2_0.py
--- a/rhodecode/lib/dbmigrate/versions/003_version_1_2_0.py
+++ b/rhodecode/lib/dbmigrate/versions/003_version_1_2_0.py
@@ -6,7 +6,6 @@ from sqlalchemy.exc import DatabaseError
 from sqlalchemy.orm import relation, backref, class_mapper
 from sqlalchemy.orm.session import Session
 
-
 from rhodecode.lib.dbmigrate.migrate import *
 from rhodecode.lib.dbmigrate.migrate.changeset import *
 
@@ -35,20 +34,14 @@ def upgrade(migrate_engine):
     #==========================================================================
     # Upgrade of `repositories` table
     #==========================================================================    
-    tblname = 'repositories'
-    tbl = Table(tblname, MetaData(bind=migrate_engine), autoload=True,
-                    autoload_with=migrate_engine)
+    from rhodecode.model.db import Repository
 
     #ADD group_id column#
-    group_id = Column("group_id", Integer(), #ForeignKey('groups.group_id'),
+    group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'),
                   nullable=True, unique=False, default=None)
 
-    group_id.create(tbl)
+    group_id.create(Repository().__table__)
 
-# TODO: fix this somehow ?!
-#    cons = ForeignKeyConstraint([tbl.c.group_id], ['groups.group_id'], table=tbl)
-#
-#    cons.create()
     return