Changeset - de560c47dd03
[Not reviewed]
beta
0 1 0
Marcin Kuzminski - 15 years ago 2011-01-02 20:52:24
marcin@python-works.com
Added missing FK to migration
1 file changed with 3 insertions and 10 deletions:
0 comments (0 inline, 0 general)
rhodecode/lib/dbmigrate/versions/003_version_1_2_0.py
Show inline comments
 
import logging
 
import datetime
 

	
 
from sqlalchemy import *
 
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 *
 

	
 
from rhodecode.model.meta import Base
 
from rhodecode.model.db import BaseModel
 

	
 
log = logging.getLogger(__name__)
 

	
 
def upgrade(migrate_engine):
 
    """ Upgrade operations go here. 
 
    Don't create your own engine; bind migrate_engine to your metadata
 
    """
 
@@ -26,34 +25,28 @@ def upgrade(migrate_engine):
 
    from rhodecode.model.db import Group
 
    Group().__table__.create()
 

	
 
    #==========================================================================
 
    # Add table `group_to_perm`
 
    #==========================================================================
 
    from rhodecode.model.db import GroupToPerm
 
    GroupToPerm().__table__.create()
 

	
 
    #==========================================================================
 
    # 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
 

	
 

	
 
def downgrade(migrate_engine):
 
    meta = MetaData()
 
    meta.bind = migrate_engine
 

	
 

	
0 comments (0 inline, 0 general)