Files
@ 4bdd0bf1b1f4
Branch filter:
Location: kallithea/rhodecode/lib/dbmigrate/versions/002_version_1_1_0.py
4bdd0bf1b1f4
5.1 KiB
text/x-python
security bugfix: protected feeds, from unauthorized access, even without this, the feeds would crash and were unreadable, But proper way of securing it is with the secure decarators.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 | 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.model.meta import Base
from rhodecode.model.db import BaseModel
from rhodecode.lib.dbmigrate.migrate import *
log = logging.getLogger(__name__)
def upgrade(migrate_engine):
""" Upgrade operations go here.
Don't create your own engine; bind migrate_engine to your metadata
"""
#==========================================================================
# Upgrade of `users` table
#==========================================================================
tblname = 'users'
tbl = Table(tblname, MetaData(bind=migrate_engine), autoload=True,
autoload_with=migrate_engine)
#ADD is_ldap column
is_ldap = Column("is_ldap", Boolean(), nullable=True,
unique=None, default=False)
is_ldap.create(tbl, populate_default=True)
is_ldap.alter(nullable=False)
#==========================================================================
# Upgrade of `user_logs` table
#==========================================================================
tblname = 'users'
tbl = Table(tblname, MetaData(bind=migrate_engine), autoload=True,
autoload_with=migrate_engine)
#ADD revision column
revision = Column('revision', TEXT(length=None, convert_unicode=False,
assert_unicode=None),
nullable=True, unique=None, default=None)
revision.create(tbl)
#==========================================================================
# Upgrade of `repositories` table
#==========================================================================
tblname = 'repositories'
tbl = Table(tblname, MetaData(bind=migrate_engine), autoload=True,
autoload_with=migrate_engine)
#ADD repo_type column#
repo_type = Column("repo_type", String(length=None, convert_unicode=False,
assert_unicode=None),
nullable=True, unique=False, default='hg')
repo_type.create(tbl, populate_default=True)
#repo_type.alter(nullable=False)
#ADD statistics column#
enable_statistics = Column("statistics", Boolean(), nullable=True,
unique=None, default=True)
enable_statistics.create(tbl)
#==========================================================================
# Add table `user_followings`
#==========================================================================
tblname = 'user_followings'
class UserFollowing(Base, BaseModel):
__tablename__ = 'user_followings'
__table_args__ = (UniqueConstraint('user_id', 'follows_repository_id'),
UniqueConstraint('user_id', 'follows_user_id')
, {'useexisting':True})
user_following_id = Column("user_following_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
user_id = Column("user_id", Integer(), ForeignKey(u'users.user_id'), nullable=False, unique=None, default=None)
follows_repo_id = Column("follows_repository_id", Integer(), ForeignKey(u'repositories.repo_id'), nullable=True, unique=None, default=None)
follows_user_id = Column("follows_user_id", Integer(), ForeignKey(u'users.user_id'), nullable=True, unique=None, default=None)
user = relation('User', primaryjoin='User.user_id==UserFollowing.user_id')
follows_user = relation('User', primaryjoin='User.user_id==UserFollowing.follows_user_id')
follows_repository = relation('Repository')
Base.metadata.tables[tblname].create(migrate_engine)
#==========================================================================
# Add table `cache_invalidation`
#==========================================================================
tblname = 'cache_invalidation'
class CacheInvalidation(Base, BaseModel):
__tablename__ = 'cache_invalidation'
__table_args__ = (UniqueConstraint('cache_key'), {'useexisting':True})
cache_id = Column("cache_id", Integer(), nullable=False, unique=True, default=None, primary_key=True)
cache_key = Column("cache_key", String(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)
cache_args = Column("cache_args", String(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)
cache_active = Column("cache_active", Boolean(), nullable=True, unique=None, default=False)
def __init__(self, cache_key, cache_args=''):
self.cache_key = cache_key
self.cache_args = cache_args
self.cache_active = False
def __repr__(self):
return "<CacheInvalidation('%s:%s')>" % (self.cache_id, self.cache_key)
Base.metadata.tables[tblname].create(migrate_engine)
return
def downgrade(migrate_engine):
meta = MetaData()
meta.bind = migrate_engine
|