comparison rhodecode/lib/dbmigrate/versions/010_version_1_5_2.py @ 3148:b31984972e95 beta

Migration upgrades cache for lightweight dashboard Fixed some migration issues
author Marcin Kuzminski <marcin@python-works.com>
date Sat, 05 Jan 2013 02:20:35 +0100
parents 8182ebed2922
children ffd45b185016
comparison
equal deleted inserted replaced
3147:8182ebed2922 3148:b31984972e95
10 from rhodecode.lib.dbmigrate.migrate import * 10 from rhodecode.lib.dbmigrate.migrate import *
11 from rhodecode.lib.dbmigrate.migrate.changeset import * 11 from rhodecode.lib.dbmigrate.migrate.changeset import *
12 12
13 from rhodecode.model.meta import Base 13 from rhodecode.model.meta import Base
14 from rhodecode.model import meta 14 from rhodecode.model import meta
15 from rhodecode.lib.dbmigrate.versions import _reset_base
15 16
16 log = logging.getLogger(__name__) 17 log = logging.getLogger(__name__)
17 18
18 19
19 def upgrade(migrate_engine): 20 def upgrade(migrate_engine):
20 """ 21 """
21 Upgrade operations go here. 22 Upgrade operations go here.
22 Don't create your own engine; bind migrate_engine to your metadata 23 Don't create your own engine; bind migrate_engine to your metadata
23 """ 24 """
25 _reset_base(migrate_engine)
24 #========================================================================== 26 #==========================================================================
25 # USER LOGS 27 # USER LOGS
26 #========================================================================== 28 #==========================================================================
27 from rhodecode.lib.dbmigrate.schema.db_1_5_0 import UserIpMap 29 from rhodecode.lib.dbmigrate.schema.db_1_5_2 import UserIpMap
28 tbl = UserIpMap.__table__ 30 tbl = UserIpMap.__table__
29 tbl.create() 31 tbl.create()
30 32
31 #========================================================================== 33 #==========================================================================
32 # REPOSITORIES 34 # REPOSITORIES
33 #========================================================================== 35 #==========================================================================
34 from rhodecode.lib.dbmigrate.schema.db_1_5_0 import Repository 36 from rhodecode.lib.dbmigrate.schema.db_1_5_2 import Repository
35 tbl = Repository.__table__ 37 tbl = Repository.__table__
36 changeset_cache = Column("changeset_cache", LargeBinary(), nullable=True) 38 changeset_cache = Column("changeset_cache", LargeBinary(), nullable=True)
37 # create username column 39 # create username column
38 changeset_cache.create(table=tbl) 40 changeset_cache.create(table=tbl)
39 41
40 #fix cache data 42 #fix cache data
41 _Session = Session() 43 repositories = Repository.getAll()
42 ## after adding that column fix all usernames
43 repositories = _Session.query(Repository).all()
44 for entry in repositories: 44 for entry in repositories:
45 entry.update_changeset_cache() 45 entry.update_changeset_cache()
46 _Session.commit()
47 46
48 47
49 def downgrade(migrate_engine): 48 def downgrade(migrate_engine):
50 meta = MetaData() 49 meta = MetaData()
51 meta.bind = migrate_engine 50 meta.bind = migrate_engine