backend/src/appenlight/models/services/event.py

# -*- coding: utf-8 -*-

# Copyright (C) 2010-2016  RhodeCode GmbH
#
# 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
# App Enlight Enterprise Edition, including its added features, Support
# services, and proprietary license terms, please see
# https://rhodecode.com/licenses/

import sqlalchemy as sa
from pyramid.threadlocal import get_current_registry
from paginate_sqlalchemy import SqlalchemyOrmPage
from appenlight.models import get_db_session
from appenlight.models.event import Event
from appenlight.models.services.base import BaseService


class EventService(BaseService):
    @classmethod
    def for_resource(cls, resource_ids, event_type=None, status=None,
                     since_when=None, limit=20, event_id=None,
                     target_uuid=None, order_by=None, or_target_user_id=None,
                     db_session=None):
        """
        Fetches events including based on passed params OR if target_user_id
        is present include events that just target this user
        """
        db_session = get_db_session(db_session)
        query = db_session.query(Event)
        query = query.options(sa.orm.joinedload(Event.resource))
        and_cond = [Event.resource_id.in_(resource_ids)]
        if not resource_ids:
            and_cond = [Event.resource_id == -999]

        if event_type:
            and_cond.append(Event.event_type == event_type)
        if status:
            and_cond.append(Event.status == status)
        if since_when:
            and_cond.append(Event.start_date >= since_when)
        if event_id:
            and_cond.append(Event.id == event_id)
        if target_uuid:
            and_cond.append(Event.target_uuid == target_uuid)

        or_cond = []

        if or_target_user_id:
            or_cond.append(sa.or_(Event.target_user_id == or_target_user_id))

        query = query.filter(sa.or_(sa.and_(*and_cond),
                                    *or_cond))
        if not order_by:
            query = query.order_by(sa.desc(Event.start_date))
        if limit:
            query = query.limit(limit)

        return query

    @classmethod
    def by_type_and_status(cls, event_types, status_types, since_when=None,
                           older_than=None, db_session=None, app_ids=None):
        db_session = get_db_session(db_session)
        query = db_session.query(Event)
        query = query.filter(Event.event_type.in_(event_types))
        query = query.filter(Event.status.in_(status_types))
        if since_when:
            query = query.filter(Event.start_date >= since_when)
        if older_than:
            query = query.filter(Event.start_date <= older_than)
        if app_ids:
            query = query.filter(Event.resource_id.in_(app_ids))
        return query

    @classmethod
    def latest_for_user(cls, user, db_session=None):
        registry = get_current_registry()
        resources = user.resources_with_perms(
            ['view'], resource_types=registry.resource_types)
        resource_ids = [r.resource_id for r in resources]
        db_session = get_db_session(db_session)
        return EventService.for_resource(
            resource_ids, or_target_user_id=user.id, limit=10,
            db_session=db_session)

    @classmethod
    def get_paginator(cls, user, page=1, item_count=None, items_per_page=50,
                      order_by=None, filter_settings=None, db_session=None):
        if not filter_settings:
            filter_settings = {}
        registry = get_current_registry()
        resources = user.resources_with_perms(
            ['view'], resource_types=registry.resource_types)
        resource_ids = [r.resource_id for r in resources]
        query = EventService.for_resource(
            resource_ids, or_target_user_id=user.id, limit=100,
            db_session=db_session)

        paginator = SqlalchemyOrmPage(query, page=page,
                                      items_per_page=items_per_page,
                                      **filter_settings)
        return paginator