changeset 3086:f9c44f3ed4c6 beta

fixed migrations
author Marcin Kuzminski <marcin@python-works.com>
date Wed, 12 Dec 2012 23:26:30 +0100
parents 7d0476e1f1dc
children a797ada9d2ad
files rhodecode/lib/dbmigrate/versions/008_version_1_5_0.py
diffstat 1 files changed, 14 insertions(+), 14 deletions(-) [+]
line wrap: on
line diff
--- a/rhodecode/lib/dbmigrate/versions/008_version_1_5_0.py	Tue Dec 11 01:15:30 2012 +0100
+++ b/rhodecode/lib/dbmigrate/versions/008_version_1_5_0.py	Wed Dec 12 23:26:30 2012 +0100
@@ -32,27 +32,27 @@
     # 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):