notifications.py
173 lines
| 6.8 KiB
| text/x-python
|
PythonLexer
r1824 | # -*- coding: utf-8 -*- | |||
""" | ||||
rhodecode.controllers.admin.notifications | ||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ | ||||
notifications controller for RhodeCode | ||||
:created_on: Nov 23, 2010 | ||||
:author: marcink | ||||
:copyright: (C) 2010-2012 Marcin Kuzminski <marcin@python-works.com> | ||||
:license: GPLv3, see COPYING for more details. | ||||
""" | ||||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU General Public License as published by | ||||
# the Free Software Foundation, either version 3 of the License, or | ||||
# (at your option) any later version. | ||||
# | ||||
# 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 General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
r1712 | import logging | |||
r1713 | import traceback | |||
r1712 | ||||
r1791 | from pylons import request | |||
r1713 | from pylons import tmpl_context as c, url | |||
r3428 | from pylons.controllers.util import redirect, abort | |||
r1712 | ||||
r2190 | from webhelpers.paginate import Page | |||
r1712 | from rhodecode.lib.base import BaseController, render | |||
from rhodecode.model.db import Notification | ||||
from rhodecode.model.notification import NotificationModel | ||||
r1739 | from rhodecode.lib.auth import LoginRequired, NotAnonymous | |||
r1712 | from rhodecode.lib import helpers as h | |||
r1713 | from rhodecode.model.meta import Session | |||
r2845 | from rhodecode.lib.utils2 import safe_int | |||
r1723 | ||||
r1712 | ||||
log = logging.getLogger(__name__) | ||||
r1791 | ||||
r1712 | class NotificationsController(BaseController): | |||
"""REST Controller styled on the Atom Publishing Protocol""" | ||||
# To properly map this controller, ensure your config/routing.py | ||||
# file has a resource setup: | ||||
r1818 | # map.resource('notification', 'notifications', controller='_admin/notifications', | |||
r1712 | # path_prefix='/_admin', name_prefix='_admin_') | |||
@LoginRequired() | ||||
r1739 | @NotAnonymous() | |||
r1712 | def __before__(self): | |||
super(NotificationsController, self).__before__() | ||||
def index(self, format='html'): | ||||
"""GET /_admin/notifications: All items in the collection""" | ||||
# url('notifications') | ||||
c.user = self.rhodecode_user | ||||
r2433 | notif = NotificationModel().get_for_user(self.rhodecode_user.user_id, | |||
r2503 | filter_=request.GET.getall('type')) | |||
r2845 | ||||
r3745 | p = safe_int(request.GET.get('page', 1), 1) | |||
r2190 | c.notifications = Page(notif, page=p, items_per_page=10) | |||
r2433 | c.pull_request_type = Notification.TYPE_PULL_REQUEST | |||
r2503 | c.comment_type = [Notification.TYPE_CHANGESET_COMMENT, | |||
Notification.TYPE_PULL_REQUEST_COMMENT] | ||||
r2519 | ||||
_current_filter = request.GET.getall('type') | ||||
c.current_filter = 'all' | ||||
if _current_filter == [c.pull_request_type]: | ||||
c.current_filter = 'pull_request' | ||||
elif _current_filter == c.comment_type: | ||||
c.current_filter = 'comment' | ||||
r1712 | return render('admin/notifications/notifications.html') | |||
r1791 | def mark_all_read(self): | |||
if request.environ.get('HTTP_X_PARTIAL_XHR'): | ||||
nm = NotificationModel() | ||||
# mark all read | ||||
r2433 | nm.mark_all_read_for_user(self.rhodecode_user.user_id, | |||
r2503 | filter_=request.GET.getall('type')) | |||
r2662 | Session().commit() | |||
r1791 | c.user = self.rhodecode_user | |||
r2433 | notif = nm.get_for_user(self.rhodecode_user.user_id, | |||
r2503 | filter_=request.GET.getall('type')) | |||
r2190 | c.notifications = Page(notif, page=1, items_per_page=10) | |||
r1791 | return render('admin/notifications/notifications_data.html') | |||
r1712 | def create(self): | |||
"""POST /_admin/notifications: Create a new item""" | ||||
# url('notifications') | ||||
def new(self, format='html'): | ||||
"""GET /_admin/notifications/new: Form to create a new item""" | ||||
# url('new_notification') | ||||
def update(self, notification_id): | ||||
"""PUT /_admin/notifications/id: Update an existing item""" | ||||
# Forms posted to this method should contain a hidden field: | ||||
# <input type="hidden" name="_method" value="PUT" /> | ||||
# Or using helpers: | ||||
# h.form(url('notification', notification_id=ID), | ||||
# method='put') | ||||
# url('notification', notification_id=ID) | ||||
r2610 | try: | |||
no = Notification.get(notification_id) | ||||
|
r3141 | owner = all(un.user.user_id == c.rhodecode_user.user_id | ||
for un in no.notifications_to_users) | ||||
r2610 | if h.HasPermissionAny('hg.admin')() or owner: | |||
NotificationModel().mark_read(c.rhodecode_user.user_id, no) | ||||
r2662 | Session().commit() | |||
r2610 | return 'ok' | |||
except Exception: | ||||
r3428 | Session().rollback() | |||
r2610 | log.error(traceback.format_exc()) | |||
return 'fail' | ||||
r1712 | ||||
def delete(self, notification_id): | ||||
"""DELETE /_admin/notifications/id: Delete an existing item""" | ||||
# Forms posted to this method should contain a hidden field: | ||||
# <input type="hidden" name="_method" value="DELETE" /> | ||||
# Or using helpers: | ||||
# h.form(url('notification', notification_id=ID), | ||||
# method='delete') | ||||
# url('notification', notification_id=ID) | ||||
r1713 | try: | |||
no = Notification.get(notification_id) | ||||
|
r3141 | owner = all(un.user.user_id == c.rhodecode_user.user_id | ||
for un in no.notifications_to_users) | ||||
r2610 | if h.HasPermissionAny('hg.admin')() or owner: | |||
r1713 | NotificationModel().delete(c.rhodecode_user.user_id, no) | |||
r2662 | Session().commit() | |||
r1713 | return 'ok' | |||
except Exception: | ||||
r3428 | Session().rollback() | |||
r1713 | log.error(traceback.format_exc()) | |||
r1712 | return 'fail' | |||
def show(self, notification_id, format='html'): | ||||
"""GET /_admin/notifications/id: Show a specific item""" | ||||
# url('notification', notification_id=ID) | ||||
c.user = self.rhodecode_user | ||||
r1713 | no = Notification.get(notification_id) | |||
r3428 | owner = any(un.user.user_id == c.rhodecode_user.user_id | |||
|
r3141 | for un in no.notifications_to_users) | ||
r3428 | ||||
r1713 | if no and (h.HasPermissionAny('hg.admin', 'repository.admin')() or owner): | |||
unotification = NotificationModel()\ | ||||
.get_user_notification(c.user.user_id, no) | ||||
r1712 | ||||
r1713 | # if this association to user is not valid, we don't want to show | |||
# this message | ||||
if unotification: | ||||
|
r3625 | if not unotification.read: | ||
r1713 | unotification.mark_as_read() | |||
r2662 | Session().commit() | |||
r1713 | c.notification = no | |||
r1712 | ||||
r1713 | return render('admin/notifications/show_notification.html') | |||
r1712 | ||||
r3428 | return abort(403) | |||
r1712 | ||||
def edit(self, notification_id, format='html'): | ||||
"""GET /_admin/notifications/id/edit: Form to edit an existing item""" | ||||
# url('edit_notification', notification_id=ID) | ||||