base.py
75 lines
| 2.7 KiB
| text/x-python
|
PythonLexer
r547 | """The base Controller API | |||
Provides the BaseController class for subclassing. | ||||
""" | ||||
from pylons import config, tmpl_context as c, request, session | ||||
from pylons.controllers import WSGIController | ||||
from pylons.templating import render_mako as render | ||||
r1304 | ||||
r547 | from rhodecode import __version__ | |||
r1117 | from rhodecode.lib.auth import AuthUser | |||
r547 | from rhodecode.lib.utils import get_repo_slug | |||
from rhodecode.model import meta | ||||
r691 | from rhodecode.model.scm import ScmModel | |||
r710 | from rhodecode import BACKENDS | |||
r1366 | from rhodecode.model.db import Repository | |||
r665 | ||||
r1307 | ||||
r547 | class BaseController(WSGIController): | |||
r659 | ||||
r547 | def __before__(self): | |||
r548 | c.rhodecode_version = __version__ | |||
r890 | c.rhodecode_name = config.get('rhodecode_title') | |||
r891 | c.ga_code = config.get('rhodecode_ga_code') | |||
r547 | c.repo_name = get_repo_slug(request) | |||
r659 | c.backends = BACKENDS.keys() | |||
r890 | self.cut_off_limit = int(config.get('cut_off_limit')) | |||
r1036 | ||||
r784 | self.sa = meta.Session() | |||
r1045 | self.scm_model = ScmModel(self.sa) | |||
r1366 | ||||
r784 | #c.unread_journal = scm_model.get_unread_journal() | |||
r547 | def __call__(self, environ, start_response): | |||
"""Invoke the Controller""" | ||||
# WSGIController.__call__ dispatches to the Controller method | ||||
# the request is routed to. This routing information is | ||||
# available in environ['pylons.routes_dict'] | ||||
try: | ||||
r1307 | # putting this here makes sure that we update permissions each time | |||
r1117 | api_key = request.GET.get('api_key') | |||
user_id = getattr(session.get('rhodecode_user'), 'user_id', None) | ||||
self.rhodecode_user = c.rhodecode_user = AuthUser(user_id, api_key) | ||||
self.rhodecode_user.set_authenticated( | ||||
getattr(session.get('rhodecode_user'), | ||||
'is_authenticated', False)) | ||||
session['rhodecode_user'] = self.rhodecode_user | ||||
session.save() | ||||
r547 | return WSGIController.__call__(self, environ, start_response) | |||
finally: | ||||
meta.Session.remove() | ||||
r1045 | ||||
class BaseRepoController(BaseController): | ||||
""" | ||||
Base class for controllers responsible for loading all needed data | ||||
for those controllers, loaded items are | ||||
r1203 | ||||
r1045 | c.rhodecode_repo: instance of scm repository (taken from cache) | |||
r1203 | ||||
r1045 | """ | |||
def __before__(self): | ||||
super(BaseRepoController, self).__before__() | ||||
if c.repo_name: | ||||
r1366 | c.rhodecode_repo = Repository.by_repo_name(c.repo_name).scm_instance | |||
r1282 | ||||
r1340 | if c.rhodecode_repo is not None: | |||
r1307 | c.repository_followers = \ | |||
self.scm_model.get_followers(c.repo_name) | ||||
r1045 | c.repository_forks = self.scm_model.get_forks(c.repo_name) | |||
else: | ||||
c.repository_followers = 0 | ||||
c.repository_forks = 0 | ||||
r1304 | ||||