kallithea Changeset - e4d2271e1193

Changeset - e4d2271e1193

Parent rev.

Child rev.

[Not reviewed]

beta

0 1 0

Marcin Kuzminski - 13 years ago 2012-07-01 12:40:00
marcin@python-works.com

Added get_all method to BaseModel, requires to define cls variable in child models

1 file changed with 12 insertions and 2 deletions:

rhodecode/model/__init__.py

0 comments (0 inline, 0 general)

rhodecode/model/__init__.py

➞

Show inline comments

@@ @@ -39,13 +39,12 @@ @@
 # GNU General Public License for more details.
+#
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 import logging
 from rhodecode.model.db import User, Repository, Permission
 from rhodecode.model import meta
 log = logging.getLogger(__name__)
 def init_model(engine):
@@ @@ -65,17 +64,19 @@ class BaseModel(object): @@
     Base Model for all RhodeCode models, it adds sql alchemy session
     into instance of model
     :param sa: If passed it reuses this session instead of creating a new one
     """
     cls = None  # override in child class
     def __init__(self, sa=None):
         if sa is not None:
             self.sa = sa
         else:
             self.sa = meta.Session
+            self.sa = meta.Session()
     def _get_instance(self, cls, instance, callback=None):
         """
         Get's instance of given cls using some simple lookup mechanism.
         :param cls: class to fetch
@@ @@ -101,28 +102,37 @@ class BaseModel(object): @@
         """
         Helper method to get user by ID, or username fallback
         :param user:
         :type user: UserID, username, or User instance
         """
         from rhodecode.model.db import User
         return self._get_instance(User, user,
                                   callback=User.get_by_username)
     def _get_repo(self, repository):
         """
         Helper method to get repository by ID, or repository name
         :param repository:
         :type repository: RepoID, repository name or Repository Instance
         """
         from rhodecode.model.db import Repository
         return self._get_instance(Repository, repository,
                                   callback=Repository.get_by_repo_name)
     def _get_perm(self, permission):
         """
         Helper method to get permission by ID, or permission name
         :param permission:
         :type permission: PermissionID, permission_name or Permission instance
         """
         from rhodecode.model.db import Permission
         return self._get_instance(Permission, permission,
                                   callback=Permission.get_by_key)
     def get_all(self):
         """
         Returns all instances of what is defined in `cls` class variable
         """
         return self.cls.getAll()

0 comments (0 inline, 0 general)