kallithea Changeset - f9c44f3ed4c6

Changeset - f9c44f3ed4c6

Parent rev.

Child rev.

[Not reviewed]

beta

0 1 0

Marcin Kuzminski - 13 years ago 2012-12-12 23:26:30
marcin@python-works.com

fixed migrations

1 file changed with 14 insertions and 14 deletions:

rhodecode/lib/dbmigrate/versions/008_version_1_5_0.py

0 comments (0 inline, 0 general)

rhodecode/lib/dbmigrate/versions/008_version_1_5_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, joinedload
 from sqlalchemy.orm.session import Session
 from sqlalchemy.ext.declarative import declarative_base
 from rhodecode.lib.dbmigrate.migrate import *
 from rhodecode.lib.dbmigrate.migrate.changeset import *
 from rhodecode.model.meta import Base
 from rhodecode.model import meta
 log = logging.getLogger(__name__)
 def upgrade(migrate_engine):
     """
     Upgrade operations go here.
     Don't create your own engine; bind migrate_engine to your metadata
     """
     #==========================================================================
     # USER LOGS
     #==========================================================================
     from rhodecode.lib.dbmigrate.schema.db_1_5_0 import UserLog
     tbl = UserLog.__table__
     username = Column("username", String(255, convert_unicode=False,
                                          assert_unicode=None), nullable=True,
                       unique=None, default=None)
     # create username column
     username.create(table=tbl)
     #alter user_id to not null
     _Session = Session()
     ## after adding that column fix all usernames
     users_log = _Session.query(UserLog)\
             .options(joinedload(UserLog.user))\
             .options(joinedload(UserLog.repository)).all()
     for entry in users_log:
         entry.username = entry.user.username
         _Session.add(entry)
     _Session.commit()
     #alter username to not null
     from rhodecode.lib.dbmigrate.schema.db_1_5_0 import UserLog
     tbl_name = UserLog.__tablename__
     tbl = Table(tbl_name,
                 MetaData(bind=migrate_engine), autoload=True,
                 autoload_with=migrate_engine)
-    col = tbl.columns.user_id
+    col = tbl.columns.username
     # remove nullability from revision field
     col.alter(nullable=True)
     ## after adding that column fix all usernames
     users_log = UserLog.query()\
             .options(joinedload(UserLog.user))\
             .options(joinedload(UserLog.repository)).all()
     for entry in users_log:
         entry.username = entry.user.username
         Session().add(entry)
     Session().commit()
     col.alter(nullable=False)
 def downgrade(migrate_engine):
     meta = MetaData()
     meta.bind = migrate_engine

0 comments (0 inline, 0 general)