view kallithea/controllers/admin/notifications.py @ 6452:3dcf1f82311a

controllers: avoid setting request state in controller instances - set it in the thread global request variable In TurboGears, controllers are singletons and we should avoid using instance variables for any volatile data. Instead, use the "global thread local" request context. With everything in request, some use of c is dropped. Note: kallithea/controllers/api/__init__.py still use instance variables that will cause problems with TurboGears.
author Mads Kiilerich <mads@kiilerich.com>
date Sat, 24 Dec 2016 01:27:47 +0100
parents e33b17db388d
children e9ac5698281d
line wrap: on
line source

# -*- coding: utf-8 -*-
# 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/>.
"""
kallithea.controllers.admin.notifications
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

notifications controller for Kallithea

This file was forked by the Kallithea project in July 2014.
Original author and date, and relevant copyright and licensing information is below:
:created_on: Nov 23, 2010
:author: marcink
:copyright: (c) 2013 RhodeCode GmbH, and others.
:license: GPLv3, see LICENSE.md for more details.
"""

import logging
import traceback

from pylons import request
from pylons import tmpl_context as c
from webob.exc import HTTPBadRequest, HTTPForbidden

from kallithea.model.db import Notification
from kallithea.model.notification import NotificationModel
from kallithea.model.meta import Session
from kallithea.lib.auth import LoginRequired, NotAnonymous
from kallithea.lib.base import BaseController, render
from kallithea.lib import helpers as h
from kallithea.lib.page import Page
from kallithea.lib.utils2 import safe_int


log = logging.getLogger(__name__)


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:
    #     map.resource('notification', 'notifications', controller='_admin/notifications',
    #         path_prefix='/_admin', name_prefix='_admin_')

    @LoginRequired()
    @NotAnonymous()
    def __before__(self):
        super(NotificationsController, self).__before__()

    def index(self, format='html'):
        c.user = request.authuser
        notif = NotificationModel().query_for_user(request.authuser.user_id,
                                            filter_=request.GET.getall('type'))

        p = safe_int(request.GET.get('page'), 1)
        c.notifications = Page(notif, page=p, items_per_page=10)
        c.pull_request_type = Notification.TYPE_PULL_REQUEST
        c.comment_type = [Notification.TYPE_CHANGESET_COMMENT,
                          Notification.TYPE_PULL_REQUEST_COMMENT]

        _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'

        return render('admin/notifications/notifications.html')

    def mark_all_read(self):
        if request.environ.get('HTTP_X_PARTIAL_XHR'):
            nm = NotificationModel()
            # mark all read
            nm.mark_all_read_for_user(request.authuser.user_id,
                                      filter_=request.GET.getall('type'))
            Session().commit()
            c.user = request.authuser
            notif = nm.query_for_user(request.authuser.user_id,
                                      filter_=request.GET.getall('type'))
            c.notifications = Page(notif, page=1, items_per_page=10)
            return render('admin/notifications/notifications_data.html')

    def update(self, notification_id):
        try:
            no = Notification.get(notification_id)
            owner = all(un.user_id == request.authuser.user_id
                        for un in no.notifications_to_users)
            if h.HasPermissionAny('hg.admin')() or owner:
                # deletes only notification2user
                NotificationModel().mark_read(request.authuser.user_id, no)
                Session().commit()
                return 'ok'
        except Exception:
            Session().rollback()
            log.error(traceback.format_exc())
        raise HTTPBadRequest()

    def delete(self, notification_id):
        try:
            no = Notification.get(notification_id)
            owner = any(un.user_id == request.authuser.user_id
                        for un in no.notifications_to_users)
            if h.HasPermissionAny('hg.admin')() or owner:
                # deletes only notification2user
                NotificationModel().delete(request.authuser.user_id, no)
                Session().commit()
                return 'ok'
        except Exception:
            Session().rollback()
            log.error(traceback.format_exc())
        raise HTTPBadRequest()

    def show(self, notification_id, format='html'):
        notification = Notification.get_or_404(notification_id)

        unotification = NotificationModel() \
            .get_user_notification(request.authuser.user_id, notification)

        # if this association to user is not valid, we don't want to show
        # this message
        if unotification is None:
            raise HTTPForbidden()

        if not unotification.read:
            unotification.mark_as_read()
            Session().commit()

        c.notification = notification
        c.user = request.authuser
        return render('admin/notifications/show_notification.html')