tracim_v2/tracim/lib/core/group.py

# -*- coding: utf-8 -*-
import typing

from sqlalchemy.orm.exc import NoResultFound

from tracim.exceptions import GroupNotExist

__author__ = 'damien'

from tracim.models.auth import Group, User
from sqlalchemy.orm import Query
from sqlalchemy.orm import Session


class GroupApi(object):

    def __init__(
            self,
            session: Session,
            current_user: typing.Optional[User],
    ):
        self._user = current_user
        self._session = session

    def _base_query(self) -> Query:
        return self._session.query(Group)

    def get_one(self, group_id) -> Group:
        try:
            group = self._base_query().filter(Group.group_id == group_id).one()
            return group
        except NoResultFound:
            raise GroupNotExist()

    def get_one_with_name(self, group_name) -> Group:
        try:
            group = self._base_query().filter(Group.group_name == group_name).one()
            return group
        except NoResultFound:
            raise GroupNotExist()

    def get_all(self):
        return self._base_query().order_by(Group.group_id).all()