Added associated classes into child models

rhodecode/model/changeset_status.py

@@ -34,6 +34,8 b' log = logging.getLogger(__name__)'
 
 class ChangesetStatusModel(BaseModel):
 
+    cls = ChangesetStatus
+
     def __get_changeset_status(self, changeset_status):
         return self._get_instance(ChangesetStatus, changeset_status)
 

rhodecode/model/comment.py

@@ -41,6 +41,8 b' log = logging.getLogger(__name__)'
 
 class ChangesetCommentsModel(BaseModel):
 
+    cls = ChangesetComment
+
     def __get_changeset_comment(self, changeset_comment):
         return self._get_instance(ChangesetComment, changeset_comment)
 

rhodecode/model/notification.py

@@ -40,6 +40,8 b' log = logging.getLogger(__name__)'
 
 class NotificationModel(BaseModel):
 
+    cls = Notification
+
     def __get_notification(self, notification):
         if isinstance(notification, Notification):
             return notification

rhodecode/model/permission.py

@@ -42,6 +42,8 b' class PermissionModel(BaseModel):'
     Permissions model for RhodeCode
     """
 
+    cls = Permission
+
     def get_permission(self, permission_id, cache=False):
         """
         Get's permissions by id

rhodecode/model/pull_request.py

@@ -40,6 +40,8 b' log = logging.getLogger(__name__)'
 
 class PullRequestModel(BaseModel):
 
+    cls = PullRequest
+
     def get_all(self, repo):
         repo = self._get_repo(repo)
         return PullRequest.query().filter(PullRequest.other_repo == repo).all()

rhodecode/model/repo.py

@@ -48,6 +48,8 b' log = logging.getLogger(__name__)'
 
 class RepoModel(BaseModel):
 
+    cls = Repository
+
     def __get_users_group(self, users_group):
         return self._get_instance(UsersGroup, users_group,
                                   callback=UsersGroup.get_by_group_name)

rhodecode/model/repo_permission.py

@@ -34,6 +34,8 b' log = logging.getLogger(__name__)'
 
 class RepositoryPermissionModel(BaseModel):
 
+    cls = UserRepoToPerm
+
     def get_user_permission(self, repository, user):
         repository = self._get_repo(repository)
         user = self._get_user(user)

rhodecode/model/repos_group.py

@@ -39,6 +39,8 b' log = logging.getLogger(__name__)'
 
 class ReposGroupModel(BaseModel):
 
+    cls = RepoGroup
+
     def __get_users_group(self, users_group):
         return self._get_instance(UsersGroup, users_group,
                                   callback=UsersGroup.get_by_group_name)

rhodecode/model/user.py

@@ -59,6 +59,7 b' PERM_WEIGHTS = {'
 
 
 class UserModel(BaseModel):
+    cls = User
 
     def get(self, user_id, cache=False):
         user = self.sa.query(User)
@@ -82,6 +83,9 b' class UserModel(BaseModel):'
                                           "get_user_%s" % username))
         return user.scalar()
 
+    def get_by_email(self, email, cache=False, case_insensitive=False):
+        return User.get_by_email(email, case_insensitive, cache)
+
     def get_by_api_key(self, api_key, cache=False):
         return User.get_by_api_key(api_key, cache)
 

rhodecode/model/users_group.py

@@ -44,6 +44,9 b' class UsersGroupModel(BaseModel):'
     def get(self, users_group_id, cache=False):
         return UsersGroup.get(users_group_id)
 
+    def get_group(self, users_group):
+        return self.__get_users_group(users_group)
+
     def get_by_name(self, name, cache=False, case_insensitive=False):
         return UsersGroup.get_by_group_name(name, cache, case_insensitive)