Show More
views.py
386 lines
| 14.2 KiB
| text/x-python
|
PythonLexer
r1933 | # -*- coding: utf-8 -*- | |||
r2487 | # Copyright (C) 2010-2018 RhodeCode GmbH | |||
r1933 | # | |||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU Affero General Public License, version 3 | ||||
# (only), as published by the Free Software Foundation. | ||||
# | ||||
# This program is distributed in the hope that it will be useful, | ||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
# | ||||
# You should have received a copy of the GNU Affero General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
# | ||||
# This program is dual-licensed. If you wish to learn more about the | ||||
# RhodeCode Enterprise Edition, including its added features, Support services, | ||||
# and proprietary license terms, please see https://rhodecode.com/licenses/ | ||||
import logging | ||||
import itertools | ||||
from webhelpers.feedgenerator import Atom1Feed, Rss201rev2Feed | ||||
from pyramid.view import view_config | ||||
from pyramid.httpexceptions import HTTPBadRequest | ||||
from pyramid.response import Response | ||||
from pyramid.renderers import render | ||||
from rhodecode.apps._base import BaseAppView | ||||
from rhodecode.model.db import ( | ||||
or_, joinedload, UserLog, UserFollowing, User, UserApiKeys) | ||||
from rhodecode.model.meta import Session | ||||
import rhodecode.lib.helpers as h | ||||
from rhodecode.lib.helpers import Page | ||||
from rhodecode.lib.user_log_filter import user_log_filter | ||||
from rhodecode.lib.auth import LoginRequired, NotAnonymous, CSRFRequired | ||||
r2071 | from rhodecode.lib.utils2 import safe_int, AttributeDict, md5_safe | |||
r1933 | from rhodecode.model.scm import ScmModel | |||
log = logging.getLogger(__name__) | ||||
class JournalView(BaseAppView): | ||||
def load_default_context(self): | ||||
c = self._get_local_tmpl_context(include_app_defaults=True) | ||||
r2351 | ||||
r1933 | self._load_defaults(c.rhodecode_name) | |||
# TODO(marcink): what is this, why we need a global register ? | ||||
c.search_term = self.request.GET.get('filter') or '' | ||||
return c | ||||
def _get_config(self, rhodecode_name): | ||||
import rhodecode | ||||
config = rhodecode.CONFIG | ||||
return { | ||||
'language': 'en-us', | ||||
'feed_ttl': '5', # TTL of feed, | ||||
'feed_items_per_page': | ||||
safe_int(config.get('rss_items_per_page', 20)), | ||||
'rhodecode_name': rhodecode_name | ||||
} | ||||
def _load_defaults(self, rhodecode_name): | ||||
config = self._get_config(rhodecode_name) | ||||
# common values for feeds | ||||
self.language = config["language"] | ||||
self.ttl = config["feed_ttl"] | ||||
self.feed_items_per_page = config['feed_items_per_page'] | ||||
self.rhodecode_name = config['rhodecode_name'] | ||||
def _get_daily_aggregate(self, journal): | ||||
groups = [] | ||||
for k, g in itertools.groupby(journal, lambda x: x.action_as_day): | ||||
user_group = [] | ||||
# groupby username if it's a present value, else | ||||
# fallback to journal username | ||||
for _, g2 in itertools.groupby( | ||||
list(g), lambda x: x.user.username if x.user else x.username): | ||||
l = list(g2) | ||||
user_group.append((l[0].user, l)) | ||||
groups.append((k, user_group,)) | ||||
return groups | ||||
def _get_journal_data(self, following_repos, search_term): | ||||
repo_ids = [x.follows_repository.repo_id for x in following_repos | ||||
if x.follows_repository is not None] | ||||
user_ids = [x.follows_user.user_id for x in following_repos | ||||
if x.follows_user is not None] | ||||
filtering_criterion = None | ||||
if repo_ids and user_ids: | ||||
filtering_criterion = or_(UserLog.repository_id.in_(repo_ids), | ||||
UserLog.user_id.in_(user_ids)) | ||||
if repo_ids and not user_ids: | ||||
filtering_criterion = UserLog.repository_id.in_(repo_ids) | ||||
if not repo_ids and user_ids: | ||||
filtering_criterion = UserLog.user_id.in_(user_ids) | ||||
if filtering_criterion is not None: | ||||
journal = Session().query(UserLog)\ | ||||
.options(joinedload(UserLog.user))\ | ||||
.options(joinedload(UserLog.repository)) | ||||
# filter | ||||
try: | ||||
journal = user_log_filter(journal, search_term) | ||||
except Exception: | ||||
# we want this to crash for now | ||||
raise | ||||
journal = journal.filter(filtering_criterion)\ | ||||
.order_by(UserLog.action_date.desc()) | ||||
else: | ||||
journal = [] | ||||
return journal | ||||
r2071 | def feed_uid(self, entry_id): | |||
return '{}:{}'.format('journal', md5_safe(entry_id)) | ||||
r1933 | def _atom_feed(self, repos, search_term, public=True): | |||
_ = self.request.translate | ||||
journal = self._get_journal_data(repos, search_term) | ||||
if public: | ||||
_link = h.route_url('journal_public_atom') | ||||
_desc = '%s %s %s' % (self.rhodecode_name, _('public journal'), | ||||
'atom feed') | ||||
else: | ||||
_link = h.route_url('journal_atom') | ||||
_desc = '%s %s %s' % (self.rhodecode_name, _('journal'), 'atom feed') | ||||
feed = Atom1Feed( | ||||
title=_desc, link=_link, description=_desc, | ||||
language=self.language, ttl=self.ttl) | ||||
for entry in journal[:self.feed_items_per_page]: | ||||
user = entry.user | ||||
if user is None: | ||||
# fix deleted users | ||||
user = AttributeDict({'short_contact': entry.username, | ||||
'email': '', | ||||
'full_contact': ''}) | ||||
r2351 | action, action_extra, ico = h.action_parser( | |||
self.request, entry, feed=True) | ||||
r1933 | title = "%s - %s %s" % (user.short_contact, action(), | |||
entry.repository.repo_name) | ||||
desc = action_extra() | ||||
_url = h.route_url('home') | ||||
if entry.repository is not None: | ||||
_url = h.route_url('repo_changelog', | ||||
repo_name=entry.repository.repo_name) | ||||
r2071 | feed.add_item( | |||
unique_id=self.feed_uid(entry.user_log_id), | ||||
title=title, | ||||
pubdate=entry.action_date, | ||||
link=_url, | ||||
author_email=user.email, | ||||
author_name=user.full_contact, | ||||
description=desc) | ||||
r1933 | ||||
response = Response(feed.writeString('utf-8')) | ||||
response.content_type = feed.mime_type | ||||
return response | ||||
def _rss_feed(self, repos, search_term, public=True): | ||||
_ = self.request.translate | ||||
journal = self._get_journal_data(repos, search_term) | ||||
if public: | ||||
_link = h.route_url('journal_public_atom') | ||||
_desc = '%s %s %s' % ( | ||||
self.rhodecode_name, _('public journal'), 'rss feed') | ||||
else: | ||||
_link = h.route_url('journal_atom') | ||||
_desc = '%s %s %s' % ( | ||||
self.rhodecode_name, _('journal'), 'rss feed') | ||||
feed = Rss201rev2Feed( | ||||
title=_desc, link=_link, description=_desc, | ||||
language=self.language, ttl=self.ttl) | ||||
for entry in journal[:self.feed_items_per_page]: | ||||
user = entry.user | ||||
if user is None: | ||||
# fix deleted users | ||||
user = AttributeDict({'short_contact': entry.username, | ||||
'email': '', | ||||
'full_contact': ''}) | ||||
r2351 | action, action_extra, ico = h.action_parser( | |||
self.request, entry, feed=True) | ||||
r1933 | title = "%s - %s %s" % (user.short_contact, action(), | |||
entry.repository.repo_name) | ||||
desc = action_extra() | ||||
_url = h.route_url('home') | ||||
if entry.repository is not None: | ||||
_url = h.route_url('repo_changelog', | ||||
repo_name=entry.repository.repo_name) | ||||
r2071 | feed.add_item( | |||
unique_id=self.feed_uid(entry.user_log_id), | ||||
title=title, | ||||
pubdate=entry.action_date, | ||||
link=_url, | ||||
author_email=user.email, | ||||
author_name=user.full_contact, | ||||
description=desc) | ||||
r1933 | ||||
response = Response(feed.writeString('utf-8')) | ||||
response.content_type = feed.mime_type | ||||
return response | ||||
@LoginRequired() | ||||
@NotAnonymous() | ||||
@view_config( | ||||
route_name='journal', request_method='GET', | ||||
renderer=None) | ||||
def journal(self): | ||||
c = self.load_default_context() | ||||
p = safe_int(self.request.GET.get('page', 1), 1) | ||||
c.user = User.get(self._rhodecode_user.user_id) | ||||
following = Session().query(UserFollowing)\ | ||||
.filter(UserFollowing.user_id == self._rhodecode_user.user_id)\ | ||||
.options(joinedload(UserFollowing.follows_repository))\ | ||||
.all() | ||||
journal = self._get_journal_data(following, c.search_term) | ||||
def url_generator(**kw): | ||||
query_params = { | ||||
'filter': c.search_term | ||||
} | ||||
query_params.update(kw) | ||||
return self.request.current_route_path(_query=query_params) | ||||
c.journal_pager = Page( | ||||
journal, page=p, items_per_page=20, url=url_generator) | ||||
c.journal_day_aggreagate = self._get_daily_aggregate(c.journal_pager) | ||||
c.journal_data = render( | ||||
'rhodecode:templates/journal/journal_data.mako', | ||||
self._get_template_context(c), self.request) | ||||
if self.request.is_xhr: | ||||
return Response(c.journal_data) | ||||
html = render( | ||||
'rhodecode:templates/journal/journal.mako', | ||||
self._get_template_context(c), self.request) | ||||
return Response(html) | ||||
@LoginRequired(auth_token_access=[UserApiKeys.ROLE_FEED]) | ||||
@NotAnonymous() | ||||
@view_config( | ||||
route_name='journal_atom', request_method='GET', | ||||
renderer=None) | ||||
def journal_atom(self): | ||||
""" | ||||
Produce an atom-1.0 feed via feedgenerator module | ||||
""" | ||||
c = self.load_default_context() | ||||
following_repos = Session().query(UserFollowing)\ | ||||
.filter(UserFollowing.user_id == self._rhodecode_user.user_id)\ | ||||
.options(joinedload(UserFollowing.follows_repository))\ | ||||
.all() | ||||
return self._atom_feed(following_repos, c.search_term, public=False) | ||||
@LoginRequired(auth_token_access=[UserApiKeys.ROLE_FEED]) | ||||
@NotAnonymous() | ||||
@view_config( | ||||
route_name='journal_rss', request_method='GET', | ||||
renderer=None) | ||||
def journal_rss(self): | ||||
""" | ||||
Produce an rss feed via feedgenerator module | ||||
""" | ||||
c = self.load_default_context() | ||||
following_repos = Session().query(UserFollowing)\ | ||||
.filter(UserFollowing.user_id == self._rhodecode_user.user_id)\ | ||||
.options(joinedload(UserFollowing.follows_repository))\ | ||||
.all() | ||||
return self._rss_feed(following_repos, c.search_term, public=False) | ||||
@LoginRequired() | ||||
@NotAnonymous() | ||||
@CSRFRequired() | ||||
@view_config( | ||||
route_name='toggle_following', request_method='POST', | ||||
renderer='json_ext') | ||||
def toggle_following(self): | ||||
user_id = self.request.POST.get('follows_user_id') | ||||
if user_id: | ||||
try: | ||||
ScmModel().toggle_following_user( | ||||
user_id, self._rhodecode_user.user_id) | ||||
Session().commit() | ||||
return 'ok' | ||||
except Exception: | ||||
raise HTTPBadRequest() | ||||
repo_id = self.request.POST.get('follows_repo_id') | ||||
if repo_id: | ||||
try: | ||||
ScmModel().toggle_following_repo( | ||||
repo_id, self._rhodecode_user.user_id) | ||||
Session().commit() | ||||
return 'ok' | ||||
except Exception: | ||||
raise HTTPBadRequest() | ||||
raise HTTPBadRequest() | ||||
@LoginRequired() | ||||
@view_config( | ||||
route_name='journal_public', request_method='GET', | ||||
renderer=None) | ||||
def journal_public(self): | ||||
c = self.load_default_context() | ||||
# Return a rendered template | ||||
p = safe_int(self.request.GET.get('page', 1), 1) | ||||
c.following = Session().query(UserFollowing)\ | ||||
.filter(UserFollowing.user_id == self._rhodecode_user.user_id)\ | ||||
.options(joinedload(UserFollowing.follows_repository))\ | ||||
.all() | ||||
journal = self._get_journal_data(c.following, c.search_term) | ||||
def url_generator(**kw): | ||||
query_params = {} | ||||
query_params.update(kw) | ||||
return self.request.current_route_path(_query=query_params) | ||||
c.journal_pager = Page( | ||||
journal, page=p, items_per_page=20, url=url_generator) | ||||
c.journal_day_aggreagate = self._get_daily_aggregate(c.journal_pager) | ||||
c.journal_data = render( | ||||
'rhodecode:templates/journal/journal_data.mako', | ||||
self._get_template_context(c), self.request) | ||||
if self.request.is_xhr: | ||||
return Response(c.journal_data) | ||||
html = render( | ||||
'rhodecode:templates/journal/public_journal.mako', | ||||
self._get_template_context(c), self.request) | ||||
return Response(html) | ||||
@LoginRequired(auth_token_access=[UserApiKeys.ROLE_FEED]) | ||||
@view_config( | ||||
route_name='journal_public_atom', request_method='GET', | ||||
renderer=None) | ||||
def journal_public_atom(self): | ||||
""" | ||||
Produce an atom-1.0 feed via feedgenerator module | ||||
""" | ||||
c = self.load_default_context() | ||||
following_repos = Session().query(UserFollowing)\ | ||||
.filter(UserFollowing.user_id == self._rhodecode_user.user_id)\ | ||||
.options(joinedload(UserFollowing.follows_repository))\ | ||||
.all() | ||||
return self._atom_feed(following_repos, c.search_term) | ||||
@LoginRequired(auth_token_access=[UserApiKeys.ROLE_FEED]) | ||||
@view_config( | ||||
route_name='journal_public_rss', request_method='GET', | ||||
renderer=None) | ||||
def journal_public_rss(self): | ||||
""" | ||||
Produce an rss2 feed via feedgenerator module | ||||
""" | ||||
c = self.load_default_context() | ||||
following_repos = Session().query(UserFollowing)\ | ||||
.filter(UserFollowing.user_id == self._rhodecode_user.user_id)\ | ||||
.options(joinedload(UserFollowing.follows_repository))\ | ||||
.all() | ||||
return self._rss_feed(following_repos, c.search_term) | ||||