diff options
| author | Nicolas VINOT <aeris@imirhil.fr> | 2011-10-23 01:16:55 +0200 |
|---|---|---|
| committer | Nicolas VINOT <aeris@imirhil.fr> | 2011-10-23 01:16:55 +0200 |
| commit | 52272d32c5ba072a2488802a4404d54ce81471f4 (patch) | |
| tree | 90861f375e2296aa3bcf041c66dd581f2f709a9a | |
| parent | 3e83af842f15de11dbd052182dc2ead6aba0a602 (diff) | |
Correct code style
--HG--
branch : beta
| -rw-r--r-- | rhodecode/controllers/api/api.py | 202 | ||||
| -rwxr-xr-x | rhodecode/model/db.py | 842 | ||||
| -rw-r--r-- | rhodecode/model/repo_permission.py | 30 | ||||
| -rw-r--r-- | rhodecode/model/repos_group.py | 100 | ||||
| -rw-r--r-- | rhodecode/model/user.py | 18 | ||||
| -rw-r--r-- | rhodecode/model/users_group.py | 40 |
6 files changed, 616 insertions, 616 deletions
diff --git a/rhodecode/controllers/api/api.py b/rhodecode/controllers/api/api.py index 97bdaf65..82319427 100644 --- a/rhodecode/controllers/api/api.py +++ b/rhodecode/controllers/api/api.py @@ -14,10 +14,10 @@ from rhodecode.model import users_group from rhodecode.model.repos_group import ReposGroupModel from sqlalchemy.orm.exc import NoResultFound -log = logging.getLogger( __name__ ) +log = logging.getLogger(__name__) -class ApiController( JSONRPCController ): +class ApiController(JSONRPCController): """ API Controller @@ -33,8 +33,8 @@ class ApiController( JSONRPCController ): """ - @HasPermissionAllDecorator( 'hg.admin' ) - def pull( self, apiuser, repo ): + @HasPermissionAllDecorator('hg.admin') + def pull(self, apiuser, repo): """ Dispatch pull action on given repo @@ -43,17 +43,17 @@ class ApiController( JSONRPCController ): :param repo: """ - if Repository.is_valid( repo ) is False: - raise JSONRPCError( 'Unknown repo "%s"' % repo ) + if Repository.is_valid(repo) is False: + raise JSONRPCError('Unknown repo "%s"' % repo) try: - ScmModel().pull_changes( repo, self.rhodecode_user.username ) + ScmModel().pull_changes(repo, self.rhodecode_user.username) return 'Pulled from %s' % repo except Exception: - raise JSONRPCError( 'Unable to pull changes from "%s"' % repo ) + raise JSONRPCError('Unable to pull changes from "%s"' % repo) - @HasPermissionAllDecorator( 'hg.admin' ) - def get_user( self, apiuser, username ): + @HasPermissionAllDecorator('hg.admin') + def get_user(self, apiuser, username): """" Get a user by username @@ -61,21 +61,21 @@ class ApiController( JSONRPCController ): :param username """ - user = User.get_by_username( username ) + user = User.get_by_username(username) if not user: return None - return dict( id = user.user_id, + return dict(id = user.user_id, username = user.username, firstname = user.name, lastname = user.lastname, email = user.email, active = user.active, admin = user.admin, - ldap = user.ldap_dn ) + ldap = user.ldap_dn) - @HasPermissionAllDecorator( 'hg.admin' ) - def get_users( self, apiuser ): + @HasPermissionAllDecorator('hg.admin') + def get_users(self, apiuser): """" Get all users @@ -84,19 +84,19 @@ class ApiController( JSONRPCController ): result = [] for user in User.getAll(): - result.append( dict( id = user.user_id, + result.append(dict(id = user.user_id, username = user.username, firstname = user.name, lastname = user.lastname, email = user.email, active = user.active, admin = user.admin, - ldap = user.ldap_dn ) ) + ldap = user.ldap_dn)) return result - @HasPermissionAllDecorator( 'hg.admin' ) - def create_user( self, apiuser, username, password, firstname, - lastname, email, active = True, admin = False, ldap_dn = None ): + @HasPermissionAllDecorator('hg.admin') + def create_user(self, apiuser, username, password, firstname, + lastname, email, active = True, admin = False, ldap_dn = None): """ Create new user @@ -111,26 +111,26 @@ class ApiController( JSONRPCController ): :param ldap_dn: """ - if self.get_user( apiuser, username ): - raise JSONRPCError( "user %s already exist" % username ) + if self.get_user(apiuser, username): + raise JSONRPCError("user %s already exist" % username) try: - form_data = dict( username = username, + form_data = dict(username = username, password = password, active = active, admin = admin, name = firstname, lastname = lastname, email = email, - ldap_dn = ldap_dn ) - UserModel().create_ldap( username, password, ldap_dn, form_data ) - return dict( msg = 'created new user %s' % username ) + ldap_dn = ldap_dn) + UserModel().create_ldap(username, password, ldap_dn, form_data) + return dict(msg = 'created new user %s' % username) except Exception: - log.error( traceback.format_exc() ) - raise JSONRPCError( 'failed to create user %s' % username ) + log.error(traceback.format_exc()) + raise JSONRPCError('failed to create user %s' % username) - @HasPermissionAllDecorator( 'hg.admin' ) - def get_users_group( self, apiuser, group_name ): + @HasPermissionAllDecorator('hg.admin') + def get_users_group(self, apiuser, group_name): """" Get users group by name @@ -138,29 +138,29 @@ class ApiController( JSONRPCController ): :param group_name """ - users_group = UsersGroup.get_by_group_name( group_name ) + users_group = UsersGroup.get_by_group_name(group_name) if not users_group: return None members = [] for user in users_group.members: user = user.user - members.append( dict( id = user.user_id, + members.append(dict(id = user.user_id, username = user.username, firstname = user.name, lastname = user.lastname, email = user.email, active = user.active, admin = user.admin, - ldap = user.ldap_dn ) ) + ldap = user.ldap_dn)) - return dict( id = users_group.users_group_id, + return dict(id = users_group.users_group_id, name = users_group.users_group_name, active = users_group.users_group_active, - members = members ) + members = members) - @HasPermissionAllDecorator( 'hg.admin' ) - def get_users_groups( self, apiuser ): + @HasPermissionAllDecorator('hg.admin') + def get_users_groups(self, apiuser): """" Get all users groups @@ -172,23 +172,23 @@ class ApiController( JSONRPCController ): members = [] for user in users_group.members: user = user.user - members.append( dict( id = user.user_id, + members.append(dict(id = user.user_id, username = user.username, firstname = user.name, lastname = user.lastname, email = user.email, active = user.active, admin = user.admin, - ldap = user.ldap_dn ) ) + ldap = user.ldap_dn)) - result.append( dict( id = users_group.users_group_id, + result.append(dict(id = users_group.users_group_id, name = users_group.users_group_name, active = users_group.users_group_active, - members = members ) ) + members = members)) return result - @HasPermissionAllDecorator( 'hg.admin' ) - def create_users_group( self, apiuser, name, active = True ): + @HasPermissionAllDecorator('hg.admin') + def create_users_group(self, apiuser, name, active = True): """ Creates an new usergroup @@ -196,21 +196,21 @@ class ApiController( JSONRPCController ): :param active: """ - if self.get_users_group( apiuser, name ): - raise JSONRPCError( "users group %s already exist" % name ) + if self.get_users_group(apiuser, name): + raise JSONRPCError("users group %s already exist" % name) try: - form_data = dict( users_group_name = name, - users_group_active = active ) - ug = UsersGroup.create( form_data ) - return dict( id = ug.users_group_id, - msg = 'created new users group %s' % name ) + form_data = dict(users_group_name = name, + users_group_active = active) + ug = UsersGroup.create(form_data) + return dict(id = ug.users_group_id, + msg = 'created new users group %s' % name) except Exception: - log.error( traceback.format_exc() ) - raise JSONRPCError( 'failed to create group %s' % name ) + log.error(traceback.format_exc()) + raise JSONRPCError('failed to create group %s' % name) - @HasPermissionAllDecorator( 'hg.admin' ) - def add_user_to_users_group( self, apiuser, group_name, user_name ): + @HasPermissionAllDecorator('hg.admin') + def add_user_to_users_group(self, apiuser, group_name, user_name): """" Add a user to a group @@ -220,25 +220,25 @@ class ApiController( JSONRPCController ): """ try: - users_group = UsersGroup.get_by_group_name( group_name ) + users_group = UsersGroup.get_by_group_name(group_name) if not users_group: - raise JSONRPCError( 'unknown users group %s' % group_name ) + raise JSONRPCError('unknown users group %s' % group_name) try: - user = User.get_by_username( user_name ) + user = User.get_by_username(user_name) except NoResultFound: - raise JSONRPCError( 'unknown user %s' % user_name ) + raise JSONRPCError('unknown user %s' % user_name) - ugm = UsersGroupModel().add_user_to_group( users_group, user ) + ugm = UsersGroupModel().add_user_to_group(users_group, user) - return dict( id = ugm.users_group_member_id, - msg = 'created new users group member' ) + return dict(id = ugm.users_group_member_id, + msg = 'created new users group member') except Exception: - log.error( traceback.format_exc() ) - raise JSONRPCError( 'failed to create users group member' ) + log.error(traceback.format_exc()) + raise JSONRPCError('failed to create users group member') - @HasPermissionAnyDecorator( 'hg.admin' ) - def get_repo( self, apiuser, repo_name ): + @HasPermissionAnyDecorator('hg.admin') + def get_repo(self, apiuser, repo_name): """" Get repository by name @@ -247,7 +247,7 @@ class ApiController( JSONRPCController ): """ try: - repo = Repository.get_by_repo_name( repo_name ) + repo = Repository.get_by_repo_name(repo_name) except NoResultFound: return None @@ -255,7 +255,7 @@ class ApiController( JSONRPCController ): for user in repo.repo_to_perm: perm = user.permission.permission_name user = user.user - members.append( dict( type_ = "user", + members.append(dict(type_ = "user", id = user.user_id, username = user.username, firstname = user.name, @@ -264,24 +264,24 @@ class ApiController( JSONRPCController ): active = user.active, admin = user.admin, ldap = user.ldap_dn, - permission = perm ) ) + permission = perm)) for users_group in repo.users_group_to_perm: perm = users_group.permission.permission_name users_group = users_group.users_group - members.append( dict( type_ = "users_group", + members.append(dict(type_ = "users_group", id = users_group.users_group_id, name = users_group.users_group_name, active = users_group.users_group_active, - permission = perm ) ) + permission = perm)) - return dict( id = repo.repo_id, + return dict(id = repo.repo_id, name = repo.repo_name, type = repo.repo_type, description = repo.description, - members = members ) + members = members) - @HasPermissionAnyDecorator( 'hg.admin' ) - def get_repos( self, apiuser ): + @HasPermissionAnyDecorator('hg.admin') + def get_repos(self, apiuser): """" Get all repositories @@ -290,15 +290,15 @@ class ApiController( JSONRPCController ): result = [] for repository in Repository.getAll(): - result.append( dict( id = repository.repo_id, + result.append(dict(id = repository.repo_id, name = repository.repo_name, type = repository.repo_type, - description = repository.description ) ) + description = repository.description)) return result - @HasPermissionAnyDecorator( 'hg.admin', 'hg.create.repository' ) - def create_repo( self, apiuser, name, owner_name, description = '', repo_type = 'hg', \ - private = False ): + @HasPermissionAnyDecorator('hg.admin', 'hg.create.repository') + def create_repo(self, apiuser, name, owner_name, description = '', repo_type = 'hg', \ + private = False): """ Create a repository @@ -312,38 +312,38 @@ class ApiController( JSONRPCController ): try: try: - owner = User.get_by_username( owner_name ) + owner = User.get_by_username(owner_name) except NoResultFound: - raise JSONRPCError( 'unknown user %s' % owner ) + raise JSONRPCError('unknown user %s' % owner) - if self.get_repo( apiuser, name ): - raise JSONRPCError( "repo %s already exist" % name ) + if self.get_repo(apiuser, name): + raise JSONRPCError("repo %s already exist" % name) - groups = name.split( '/' ) + groups = name.split('/') real_name = groups[-1] groups = groups[:-1] parent_id = None for g in groups: - group = Group.get_by_group_name( g ) + group = Group.get_by_group_name(g) if not group: - group = ReposGroupModel().create( dict( group_name = g, + group = ReposGroupModel().create(dict(group_name = g, group_description = '', - group_parent_id = parent_id ) ) + group_parent_id = parent_id)) parent_id = group.group_id - RepoModel().create( dict( repo_name = real_name, + RepoModel().create(dict(repo_name = real_name, repo_name_full = name, description = description, private = private, repo_type = repo_type, repo_group = parent_id, - clone_uri = None ), owner ) + clone_uri = None), owner) except Exception: - log.error( traceback.format_exc() ) - raise JSONRPCError( 'failed to create repository %s' % name ) + log.error(traceback.format_exc()) + raise JSONRPCError('failed to create repository %s' % name) - @HasPermissionAnyDecorator( 'hg.admin' ) - def add_user_to_repo( self, apiuser, repo_name, user_name, perm ): + @HasPermissionAnyDecorator('hg.admin') + def add_user_to_repo(self, apiuser, repo_name, user_name, perm): """ Add permission for a user to a repository @@ -355,18 +355,18 @@ class ApiController( JSONRPCController ): try: try: - repo = Repository.get_by_repo_name( repo_name ) + repo = Repository.get_by_repo_name(repo_name) except NoResultFound: - raise JSONRPCError( 'unknown repository %s' % repo ) + raise JSONRPCError('unknown repository %s' % repo) try: - user = User.get_by_username( user_name ) + user = User.get_by_username(user_name) except NoResultFound: - raise JSONRPCError( 'unknown user %s' % user ) + raise JSONRPCError('unknown user %s' % user) - RepositoryPermissionModel().updateOrDeleteUserPermission( repo, user, perm ) + RepositoryPermissionModel().update_or_delete_user_permission(repo, user, perm) except Exception: - log.error( traceback.format_exc() ) - raise JSONRPCError( 'failed to edit permission %(repo)s for %(user)s' - % dict( user = user_name, repo = repo_name ) ) + log.error(traceback.format_exc()) + raise JSONRPCError('failed to edit permission %(repo)s for %(user)s' + % dict(user = user_name, repo = repo_name)) diff --git a/rhodecode/model/db.py b/rhodecode/model/db.py index f4c1b639..f240370e 100755 --- a/rhodecode/model/db.py +++ b/rhodecode/model/db.py @@ -51,13 +51,13 @@ from rhodecode.model.meta import Base, Session from rhodecode.model.caching_query import FromCache -log = logging.getLogger( __name__ ) +log = logging.getLogger(__name__) #============================================================================== # BASE CLASSES #============================================================================== -class ModelSerializer( json.JSONEncoder ): +class ModelSerializer(json.JSONEncoder): """ Simple Serializer for JSON, @@ -80,120 +80,120 @@ class ModelSerializer( json.JSONEncoder ): """ - def default( self, obj ): + def default(self, obj): - if hasattr( obj, '__json__' ): + if hasattr(obj, '__json__'): return obj.__json__() else: - return json.JSONEncoder.default( self, obj ) + return json.JSONEncoder.default(self, obj) -class BaseModel( object ): +class BaseModel(object): """Base Model for all classess """ @classmethod - def _get_keys( cls ): + def _get_keys(cls): """return column names for this model """ - return class_mapper( cls ).c.keys() + return class_mapper(cls).c.keys() - def get_dict( self ): + def get_dict(self): """return dict with keys and values corresponding to this model data """ d = {} for k in self._get_keys(): - d[k] = getattr( self, k ) + d[k] = getattr(self, k) return d - def get_appstruct( self ): + def get_appstruct(self): """return list with keys and values tupples corresponding to this model data """ l = [] for k in self._get_keys(): - l.append( ( k, getattr( self, k ), ) ) + l.append((k, getattr(self, k),)) return l - def populate_obj( self, populate_dict ): + def populate_obj(self, populate_dict): """populate model with data from given populate_dict""" for k in self._get_keys(): if k in populate_dict: - setattr( self, k, populate_dict[k] ) + setattr(self, k, populate_dict[k]) @classmethod - def query( cls ): - return Session.query( cls ) + def query(cls): + return Session.query(cls) @classmethod - def get( cls, id_ ): - return cls.query().get( id_ ) + def get(cls, id_): + return cls.query().get(id_) @classmethod - def getAll( cls ): + def getAll(cls): return cls.query().all() @classmethod - def delete( cls, id_ ): - obj = cls.query().get( id_ ) - Session.delete( obj ) + def delete(cls, id_): + obj = cls.query().get(id_) + Session.delete(obj) Session.commit() -class RhodeCodeSettings( Base, BaseModel ): +class RhodeCodeSettings(Base, BaseModel): __tablename__ = 'rhodecode_settings' - __table_args__ = ( UniqueConstraint( 'app_settings_name' ), {'extend_existing':True} ) - app_settings_id = Column( "app_settings_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - app_settings_name = Column( "app_settings_name", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - _app_settings_value = Column( "app_settings_value", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) + __table_args__ = (UniqueConstraint('app_settings_name'), {'extend_existing':True}) + app_settings_id = Column("app_settings_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + app_settings_name = Column("app_settings_name", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + _app_settings_value = Column("app_settings_value", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) - def __init__( self, k = '', v = '' ): + def __init__(self, k = '', v = ''): self.app_settings_name = k self.app_settings_value = v - @validates( '_app_settings_value' ) - def validate_settings_value( self, key, val ): - assert type( val ) == unicode + @validates('_app_settings_value') + def validate_settings_value(self, key, val): + assert type(val) == unicode return val @hybrid_property - def app_settings_value( self ): + def app_settings_value(self): v = self._app_settings_value if v == 'ldap_active': - v = str2bool( v ) + v = str2bool(v) return v @app_settings_value.setter - def app_settings_value( self, val ): + def app_settings_value(self, val): """ Setter that will always make sure we use unicode in app_settings_value :param val: """ - self._app_settings_value = safe_unicode( val ) + self._app_settings_value = safe_unicode(val) - def __repr__( self ): - return "<%s('%s:%s')>" % ( self.__class__.__name__, - self.app_settings_name, self.app_settings_value ) + def __repr__(self): + return "<%s('%s:%s')>" % (self.__class__.__name__, + self.app_settings_name, self.app_settings_value) @classmethod - def get_by_name( cls, ldap_key ): + def get_by_name(cls, ldap_key): return cls.query()\ - .filter( cls.app_settings_name == ldap_key ).scalar() + .filter(cls.app_settings_name == ldap_key).scalar() @classmethod - def get_app_settings( cls, cache = False ): + def get_app_settings(cls, cache = False): ret = cls.query() if cache: - ret = ret.options( FromCache( "sql_cache_short", "get_hg_settings" ) ) + ret = ret.options(FromCache("sql_cache_short", "get_hg_settings")) if not ret: - raise Exception( 'Could not get application settings !' ) + raise Exception('Could not get application settings !') settings = {} for each in ret: settings['rhodecode_' + each.app_settings_name] = \ @@ -202,222 +202,222 @@ class RhodeCodeSettings( Base, BaseModel ): return settings @classmethod - def get_ldap_settings( cls, cache = False ): + def get_ldap_settings(cls, cache = False): ret = cls.query()\ - .filter( cls.app_settings_name.startswith( 'ldap_' ) ).all() + .filter(cls.app_settings_name.startswith('ldap_')).all() fd = {} for row in ret: - fd.update( {row.app_settings_name:row.app_settings_value} ) + fd.update({row.app_settings_name:row.app_settings_value}) return fd -class RhodeCodeUi( Base, BaseModel ): +class RhodeCodeUi(Base, BaseModel): __tablename__ = 'rhodecode_ui' - __table_args__ = ( UniqueConstraint( 'ui_key' ), {'extend_existing':True} ) + __table_args__ = (UniqueConstraint('ui_key'), {'extend_existing':True}) HOOK_UPDATE = 'changegroup.update' HOOK_REPO_SIZE = 'changegroup.repo_size' HOOK_PUSH = 'pretxnchangegroup.push_logger' HOOK_PULL = 'preoutgoing.pull_logger' - ui_id = Column( "ui_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - ui_section = Column( "ui_section", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - ui_key = Column( "ui_key", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - ui_value = Column( "ui_value", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - ui_active = Column( "ui_active", Boolean(), nullable = True, unique = None, default = True ) + ui_id = Column("ui_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + ui_section = Column("ui_section", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + ui_key = Column("ui_key", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + ui_value = Column("ui_value", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + ui_active = Column("ui_active", Boolean(), nullable = True, unique = None, default = True) @classmethod - def get_by_key( cls, key ): - return cls.query().filter( cls.ui_key == key ) + def get_by_key(cls, key): + return cls.query().filter(cls.ui_key == key) @classmethod - def get_builtin_hooks( cls ): + def get_builtin_hooks(cls): q = cls.query() - q = q.filter( cls.ui_key.in_( [cls.HOOK_UPDATE, + q = q.filter(cls.ui_key.in_([cls.HOOK_UPDATE, cls.HOOK_REPO_SIZE, - cls.HOOK_PUSH, cls.HOOK_PULL] ) ) + cls.HOOK_PUSH, cls.HOOK_PULL])) return q.all() @classmethod - def get_custom_hooks( cls ): + def get_custom_hooks(cls): q = cls.query() - q = q.filter( ~cls.ui_key.in_( [cls.HOOK_UPDATE, + q = q.filter(~cls.ui_key.in_([cls.HOOK_UPDATE, cls.HOOK_REPO_SIZE, - cls.HOOK_PUSH, cls.HOOK_PULL] ) ) - q = q.filter( cls.ui_section == 'hooks' ) + cls.HOOK_PUSH, cls.HOOK_PULL])) + q = q.filter(cls.ui_section == 'hooks') return q.all() @classmethod - def create_or_update_hook( cls, key, val ): - new_ui = cls.get_by_key( key ).scalar() or cls() + def create_or_update_hook(cls, key, val): + new_ui = cls.get_by_key(key).scalar() or cls() new_ui.ui_section = 'hooks' new_ui.ui_active = True new_ui.ui_key = key new_ui.ui_value = val - Session.add( new_ui ) + Session.add(new_ui) Session.commit() -class User( Base, BaseModel ): +class User(Base, BaseModel): __tablename__ = 'users' - __table_args__ = ( UniqueConstraint( 'username' ), UniqueConstraint( 'email' ), {'extend_existing':True} ) - user_id = Column( "user_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - username = Column( "username", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - password = Column( "password", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - active = Column( "active", Boolean(), nullable = True, unique = None, default = None ) - admin = Column( "admin", Boolean(), nullable = True, unique = None, default = False ) - name = Column( "name", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - lastname = Column( "lastname", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - email = Column( "email", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - last_login = Column( "last_login", DateTime( timezone = False ), nullable = True, unique = None, default = None ) - ldap_dn = Column( "ldap_dn", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - api_key = Column( "api_key", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - - user_log = relationship( 'UserLog', cascade = 'all' ) - user_perms = relationship( 'UserToPerm', primaryjoin = "User.user_id==UserToPerm.user_id", cascade = 'all' ) - - repositories = relationship( 'Repository' ) - user_followers = relationship( 'UserFollowing', primaryjoin = 'UserFollowing.follows_user_id==User.user_id', cascade = 'all' ) - repo_to_perm = relationship( 'RepoToPerm', primaryjoin = 'RepoToPerm.user_id==User.user_id', cascade = 'all' ) - - group_member = relationship( 'UsersGroupMember', cascade = 'all' ) + __table_args__ = (UniqueConstraint('username'), UniqueConstraint('email'), {'extend_existing':True}) + user_id = Column("user_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + username = Column("username", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + password = Column("password", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + active = Column("active", Boolean(), nullable = True, unique = None, default = None) + admin = Column("admin", Boolean(), nullable = True, unique = None, default = False) + name = Column("name", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + lastname = Column("lastname", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + email = Column("email", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + last_login = Column("last_login", DateTime(timezone = False), nullable = True, unique = None, default = None) + ldap_dn = Column("ldap_dn", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + api_key = Column("api_key", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + + user_log = relationship('UserLog', cascade = 'all') + user_perms = relationship('UserToPerm', primaryjoin = "User.user_id==UserToPerm.user_id", cascade = 'all') + + repositories = relationship('Repository') + user_followers = relationship('UserFollowing', primaryjoin = 'UserFollowing.follows_user_id==User.user_id', cascade = 'all') + repo_to_perm = relationship('RepoToPerm', primaryjoin = 'RepoToPerm.user_id==User.user_id', cascade = 'all') + + group_member = relationship('UsersGroupMember', cascade = 'all') @property - def full_contact( self ): - return '%s %s <%s>' % ( self.name, self.lastname, self.email ) + def full_contact(self): + return '%s %s <%s>' % (self.name, self.lastname, self.email) @property - def short_contact( self ): - return '%s %s' % ( self.name, self.lastname ) + def short_contact(self): + return '%s %s' % (self.name, self.lastname) @property - def is_admin( self ): + def is_admin(self): return self.admin - def __repr__( self ): + def __repr__(self): try: - return "<%s('id:%s:%s')>" % ( self.__class__.__name__, - self.user_id, self.username ) + return "<%s('id:%s:%s')>" % (self.__class__.__name__, + self.user_id, self.username) except: return self.__class__.__name__ @classmethod - def get_by_username( cls, username, case_insensitive = False ): + def get_by_username(cls, username, case_insensitive = False): if case_insensitive: - return Session.query( cls ).filter( cls.username.ilike( username ) ).scalar() + return Session.query(cls).filter(cls.username.ilike(username)).scalar() else: - return Session.query( cls ).filter( cls.username == username ).scalar() + return Session.query(cls).filter(cls.username == username).scalar() @classmethod - def get_by_api_key( cls, api_key ): - return cls.query().filter( cls.api_key == api_key ).one() + def get_by_api_key(cls, api_key): + return cls.query().filter(cls.api_key == api_key).one() - def update_lastlogin( self ): + def update_lastlogin(self): """Update user lastlogin""" self.last_login = datetime.datetime.now() - Session.add( self ) + Session.add(self) Session.commit() - log.debug( 'updated user %s lastlogin', self.username ) + log.debug('updated user %s lastlogin', self.username) @classmethod - def create( cls, form_data ): + def create(cls, form_data): from rhodecode.lib.auth import get_crypt_password try: new_user = cls() for k, v in form_data.items(): if k == 'password': - v = get_crypt_password( v ) - setattr( new_user, k, v ) + v = get_crypt_password(v) + setattr(new_user, k, v) - new_user.api_key = generate_api_key( form_data['username'] ) - Session.add( new_user ) + new_user.api_key = generate_api_key(form_data['username']) + Session.add(new_user) Session.commit() return new_user except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) Session.rollback() raise -class UserLog( Base, BaseModel ): +class UserLog(Base, BaseModel): __tablename__ = 'user_logs' __table_args__ = {'extend_existing':True} - user_log_id = Column( "user_log_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - user_id = Column( "user_id", Integer(), ForeignKey( 'users.user_id' ), nullable = False, unique = None, default = None ) - repository_id = Column( "repository_id", Integer(), ForeignKey( 'repositories.repo_id' ), nullable = False, unique = None, default = None ) - repository_name = Column( "repository_name", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - user_ip = Column( "user_ip", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - action = Column( "action", UnicodeText( length = 1200000, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - action_date = Column( "action_date", DateTime( timezone = False ), nullable = True, unique = None, default = None ) + user_log_id = Column("user_log_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable = False, unique = None, default = None) + repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable = False, unique = None, default = None) + repository_name = Column("repository_name", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + user_ip = Column("user_ip", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + action = Column("action", UnicodeText(length = 1200000, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + action_date = Column("action_date", DateTime(timezone = False), nullable = True, unique = None, default = None) @property - def action_as_day( self ): - return date( *self.action_date.timetuple()[:3] ) + def action_as_day(self): + return date(*self.action_date.timetuple()[:3]) - user = relationship( 'User' ) - repository = relationship( 'Repository' ) + user = relationship('User') + repository = relationship('Repository') -class UsersGroup( Base, BaseModel ): +class UsersGroup(Base, BaseModel): __tablename__ = 'users_groups' __table_args__ = {'extend_existing':True} - users_group_id = Column( "users_group_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - users_group_name = Column( "users_group_name", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = False, unique = True, default = None ) - users_group_active = Column( "users_group_active", Boolean(), nullable = True, unique = None, default = None ) + users_group_id = Column("users_group_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + users_group_name = Column("users_group_name", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = False, unique = True, default = None) + users_group_active = Column("users_group_active", Boolean(), nullable = True, unique = None, default = None) - members = relationship( 'UsersGroupMember', cascade = "all, delete, delete-orphan", lazy = "joined" ) + members = relationship('UsersGroupMember', cascade = "all, delete, delete-orphan", lazy = "joined") - def __repr__( self ): - return '<userGroup(%s)>' % ( self.users_group_name ) + def __repr__(self): + return '<userGroup(%s)>' % (self.users_group_name) @classmethod - def get_by_group_name( cls, group_name, cache = False, case_insensitive = False ): + def get_by_group_name(cls, group_name, cache = False, case_insensitive = False): if case_insensitive: gr = cls.query()\ - .filter( cls.users_group_name.ilike( group_name ) ) + .filter(cls.users_group_name.ilike(group_name)) else: gr = cls.query()\ - .filter( cls.users_group_name == group_name ) + .filter(cls.users_group_name == group_name) if cache: - gr = gr.options( FromCache( "sql_cache_short", - "get_user_%s" % group_name ) ) + gr = gr.options(FromCache("sql_cache_short", + "get_user_%s" % group_name)) return gr.scalar() @classmethod - def get( cls, users_group_id, cache = False ): + def get(cls, users_group_id, cache = False): users_group = cls.query() if cache: - users_group = users_group.options( FromCache( "sql_cache_short", - "get_users_group_%s" % users_group_id ) ) - return users_group.get( users_group_id ) + users_group = users_group.options(FromCache("sql_cache_short", + "get_users_group_%s" % users_group_id)) + return users_group.get(users_group_id) @classmethod - def create( cls, form_data ): + def create(cls, form_data): try: new_users_group = cls() for k, v in form_data.items(): - setattr( new_users_group, k, v ) + setattr(new_users_group, k, v) - Session.add( new_users_group ) + Session.add(new_users_group) Session.commit() return new_users_group except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) Session.rollback() raise @classmethod - def update( cls, users_group_id, form_data ): + def update(cls, users_group_id, form_data): try: - users_group = cls.get( users_group_id, cache = False ) + users_group = cls.get(users_group_id, cache = False) for k, v in form_data.items(): if k == 'users_group_members': @@ -427,181 +427,181 @@ class UsersGroup( Base, BaseModel ): if v: v = [v] if isinstance(v, basestring) else v for u_id in set(v): - member = UsersGroupMember( users_group_id, u_id ) - members_list.append( member ) - setattr( users_group, 'members', members_list ) - setattr( users_group, k, v ) + member = UsersGroupMember(users_group_id, u_id) + members_list.append(member) + setattr(users_group, 'members', members_list) + setattr(users_group, k, v) - Session.add( users_group ) + Session.add(users_group) Session.commit() except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) Session.rollback() raise @classmethod - def delete( cls, users_group_id ): + def delete(cls, users_group_id): try: # check if this group is not assigned to repo assigned_groups = UsersGroupRepoToPerm.query()\ - .filter( UsersGroupRepoToPerm.users_group_id == - users_group_id ).all() + .filter(UsersGroupRepoToPerm.users_group_id == + users_group_id).all() if assigned_groups: - raise UsersGroupsAssignedException( 'Group assigned to %s' % - assigned_groups ) + raise UsersGroupsAssignedException('Group assigned to %s' % + assigned_groups) - users_group = cls.get( users_group_id, cache = False ) - Session.delete( users_group ) + users_group = cls.get(users_group_id, cache = False) + Session.delete(users_group) Session.commit() except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) Session.rollback() raise -class UsersGroupMember( Base, BaseModel ): +class UsersGroupMember(Base, BaseModel): __tablename__ = 'users_groups_members' __table_args__ = {'extend_existing':True} - users_group_member_id = Column( "users_group_member_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - users_group_id = Column( "users_group_id", Integer(), ForeignKey( 'users_groups.users_group_id' ), nullable = False, unique = None, default = None ) - user_id = Column( "user_id", Integer(), ForeignKey( 'users.user_id' ), nullable = False, unique = None, default = None ) + users_group_member_id = Column("users_group_member_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable = False, unique = None, default = None) + user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable = False, unique = None, default = None) - user = relationship( 'User', lazy = 'joined' ) - users_group = relationship( 'UsersGroup' ) + user = relationship('User', lazy = 'joined') + users_group = relationship('UsersGroup') - def __init__( self, gr_id = '', u_id = '' ): + def __init__(self, gr_id = '', u_id = ''): self.users_group_id = gr_id self.user_id = u_id @staticmethod - def add_user_to_group( group, user ): + def add_user_to_group(group, user): ugm = UsersGroupMember() ugm.users_group = group ugm.user = user - Session.add( ugm ) + Session.add(ugm) Session.commit() return ugm -class Repository( Base, BaseModel ): +class Repository(Base, BaseModel): __tablename__ = 'repositories' - __table_args__ = ( UniqueConstraint( 'repo_name' ), {'extend_existing':True}, ) + __table_args__ = (UniqueConstraint('repo_name'), {'extend_existing':True},) - repo_id = Column( "repo_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - repo_name = Column( "repo_name", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = False, unique = True, default = None ) - clone_uri = Column( "clone_uri", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = False, default = None ) - repo_type = Column( "repo_type", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = False, unique = False, default = 'hg' ) - user_id = Column( "user_id", Integer(), ForeignKey( 'users.user_id' ), nullable = False, unique = False, default = None ) - private = Column( "private", Boolean(), nullable = True, unique = None, default = None ) - enable_statistics = Column( "statistics", Boolean(), nullable = True, unique = None, default = True ) - enable_downloads = Column( "downloads", Boolean(), nullable = True, unique = None, default = True ) - description = Column( "description", String( length = 10000, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - created_on = Column( 'created_on', DateTime( timezone = False ), nullable = True, unique = None, default = datetime.datetime.now ) + repo_id = Column("repo_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + repo_name = Column("repo_name", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = False, unique = True, default = None) + clone_uri = Column("clone_uri", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = False, default = None) + repo_type = Column("repo_type", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = False, unique = False, default = 'hg') + user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable = False, unique = False, default = None) + private = Column("private", Boolean(), nullable = True, unique = None, default = None) + enable_statistics = Column("statistics", Boolean(), nullable = True, unique = None, default = True) + enable_downloads = Column("downloads", Boolean(), nullable = True, unique = None, default = True) + description = Column("description", String(length = 10000, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + created_on = Column('created_on', DateTime(timezone = False), nullable = True, unique = None, default = datetime.datetime.now) - fork_id = Column( "fork_id", Integer(), ForeignKey( 'repositories.repo_id' ), nullable = True, unique = False, default = None ) - group_id = Column( "group_id", Integer(), ForeignKey( 'groups.group_id' ), nullable = True, unique = False, default = None ) + fork_id = Column("fork_id", Integer(), ForeignKey('repositories.repo_id'), nullable = True, unique = False, default = None) + group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable = True, unique = False, default = None) - user = relationship( 'User' ) - fork = relationship( 'Repository', remote_side = repo_id ) - group = relationship( 'Group' ) - repo_to_perm = relationship( 'RepoToPerm', cascade = 'all', order_by = 'RepoToPerm.repo_to_perm_id' ) - users_group_to_perm = relationship( 'UsersGroupRepoToPerm', cascade = 'all' ) - stats = relationship( 'Statistics', cascade = 'all', uselist = False ) + user = relationship('User') + fork = relationship('Repository', remote_side = repo_id) + group = relationship('Group') + repo_to_perm = relationship('RepoToPerm', cascade = 'all', order_by = 'RepoToPerm.repo_to_perm_id') + users_group_to_perm = relationship('UsersGroupRepoToPerm', cascade = 'all') + stats = relationship('Statistics', cascade = 'all', uselist = False) - followers = relationship( 'UserFollowing', primaryjoin = 'UserFollowing.follows_repo_id==Repository.repo_id', cascade = 'all' ) + followers = relationship('UserFollowing', primaryjoin = 'UserFollowing.follows_repo_id==Repository.repo_id', cascade = 'all') - logs = relationship( 'UserLog', cascade = 'all' ) + logs = relationship('UserLog', cascade = 'all') - def __repr__( self ): - return "<%s('%s:%s')>" % ( self.__class__.__name__, - self.repo_id, self.repo_name ) + def __repr__(self): + return "<%s('%s:%s')>" % (self.__class__.__name__, + self.repo_id, self.repo_name) @classmethod - def url_sep( cls ): + def url_sep(cls): return '/' @classmethod - def get_by_repo_name( cls, repo_name ): - q = Session.query( cls ).filter( cls.repo_name == repo_name ) - q = q.options( joinedload( Repository.fork ) )\ - .options( joinedload( Repository.user ) )\ - .options( joinedload( Repository.group ) ) + def get_by_repo_name(cls, repo_name): + q = Session.query(cls).filter(cls.repo_name == repo_name) + q = q.options(joinedload(Repository.fork))\ + .options(joinedload(Repository.user))\ + .options(joinedload(Repository.group)) return q.one() @classmethod - def get_repo_forks( cls, repo_id ): - return cls.query().filter( Repository.fork_id == repo_id ) + def get_repo_forks(cls, repo_id): + return cls.query().filter(Repository.fork_id == repo_id) @classmethod - def base_path( cls ): + def base_path(cls): """ Returns base path when all repos are stored :param cls: """ - q = Session.query( RhodeCodeUi ).filter( RhodeCodeUi.ui_key == - cls.url_sep() ) - q.options( FromCache( "sql_cache_short", "repository_repo_path" ) ) + q = Session.query(RhodeCodeUi).filter(RhodeCodeUi.ui_key == + cls.url_sep()) + q.options(FromCache("sql_cache_short", "repository_repo_path")) return q.one().ui_value @property - def just_name( self ): - return self.repo_name.split( Repository.url_sep() )[-1] + def just_name(self): + return self.repo_name.split(Repository.url_sep())[-1] @property - def groups_with_parents( self ): + def groups_with_parents(self): groups = [] if self.group is None: return groups cur_gr = self.group - groups.insert( 0, cur_gr ) + groups.insert(0, cur_gr) while 1: - gr = getattr( cur_gr, 'parent_group', None ) + gr = getattr(cur_gr, 'parent_group', None) cur_gr = cur_gr.parent_group if gr is None: break - groups.insert( 0, gr ) + groups.insert(0, gr) return groups @property - def groups_and_repo( self ): + def groups_and_repo(self): return self.groups_with_parents, self.just_name @LazyProperty - def repo_path( self ): + def repo_path(self): """ Returns base full path for that repository means where it actually exists on a filesystem """ - q = Session.query( RhodeCodeUi ).filter( RhodeCodeUi.ui_key == - Repository.url_sep() ) - q.options( FromCache( "sql_cache_short", "repository_repo_path" ) ) + q = Session.query(RhodeCodeUi).filter(RhodeCodeUi.ui_key == + Repository.url_sep()) + q.options(FromCache("sql_cache_short", "repository_repo_path")) return q.one().ui_value @property - def repo_full_path( self ): + def repo_full_path(self): p = [self.repo_path] # we need to split the name by / since this is how we store the # names in the database, but that eventually needs to be converted # into a valid system path - p += self.repo_name.split( Repository.url_sep() ) - return os.path.join( *p ) + p += self.repo_name.split(Repository.url_sep()) + return os.path.join(*p) - def get_new_name( self, repo_name ): + def get_new_name(self, repo_name): """ returns new full repository name based on assigned group and new new :param group_name: """ path_prefix = self.group.full_path_splitted if self.group else [] - return Repository.url_sep().join( path_prefix + [repo_name] ) + return Repository.url_sep().join(path_prefix + [repo_name]) @property - def _ui( self ): + def _ui(self): """ Creates an db based ui object for this repository """ @@ -616,19 +616,19 @@ class Repository( Base, BaseModel ): ret = RhodeCodeUi.query()\ - .options( FromCache( "sql_cache_short", "repository_repo_ui" ) ).all() + .options(FromCache("sql_cache_short", "repository_repo_ui")).all() hg_ui = ret for ui_ in hg_ui: if ui_.ui_active: - log.debug( 'settings ui from db[%s]%s:%s', ui_.ui_section, - ui_.ui_key, ui_.ui_value ) - baseui.setconfig( ui_.ui_section, ui_.ui_key, ui_.ui_value ) + log.debug('settings ui from db[%s]%s:%s', ui_.ui_section, + ui_.ui_key, ui_.ui_value) + baseui.setconfig(ui_.ui_section, ui_.ui_key, ui_.ui_value) return baseui @classmethod - def is_valid( cls, repo_name ): + def is_valid(cls, repo_name): """ returns True if given repo name is a valid filesystem repository @@ -637,26 +637,26 @@ class Repository( Base, BaseModel ): """ from rhodecode.lib.utils import is_valid_repo - return is_valid_repo( repo_name, cls.base_path() ) + return is_valid_repo(repo_name, cls.base_path()) #========================================================================== # SCM PROPERTIES #========================================================================== - def get_changeset( self, rev ): - return get_changeset_safe( self.scm_instance, rev ) + def get_changeset(self, rev): + return get_changeset_safe(self.scm_instance, rev) @property - def tip( self ): - return self.get_changeset( 'tip' ) + def tip(self): + return self.get_changeset('tip') @property - def author( self ): + def author(self): return self.tip.author @property - def last_change( self ): + def last_change(self): return self.scm_instance.last_change #========================================================================== @@ -664,407 +664,407 @@ class Repository( Base, BaseModel ): #========================================================================== @property - def invalidate( self ): + def invalidate(self): """ Returns Invalidation object if this repo should be invalidated None otherwise. `cache_active = False` means that this cache state is not valid and needs to be invalidated """ return CacheInvalidation.query()\ - .filter( CacheInvalidation.cache_key == self.repo_name )\ - .filter( CacheInvalidation.cache_active == False )\ + .filter(CacheInvalidation.cache_key == self.repo_name)\ + .filter(CacheInvalidation.cache_active == False)\ .scalar() - def set_invalidate( self ): + def set_invalidate(self): """ set a cache for invalidation for this instance """ inv = CacheInvalidation.query()\ - .filter( CacheInvalidation.cache_key == self.repo_name )\ + .filter(CacheInvalidation.cache_key == self.repo_name)\ .scalar() if inv is None: - inv = CacheInvalidation( self.repo_name ) + inv = CacheInvalidation(self.repo_name) inv.cache_active = True - Session.add( inv ) + Session.add(inv) Session.commit() @LazyProperty - def scm_instance( self ): + def scm_instance(self): return self.__get_instance() @property - def scm_instance_cached( self ): - @cache_region( 'long_term' ) - def _c( repo_name ): + def scm_instance_cached(self): + @cache_region('long_term') + def _c(repo_name): return self.__get_instance() # TODO: remove this trick when beaker 1.6 is released # and have fixed this issue with not supporting unicode keys - rn = safe_str( self.repo_name ) + rn = safe_str(self.repo_name) inv = self.invalidate if inv is not None: - region_invalidate( _c, None, rn ) + region_invalidate(_c, None, rn) # update our cache inv.cache_active = True - Session.add( inv ) + Session.add(inv) Session.commit() - return _c( rn ) + return _c(rn) - def __get_instance( self ): + def __get_instance(self): repo_full_path = self.repo_full_path try: - alias = get_scm( repo_full_path )[0] - log.debug( 'Creating instance of %s repository', alias ) - backend = get_backend( alias ) + alias = get_scm(repo_full_path)[0] + log.debug('Creating instance of %s repository', alias) + backend = get_backend(alias) except VCSError: - log.error( traceback.format_exc() ) - log.error( 'Perhaps this repository is in db and not in ' + log.error(traceback.format_exc()) + log.error('Perhaps this repository is in db and not in ' 'filesystem run rescan repositories with ' - '"destroy old data " option from admin panel' ) + '"destroy old data " option from admin panel') return if alias == 'hg': - repo = backend( safe_str( repo_full_path ), create = False, - baseui = self._ui ) + repo = backend(safe_str(repo_full_path), create = False, + baseui = self._ui) #skip hidden web repository if repo._get_hidden(): return else: - repo = backend( repo_full_path, create = False ) + repo = backend(repo_full_path, create = False) return repo -class Group( Base, BaseModel ): +class Group(Base, BaseModel): __tablename__ = 'groups' - __table_args__ = ( UniqueConstraint( 'group_name', 'group_parent_id' ), - CheckConstraint( 'group_id != group_parent_id' ), {'extend_existing':True}, ) + __table_args__ = (UniqueConstraint('group_name', 'group_parent_id'), + CheckConstraint('group_id != group_parent_id'), {'extend_existing':True},) __mapper_args__ = {'order_by':'group_name'} - group_id = Column( "group_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - group_name = Column( "group_name", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = False, unique = True, default = None ) - group_parent_id = Column( "group_parent_id", Integer(), ForeignKey( 'groups.group_id' ), nullable = True, unique = None, default = None ) - group_description = Column( "group_description", String( length = 10000, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) + group_id = Column("group_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + group_name = Column("group_name", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = False, unique = True, default = None) + group_parent_id = Column("group_parent_id", Integer(), ForeignKey('groups.group_id'), nullable = True, unique = None, default = None) + group_description = Column("group_description", String(length = 10000, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) - parent_group = relationship( 'Group', remote_side = group_id ) + parent_group = relationship('Group', remote_side = group_id) - def __init__( self, group_name = '', parent_group = None ): + def __init__(self, group_name = '', parent_group = None): self.group_name = group_name self.parent_group = parent_group - def __repr__( self ): - return "<%s('%s:%s')>" % ( self.__class__.__name__, self.group_id, - self.group_name ) + def __repr__(self): + return "<%s('%s:%s')>" % (self.__class__.__name__, self.group_id, + self.group_name) @classmethod - def groups_choices( cls ): + def groups_choices(cls): from webhelpers.html import literal as _literal - repo_groups = [( '', '' )] + repo_groups = [('', '')] sep = ' » ' - _name = lambda k: _literal( sep.join( k ) ) + _name = lambda k: _literal(sep.join(k)) - repo_groups.extend( [( x.group_id, _name( x.full_path_splitted ) ) - for x in cls.query().all()] ) + repo_groups.extend([(x.group_id, _name(x.full_path_splitted)) + for x in cls.query().all()]) - repo_groups = sorted( repo_groups, key = lambda t: t[1].split( sep )[0] ) + repo_groups = sorted(repo_groups, key = lambda t: t[1].split(sep)[0]) return repo_groups @classmethod - def url_sep( cls ): + def url_sep(cls): return '/' @classmethod - def get_by_group_name( cls, group_name, cache = False, case_insensitive = False ): + def get_by_group_name(cls, group_name, cache = False, case_insensitive = False): if case_insensitive: gr = cls.query()\ - .filter( cls.group_name.ilike( group_name ) ) + .filter(cls.group_name.ilike(group_name)) else: gr = cls.query()\ - .filter( cls.group_name == group_name ) + .filter(cls.group_name == group_name) if cache: - gr = gr.options( FromCache( "sql_cache_short", - "get_group_%s" % group_name ) ) + gr = gr.options(FromCache("sql_cache_short", + "get_group_%s" % group_name)) return gr.scalar() @property - def parents( self ): + def parents(self): parents_recursion_limit = 5 groups = [] if self.parent_group is None: return groups cur_gr = self.parent_group - groups.insert( 0, cur_gr ) + groups.insert(0, cur_gr) cnt = 0 while 1: cnt += 1 - gr = getattr( cur_gr, 'parent_group', None ) + gr = getattr(cur_gr, 'parent_group', None) cur_gr = cur_gr.parent_group if gr is None: break if cnt == parents_recursion_limit: # this will prevent accidental infinit loops - log.error( 'group nested more than %s' % - parents_recursion_limit ) + log.error('group nested more than %s' % + parents_recursion_limit) break - groups.insert( 0, gr ) + groups.insert(0, gr) return groups @property - def children( self ): - return Group.query().filter( Group.parent_group == self ) + def children(self): + return Group.query().filter(Group.parent_group == self) @property - def name( self ): - return self.group_name.split( Group.url_sep() )[-1] + def name(self): + return self.group_name.split(Group.url_sep())[-1] @property - def full_path( self ): - return Group.url_sep().join( [g.group_name for g in self.parents] + - [self.group_name] ) + def full_path(self): + return Group.url_sep().join([g.group_name for g in self.parents] + + [self.group_name]) @property - def full_path_splitted( self ): - return self.group_name.split( Group.url_sep() ) + def full_path_splitted(self): + return self.group_name.split(Group.url_sep()) @property - def repositories( self ): - return Repository.query().filter( Repository.group == self ) + def repositories(self): + return Repository.query().filter(Repository.group == self) @property - def repositories_recursive_count( self ): + def repositories_recursive_count(self): cnt = self.repositories.count() - def children_count( group ): + def children_count(group): cnt = 0 for child in group.children: cnt += child.repositories.count() - cnt += children_count( child ) + cnt += children_count(child) return cnt - return cnt + children_count( self ) + return cnt + children_count(self) - def get_new_name( self, group_name ): + def get_new_name(self, group_name): """ returns new full group name based on parent and new name :param group_name: """ path_prefix = self.parent_group.full_path_splitted if self.parent_group else [] - return Group.url_sep().join( path_prefix + [group_name] ) + return Group.url_sep().join(path_prefix + [group_name]) -class Permission( Base, BaseModel ): +class Permission(Base, BaseModel): __tablename__ = 'permissions' __table_args__ = {'extend_existing':True} - permission_id = Column( "permission_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - permission_name = Column( "permission_name", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - permission_longname = Column( "permission_longname", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) + permission_id = Column("permission_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + permission_name = Column("permission_name", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + permission_longname = Column("permission_longname", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) - def __repr__( self ): - return "<%s('%s:%s')>" % ( self.__class__.__name__, - self.permission_id, self.permission_name ) + def __repr__(self): + return "<%s('%s:%s')>" % (self.__class__.__name__, + self.permission_id, self.permission_name) @classmethod - def get_by_key( cls, key ): - return cls.query().filter( cls.permission_name == key ).scalar() + def get_by_key(cls, key): + return cls.query().filter(cls.permission_name == key).scalar() -class RepoToPerm( Base, BaseModel ): +class RepoToPerm(Base, BaseModel): __tablename__ = 'repo_to_perm' - __table_args__ = ( UniqueConstraint( 'user_id', 'repository_id' ), {'extend_existing':True} ) - repo_to_perm_id = Column( "repo_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - user_id = Column( "user_id", Integer(), ForeignKey( 'users.user_id' ), nullable = False, unique = None, default = None ) - permission_id = Column( "permission_id", Integer(), ForeignKey( 'permissions.permission_id' ), nullable = False, unique = None, default = None ) - repository_id = Column( "repository_id", Integer(), ForeignKey( 'repositories.repo_id' ), nullable = False, unique = None, default = None ) + __table_args__ = (UniqueConstraint('user_id', 'repository_id'), {'extend_existing':True}) + repo_to_perm_id = Column("repo_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable = False, unique = None, default = None) + permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable = False, unique = None, default = None) + repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable = False, unique = None, default = None) - user = relationship( 'User' ) - permission = relationship( 'Permission' ) - repository = relationship( 'Repository' ) + user = relationship('User') + permission = relationship('Permission') + repository = relationship('Repository') -class UserToPerm( Base, BaseModel ): +class UserToPerm(Base, BaseModel): __tablename__ = 'user_to_perm' - __table_args__ = ( UniqueConstraint( 'user_id', 'permission_id' ), {'extend_existing':True} ) - user_to_perm_id = Column( "user_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - user_id = Column( "user_id", Integer(), ForeignKey( 'users.user_id' ), nullable = False, unique = None, default = None ) - permission_id = Column( "permission_id", Integer(), ForeignKey( 'permissions.permission_id' ), nullable = False, unique = None, default = None ) + __table_args__ = (UniqueConstraint('user_id', 'permission_id'), {'extend_existing':True}) + user_to_perm_id = Column("user_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable = False, unique = None, default = None) + permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable = False, unique = None, default = None) - user = relationship( 'User' ) - permission = relationship( 'Permission' ) + user = relationship('User') + permission = relationship('Permission') @classmethod - def has_perm( cls, user_id, perm ): - if not isinstance( perm, Permission ): - raise Exception( 'perm needs to be an instance of Permission class' ) + def has_perm(cls, user_id, perm): + if not isinstance(perm, Permission): + raise Exception('perm needs to be an instance of Permission class') - return cls.query().filter( cls.user_id == user_id )\ - .filter( cls.permission == perm ).scalar() is not None + return cls.query().filter(cls.user_id == user_id)\ + .filter(cls.permission == perm).scalar() is not None @classmethod - def grant_perm( cls, user_id, perm ): - if not isinstance( perm, Permission ): - raise Exception( 'perm needs to be an instance of Permission class' ) + def grant_perm(cls, user_id, perm): + if not isinstance(perm, Permission): + raise Exception('perm needs to be an instance of Permission class') new = cls() new.user_id = user_id new.permission = perm try: - Session.add( new ) + Session.add(new) Session.commit() except: Session.rollback() @classmethod - def revoke_perm( cls, user_id, perm ): - if not isinstance( perm, Permission ): - raise Exception( 'perm needs to be an instance of Permission class' ) + def revoke_perm(cls, user_id, perm): + if not isinstance(perm, Permission): + raise Exception('perm needs to be an instance of Permission class') try: - cls.query().filter( cls.user_id == user_id )\ - .filter( cls.permission == perm ).delete() + cls.query().filter(cls.user_id == user_id)\ + .filter(cls.permission == perm).delete() Session.commit() except: Session.rollback() -class UsersGroupRepoToPerm( Base, BaseModel ): +class UsersGroupRepoToPerm(Base, BaseModel): __tablename__ = 'users_group_repo_to_perm' - __table_args__ = ( UniqueConstraint( 'repository_id', 'users_group_id', 'permission_id' ), {'extend_existing':True} ) - users_group_to_perm_id = Column( "users_group_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - users_group_id = Column( "users_group_id", Integer(), ForeignKey( 'users_groups.users_group_id' ), nullable = False, unique = None, default = None ) - permission_id = Column( "permission_id", Integer(), ForeignKey( 'permissions.permission_id' ), nullable = False, unique = None, default = None ) - repository_id = Column( "repository_id", Integer(), ForeignKey( 'repositories.repo_id' ), nullable = False, unique = None, default = None ) + __table_args__ = (UniqueConstraint('repository_id', 'users_group_id', 'permission_id'), {'extend_existing':True}) + users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable = False, unique = None, default = None) + permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable = False, unique = None, default = None) + repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable = False, unique = None, default = None) - users_group = relationship( 'UsersGroup' ) - permission = relationship( 'Permission' ) - repository = relationship( 'Repository' ) + users_group = relationship('UsersGroup') + permission = relationship('Permission') + repository = relationship('Repository') - def __repr__( self ): - return '<userGroup:%s => %s >' % ( self.users_group, self.repository ) + def __repr__(self): + return '<userGroup:%s => %s >' % (self.users_group, self.repository) -class UsersGroupToPerm( Base, BaseModel ): +class UsersGroupToPerm(Base, BaseModel): __tablename__ = 'users_group_to_perm' - users_group_to_perm_id = Column( "users_group_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - users_group_id = Column( "users_group_id", Integer(), ForeignKey( 'users_groups.users_group_id' ), nullable = False, unique = None, default = None ) - permission_id = Column( "permission_id", Integer(), ForeignKey( 'permissions.permission_id' ), nullable = False, unique = None, default = None ) + users_group_to_perm_id = Column("users_group_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + users_group_id = Column("users_group_id", Integer(), ForeignKey('users_groups.users_group_id'), nullable = False, unique = None, default = None) + permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable = False, unique = None, default = None) - users_group = relationship( 'UsersGroup' ) - permission = relationship( 'Permission' ) + users_group = relationship('UsersGroup') + permission = relationship('Permission') @classmethod - def has_perm( cls, users_group_id, perm ): - if not isinstance( perm, Permission ): - raise Exception( 'perm needs to be an instance of Permission class' ) + def has_perm(cls, users_group_id, perm): + if not isinstance(perm, Permission): + raise Exception('perm needs to be an instance of Permission class') - return cls.query().filter( cls.users_group_id == - users_group_id )\ - .filter( cls.permission == perm )\ + return cls.query().filter(cls.users_group_id == + users_group_id)\ + .filter(cls.permission == perm)\ .scalar() is not None @classmethod - def grant_perm( cls, users_group_id, perm ): - if not isinstance( perm, Permission ): - raise Exception( 'perm needs to be an instance of Permission class' ) + def grant_perm(cls, users_group_id, perm): + if not isinstance(perm, Permission): + raise Exception('perm needs to be an instance of Permission class') new = cls() new.users_group_id = users_group_id new.permission = perm try: - Session.add( new ) + Session.add(new) Session.commit() except: Session.rollback() @classmethod - def revoke_perm( cls, users_group_id, perm ): - if not isinstance( perm, Permission ): - raise Exception( 'perm needs to be an instance of Permission class' ) + def revoke_perm(cls, users_group_id, perm): + if not isinstance(perm, Permission): + raise Exception('perm needs to be an instance of Permission class') try: - cls.query().filter( cls.users_group_id == users_group_id )\ - .filter( cls.permission == perm ).delete() + cls.query().filter(cls.users_group_id == users_group_id)\ + .filter(cls.permission == perm).delete() Session.commit() except: Session.rollback() -class GroupToPerm( Base, BaseModel ): +class GroupToPerm(Base, BaseModel): __tablename__ = 'group_to_perm' - __table_args__ = ( UniqueConstraint( 'group_id', 'permission_id' ), {'extend_existing':True} ) + __table_args__ = (UniqueConstraint('group_id', 'permission_id'), {'extend_existing':True}) - group_to_perm_id = Column( "group_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - user_id = Column( "user_id", Integer(), ForeignKey( 'users.user_id' ), nullable = False, unique = None, default = None ) - permission_id = Column( "permission_id", Integer(), ForeignKey( 'permissions.permission_id' ), nullable = False, unique = None, default = None ) - group_id = Column( "group_id", Integer(), ForeignKey( 'groups.group_id' ), nullable = False, unique = None, default = None ) + group_to_perm_id = Column("group_to_perm_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable = False, unique = None, default = None) + permission_id = Column("permission_id", Integer(), ForeignKey('permissions.permission_id'), nullable = False, unique = None, default = None) + group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), nullable = False, unique = None, default = None) - user = relationship( 'User' ) - permission = relationship( 'Permission' ) - group = relationship( 'Group' ) + user = relationship('User') + permission = relationship('Permission') + group = relationship('Group') -class Statistics( Base, BaseModel ): +class Statistics(Base, BaseModel): __tablename__ = 'statistics' - __table_args__ = ( UniqueConstraint( 'repository_id' ), {'extend_existing':True} ) - stat_id = Column( "stat_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - repository_id = Column( "repository_id", Integer(), ForeignKey( 'repositories.repo_id' ), nullable = False, unique = True, default = None ) - stat_on_revision = Column( "stat_on_revision", Integer(), nullable = False ) - commit_activity = Column( "commit_activity", LargeBinary( 1000000 ), nullable = False )#JSON data - commit_activity_combined = Column( "commit_activity_combined", LargeBinary(), nullable = False )#JSON data - languages = Column( "languages", LargeBinary( 1000000 ), nullable = False )#JSON data + __table_args__ = (UniqueConstraint('repository_id'), {'extend_existing':True}) + stat_id = Column("stat_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + repository_id = Column("repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable = False, unique = True, default = None) + stat_on_revision = Column("stat_on_revision", Integer(), nullable = False) + commit_activity = Column("commit_activity", LargeBinary(1000000), nullable = False)#JSON data + commit_activity_combined = Column("commit_activity_combined", LargeBinary(), nullable = False)#JSON data + languages = Column("languages", LargeBinary(1000000), nullable = False)#JSON data - repository = relationship( 'Repository', single_parent = True ) + repository = relationship('Repository', single_parent = True) -class UserFollowing( Base, BaseModel ): +class UserFollowing(Base, BaseModel): __tablename__ = 'user_followings' - __table_args__ = ( UniqueConstraint( 'user_id', 'follows_repository_id' ), - UniqueConstraint( 'user_id', 'follows_user_id' ) - , {'extend_existing':True} ) + __table_args__ = (UniqueConstraint('user_id', 'follows_repository_id'), + UniqueConstraint('user_id', 'follows_user_id') + , {'extend_existing':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( 'users.user_id' ), nullable = False, unique = None, default = None ) - follows_repo_id = Column( "follows_repository_id", Integer(), ForeignKey( 'repositories.repo_id' ), nullable = True, unique = None, default = None ) - follows_user_id = Column( "follows_user_id", Integer(), ForeignKey( 'users.user_id' ), nullable = True, unique = None, default = None ) - follows_from = Column( 'follows_from', DateTime( timezone = False ), nullable = True, unique = None, default = datetime.datetime.now ) + user_following_id = Column("user_following_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable = False, unique = None, default = None) + follows_repo_id = Column("follows_repository_id", Integer(), ForeignKey('repositories.repo_id'), nullable = True, unique = None, default = None) + follows_user_id = Column("follows_user_id", Integer(), ForeignKey('users.user_id'), nullable = True, unique = None, default = None) + follows_from = Column('follows_from', DateTime(timezone = False), nullable = True, unique = None, default = datetime.datetime.now) - user = relationship( 'User', primaryjoin = 'User.user_id==UserFollowing.user_id' ) + user = relationship('User', primaryjoin = 'User.user_id==UserFollowing.user_id') - follows_user = relationship( 'User', primaryjoin = 'User.user_id==UserFollowing.follows_user_id' ) - follows_repository = relationship( 'Repository', order_by = 'Repository.repo_name' ) + follows_user = relationship('User', primaryjoin = 'User.user_id==UserFollowing.follows_user_id') + follows_repository = relationship('Repository', order_by = 'Repository.repo_name') @classmethod - def get_repo_followers( cls, repo_id ): - return cls.query().filter( cls.follows_repo_id == repo_id ) + def get_repo_followers(cls, repo_id): + return cls.query().filter(cls.follows_repo_id == repo_id) -class CacheInvalidation( Base, BaseModel ): +class CacheInvalidation(Base, BaseModel): __tablename__ = 'cache_invalidation' - __table_args__ = ( UniqueConstraint( 'cache_key' ), {'extend_existing':True} ) - cache_id = Column( "cache_id", Integer(), nullable = False, unique = True, default = None, primary_key = True ) - cache_key = Column( "cache_key", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - cache_args = Column( "cache_args", String( length = 255, convert_unicode = False, assert_unicode = None ), nullable = True, unique = None, default = None ) - cache_active = Column( "cache_active", Boolean(), nullable = True, unique = None, default = False ) + __table_args__ = (UniqueConstraint('cache_key'), {'extend_existing':True}) + cache_id = Column("cache_id", Integer(), nullable = False, unique = True, default = None, primary_key = True) + cache_key = Column("cache_key", String(length = 255, convert_unicode = False, assert_unicode = None), nullable = True, unique = None, default = None) + cache_args = Column("cache_args", String(length = 255, 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 = '' ): + 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 "<%s('%s:%s')>" % ( self.__class__.__name__, - self.cache_id, self.cache_key ) + def __repr__(self): + return "<%s('%s:%s')>" % (self.__class__.__name__, + self.cache_id, self.cache_key) -class DbMigrateVersion( Base, BaseModel ): +class DbMigrateVersion(Base, BaseModel): __tablename__ = 'db_migrate_version' __table_args__ = {'extend_existing':True} - repository_id = Column( 'repository_id', String( 250 ), primary_key = True ) - repository_path = Column( 'repository_path', Text ) - version = Column( 'version', Integer ) + repository_id = Column('repository_id', String(250), primary_key = True) + repository_path = Column('repository_path', Text) + version = Column('version', Integer) diff --git a/rhodecode/model/repo_permission.py b/rhodecode/model/repo_permission.py index 7b5339fb..1612288e 100644 --- a/rhodecode/model/repo_permission.py +++ b/rhodecode/model/repo_permission.py @@ -27,18 +27,18 @@ import logging from rhodecode.model.db import BaseModel, RepoToPerm, Permission from rhodecode.model.meta import Session -log = logging.getLogger( __name__ ) +log = logging.getLogger(__name__) -class RepositoryPermissionModel( BaseModel ): - def getUserPermission( self, repository, user ): +class RepositoryPermissionModel(BaseModel): + def get_user_permission(self, repository, user): return RepoToPerm.query() \ - .filter( RepoToPerm.user == user ) \ - .filter( RepoToPerm.repository == repository ) \ + .filter(RepoToPerm.user == user) \ + .filter(RepoToPerm.repository == repository) \ .scalar() - def updateUserPermission( self, repository, user, permission ): - permission = Permission.get_by_key( permission ) - current = self.getUserPermission( repository, user ) + def update_user_permission(self, repository, user, permission): + permission = Permission.get_by_key(permission) + current = self.get_user_permission(repository, user) if current: if not current.permission is permission: current.permission = permission @@ -47,17 +47,17 @@ class RepositoryPermissionModel( BaseModel ): p.user = user p.repository = repository p.permission = permission - Session.add( p ) + Session.add(p) Session.commit() - def deleteUserPermission( self, repository, user ): - current = self.getUserPermission( repository, user ) + def delete_user_permission(self, repository, user): + current = self.get_user_permission(repository, user) if current: - Session.delete( current ) + Session.delete(current) Session.commit() - def updateOrDeleteUserPermission( self, repository, user, permission ): + def update_or_delete_user_permission(self, repository, user, permission): if permission: - self.updateUserPermission( repository, user, permission ) + self.update_user_permission(repository, user, permission) else: - self.deleteUserPermission( repository, user ) + self.delete_user_permission(repository, user) diff --git a/rhodecode/model/repos_group.py b/rhodecode/model/repos_group.py index 5da33bec..cad34a1b 100644 --- a/rhodecode/model/repos_group.py +++ b/rhodecode/model/repos_group.py @@ -36,21 +36,21 @@ from rhodecode.model import BaseModel from rhodecode.model.caching_query import FromCache from rhodecode.model.db import Group, RhodeCodeUi -log = logging.getLogger( __name__ ) +log = logging.getLogger(__name__) -class ReposGroupModel( BaseModel ): +class ReposGroupModel(BaseModel): @LazyProperty - def repos_path( self ): + def repos_path(self): """ Get's the repositories root path from database """ - q = RhodeCodeUi.get_by_key( '/' ).one() + q = RhodeCodeUi.get_by_key('/').one() return q.ui_value - def __create_group( self, group_name, parent_id ): + def __create_group(self, group_name, parent_id): """ makes repositories group on filesystem @@ -59,64 +59,64 @@ class ReposGroupModel( BaseModel ): """ if parent_id: - paths = Group.get( parent_id ).full_path.split( Group.url_sep() ) - parent_path = os.sep.join( paths ) + paths = Group.get(parent_id).full_path.split(Group.url_sep()) + parent_path = os.sep.join(paths) else: parent_path = '' - create_path = os.path.join( self.repos_path, parent_path, group_name ) - log.debug( 'creating new group in %s', create_path ) + create_path = os.path.join(self.repos_path, parent_path, group_name) + log.debug('creating new group in %s', create_path) - if os.path.isdir( create_path ): - raise Exception( 'That directory already exists !' ) + if os.path.isdir(create_path): + raise Exception('That directory already exists !') - os.makedirs( create_path ) + os.makedirs(create_path) - def __rename_group( self, old, old_parent_id, new, new_parent_id ): + def __rename_group(self, old, old_parent_id, new, new_parent_id): """ Renames a group on filesystem - + :param group_name: """ - log.debug( 'renaming repos group from %s to %s', old, new ) + log.debug('renaming repos group from %s to %s', old, new) if new_parent_id: - paths = Group.get( new_parent_id ).full_path.split( Group.url_sep() ) - new_parent_path = os.sep.join( paths ) + paths = Group.get(new_parent_id).full_path.split(Group.url_sep()) + new_parent_path = os.sep.join(paths) else: new_parent_path = '' if old_parent_id: - paths = Group.get( old_parent_id ).full_path.split( Group.url_sep() ) - old_parent_path = os.sep.join( paths ) + paths = Group.get(old_parent_id).full_path.split(Group.url_sep()) + old_parent_path = os.sep.join(paths) else: old_parent_path = '' - old_path = os.path.join( self.repos_path, old_parent_path, old ) - new_path = os.path.join( self.repos_path, new_parent_path, new ) + old_path = os.path.join(self.repos_path, old_parent_path, old) + new_path = os.path.join(self.repos_path, new_parent_path, new) - log.debug( 'renaming repos paths from %s to %s', old_path, new_path ) + log.debug('renaming repos paths from %s to %s', old_path, new_path) - if os.path.isdir( new_path ): - raise Exception( 'Was trying to rename to already ' - 'existing dir %s' % new_path ) - shutil.move( old_path, new_path ) + if os.path.isdir(new_path): + raise Exception('Was trying to rename to already ' + 'existing dir %s' % new_path) + shutil.move(old_path, new_path) - def __delete_group( self, group ): + def __delete_group(self, group): """ Deletes a group from a filesystem - + :param group: instance of group from database """ - paths = group.full_path.split( Group.url_sep() ) - paths = os.sep.join( paths ) + paths = group.full_path.split(Group.url_sep()) + paths = os.sep.join(paths) - rm_path = os.path.join( self.repos_path, paths ) - os.rmdir( rm_path ) + rm_path = os.path.join(self.repos_path, paths) + os.rmdir(rm_path) - def create( self, form_data ): + def create(self, form_data): try: new_repos_group = Group() new_repos_group.group_name = form_data['group_name'] @@ -124,22 +124,22 @@ class ReposGroupModel( BaseModel ): form_data['group_description'] new_repos_group.group_parent_id = form_data['group_parent_id'] - self.sa.add( new_repos_group ) + self.sa.add(new_repos_group) - self.__create_group( form_data['group_name'], - form_data['group_parent_id'] ) + self.__create_group(form_data['group_name'], + form_data['group_parent_id']) self.sa.commit() return new_repos_group except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) self.sa.rollback() raise - def update( self, repos_group_id, form_data ): + def update(self, repos_group_id, form_data): try: - repos_group = Group.get( repos_group_id ) + repos_group = Group.get(repos_group_id) old_name = repos_group.group_name old_parent_id = repos_group.group_parent_id @@ -148,27 +148,27 @@ class ReposGroupModel( BaseModel ): form_data['group_description'] repos_group.group_parent_id = form_data['group_parent_id'] - self.sa.add( repos_group ) + self.sa.add(repos_group) - if old_name != form_data['group_name'] or ( old_parent_id != - form_data['group_parent_id'] ): - self.__rename_group( old = old_name, old_parent_id = old_parent_id, + if old_name != form_data['group_name'] or (old_parent_id != + form_data['group_parent_id']): + self.__rename_group(old = old_name, old_parent_id = old_parent_id, new = form_data['group_name'], - new_parent_id = form_data['group_parent_id'] ) + new_parent_id = form_data['group_parent_id']) self.sa.commit() except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) self.sa.rollback() raise - def delete( self, users_group_id ): + def delete(self, users_group_id): try: - users_group = Group.get( users_group_id ) - self.sa.delete( users_group ) - self.__delete_group( users_group ) + users_group = Group.get(users_group_id) + self.sa.delete(users_group) + self.__delete_group(users_group) self.sa.commit() except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) self.sa.rollback() raise diff --git a/rhodecode/model/user.py b/rhodecode/model/user.py index da3db5c2..f81a89c0 100644 --- a/rhodecode/model/user.py +++ b/rhodecode/model/user.py @@ -49,14 +49,14 @@ PERM_WEIGHTS = {'repository.none': 0, class UserModel(BaseModel): - def get(self, user_id, cache=False): + def get(self, user_id, cache = False): user = self.sa.query(User) if cache: user = user.options(FromCache("sql_cache_short", "get_user_%s" % user_id)) return user.get(user_id) - def get_by_username(self, username, cache=False, case_insensitive=False): + def get_by_username(self, username, cache = False, case_insensitive = False): if case_insensitive: user = self.sa.query(User).filter(User.username.ilike(username)) @@ -68,7 +68,7 @@ class UserModel(BaseModel): "get_user_%s" % username)) return user.scalar() - def get_by_api_key(self, api_key, cache=False): + def get_by_api_key(self, api_key, cache = False): user = self.sa.query(User)\ .filter(User.api_key == api_key) @@ -103,7 +103,7 @@ class UserModel(BaseModel): """ from rhodecode.lib.auth import get_crypt_password log.debug('Checking for such ldap account in RhodeCode database') - if self.get_by_username(username, case_insensitive=True) is None: + if self.get_by_username(username, case_insensitive = True) is None: try: new_user = User() # add ldap account always lowercase @@ -152,7 +152,7 @@ class UserModel(BaseModel): def update(self, user_id, form_data): try: - user = self.get(user_id, cache=False) + user = self.get(user_id, cache = False) if user.username == 'default': raise DefaultUserException( _("You can't Edit this user since it's" @@ -174,7 +174,7 @@ class UserModel(BaseModel): def update_my_account(self, user_id, form_data): try: - user = self.get(user_id, cache=False) + user = self.get(user_id, cache = False) if user.username == 'default': raise DefaultUserException( _("You can't Edit this user since it's" @@ -196,7 +196,7 @@ class UserModel(BaseModel): def delete(self, user_id): try: - user = self.get(user_id, cache=False) + user = self.get(user_id, cache = False) if user.username == 'default': raise DefaultUserException( _("You can't remove this user since it's" @@ -222,7 +222,7 @@ class UserModel(BaseModel): from rhodecode.lib.celerylib import tasks, run_task run_task(tasks.reset_user_password, data['email']) - def fill_data(self, auth_user, user_id=None, api_key=None): + def fill_data(self, auth_user, user_id = None, api_key = None): """ Fetches auth_user by user_id,or api_key if present. Fills auth_user attributes with those taken from database. @@ -268,7 +268,7 @@ class UserModel(BaseModel): #====================================================================== # fetch default permissions #====================================================================== - default_user = self.get_by_username('default', cache=True) + default_user = self.get_by_username('default', cache = True) default_perms = self.sa.query(RepoToPerm, Repository, Permission)\ .join((Repository, RepoToPerm.repository_id == diff --git a/rhodecode/model/users_group.py b/rhodecode/model/users_group.py index b94559c9..5c013006 100644 --- a/rhodecode/model/users_group.py +++ b/rhodecode/model/users_group.py @@ -30,43 +30,43 @@ from rhodecode.model import BaseModel from rhodecode.model.caching_query import FromCache from rhodecode.model.db import UsersGroupMember, UsersGroup -log = logging.getLogger( __name__ ) +log = logging.getLogger(__name__) -class UsersGroupModel( BaseModel ): +class UsersGroupModel(BaseModel): - def get( self, users_group_id, cache = False ): + def get(self, users_group_id, cache = False): users_group = UsersGroup.query() if cache: - users_group = users_group.options( FromCache( "sql_cache_short", - "get_users_group_%s" % users_group_id ) ) - return users_group.get( users_group_id ) + users_group = users_group.options(FromCache("sql_cache_short", + "get_users_group_%s" % users_group_id)) + return users_group.get(users_group_id) - def get_by_name( self, name, cache = False, case_insensitive = False ): + def get_by_name(self, name, cache = False, case_insensitive = False): users_group = UsersGroup.query() if case_insensitive: - users_group = users_group.filter( UsersGroup.users_group_name.ilike( name ) ) + users_group = users_group.filter(UsersGroup.users_group_name.ilike(name)) else: - users_group = users_group.filter( UsersGroup.users_group_name == name ) + users_group = users_group.filter(UsersGroup.users_group_name == name) if cache: - users_group = users_group.options( FromCache( "sql_cache_short", - "get_users_group_%s" % name ) ) + users_group = users_group.options(FromCache("sql_cache_short", + "get_users_group_%s" % name)) return users_group.scalar() - def create( self, form_data ): + def create(self, form_data): try: new_users_group = UsersGroup() for k, v in form_data.items(): - setattr( new_users_group, k, v ) + setattr(new_users_group, k, v) - self.sa.add( new_users_group ) + self.sa.add(new_users_group) self.sa.commit() return new_users_group except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) self.sa.rollback() raise - def add_user_to_group( self, users_group, user ): + def add_user_to_group(self, users_group, user): for m in users_group.members: u = m.user if u.user_id == user.user_id: @@ -77,13 +77,13 @@ class UsersGroupModel( BaseModel ): users_group_member.user = user users_group_member.users_group = users_group - users_group.members.append( users_group_member ) - user.group_member.append( users_group_member ) + users_group.members.append(users_group_member) + user.group_member.append(users_group_member) - self.sa.add( users_group_member ) + self.sa.add(users_group_member) self.sa.commit() return users_group_member except: - log.error( traceback.format_exc() ) + log.error(traceback.format_exc()) self.sa.rollback() raise |