tracim_v2/tracim/views/contents_api/threads_controller.py

# coding=utf-8
import typing

import transaction
from pyramid.config import Configurator
from tracim.models.data import UserRoleInWorkspace

try:  # Python 3.5+
    from http import HTTPStatus
except ImportError:
    from http import client as HTTPStatus

from tracim import TracimRequest
from tracim.extensions import hapic
from tracim.lib.core.content import ContentApi
from tracim.views.controllers import Controller
from tracim.views.core_api.schemas import ThreadContentSchema
from tracim.views.core_api.schemas import ThreadRevisionSchema
from tracim.views.core_api.schemas import SetContentStatusSchema
from tracim.views.core_api.schemas import ThreadModifySchema
from tracim.views.core_api.schemas import WorkspaceAndContentIdPathSchema
from tracim.views.core_api.schemas import NoContentSchema
from tracim.lib.utils.authorization import require_content_types
from tracim.lib.utils.authorization import require_workspace_role
from tracim.exceptions import WorkspaceNotFound, ContentTypeNotAllowed
from tracim.exceptions import InsufficientUserWorkspaceRole
from tracim.exceptions import NotAuthenticated
from tracim.exceptions import AuthenticationFailed
from tracim.models.context_models import ContentInContext, RevisionInContext
from tracim.models.contents import ContentTypeLegacy as ContentType
from tracim.models.contents import thread_type
from tracim.models.revision_protection import new_revision

THREAD_ENDPOINTS_TAG = 'Threads'


class ThreadController(Controller):

    @hapic.with_api_doc(tags=[THREAD_ENDPOINTS_TAG])
    @hapic.handle_exception(NotAuthenticated, HTTPStatus.UNAUTHORIZED)
    @hapic.handle_exception(InsufficientUserWorkspaceRole, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(WorkspaceNotFound, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(AuthenticationFailed, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(ContentTypeNotAllowed, HTTPStatus.BAD_REQUEST)
    @require_workspace_role(UserRoleInWorkspace.READER)
    @require_content_types([thread_type])
    @hapic.input_path(WorkspaceAndContentIdPathSchema())
    @hapic.output_body(ThreadContentSchema())
    def get_thread(self, context, request: TracimRequest, hapic_data=None) -> ContentInContext:  # nopep8
        """
        Get thread content
        """
        app_config = request.registry.settings['CFG']
        api = ContentApi(
            current_user=request.current_user,
            session=request.dbsession,
            config=app_config,
        )
        content = api.get_one(
            hapic_data.path.content_id,
            content_type=ContentType.Any
        )
        return api.get_content_in_context(content)

    @hapic.with_api_doc(tags=[THREAD_ENDPOINTS_TAG])
    @hapic.handle_exception(NotAuthenticated, HTTPStatus.UNAUTHORIZED)
    @hapic.handle_exception(InsufficientUserWorkspaceRole, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(WorkspaceNotFound, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(AuthenticationFailed, HTTPStatus.FORBIDDEN)
    @require_workspace_role(UserRoleInWorkspace.CONTRIBUTOR)
    @require_content_types([thread_type])
    @hapic.input_path(WorkspaceAndContentIdPathSchema())
    @hapic.input_body(ThreadModifySchema())
    @hapic.output_body(ThreadContentSchema())
    def update_thread(self, context, request: TracimRequest, hapic_data=None) -> ContentInContext:  # nopep8
        """
        update thread
        """
        app_config = request.registry.settings['CFG']
        api = ContentApi(
            current_user=request.current_user,
            session=request.dbsession,
            config=app_config,
        )
        content = api.get_one(
            hapic_data.path.content_id,
            content_type=ContentType.Any
        )
        with new_revision(
                session=request.dbsession,
                tm=transaction.manager,
                content=content
        ):
            api.update_content(
                item=content,
                new_label=hapic_data.body.label,
                new_content=hapic_data.body.raw_content,

            )
            api.save(content)
        return api.get_content_in_context(content)

    @hapic.with_api_doc(tags=[THREAD_ENDPOINTS_TAG])
    @hapic.handle_exception(NotAuthenticated, HTTPStatus.UNAUTHORIZED)
    @hapic.handle_exception(InsufficientUserWorkspaceRole, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(WorkspaceNotFound, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(AuthenticationFailed, HTTPStatus.FORBIDDEN)
    @require_workspace_role(UserRoleInWorkspace.READER)
    @require_content_types([thread_type])
    @hapic.input_path(WorkspaceAndContentIdPathSchema())
    @hapic.output_body(ThreadRevisionSchema(many=True))
    def get_thread_revisions(
            self,
            context,
            request: TracimRequest,
            hapic_data=None
    ) -> typing.List[RevisionInContext]:
        """
        get thread revisions
        """
        app_config = request.registry.settings['CFG']
        api = ContentApi(
            current_user=request.current_user,
            session=request.dbsession,
            config=app_config,
        )
        content = api.get_one(
            hapic_data.path.content_id,
            content_type=ContentType.Any
        )
        revisions = content.revisions
        return [
            api.get_revision_in_context(revision)
            for revision in revisions
        ]

    @hapic.with_api_doc(tags=[THREAD_ENDPOINTS_TAG])
    @hapic.handle_exception(NotAuthenticated, HTTPStatus.UNAUTHORIZED)
    @hapic.handle_exception(InsufficientUserWorkspaceRole, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(WorkspaceNotFound, HTTPStatus.FORBIDDEN)
    @hapic.handle_exception(AuthenticationFailed, HTTPStatus.FORBIDDEN)
    @require_workspace_role(UserRoleInWorkspace.CONTRIBUTOR)
    @require_content_types([thread_type])
    @hapic.input_path(WorkspaceAndContentIdPathSchema())
    @hapic.input_body(SetContentStatusSchema())
    @hapic.output_body(NoContentSchema(), default_http_code=HTTPStatus.NO_CONTENT)  # nopep8
    def set_thread_status(self, context, request: TracimRequest, hapic_data=None) -> None:  # nopep8
        """
        set thread status
        """
        app_config = request.registry.settings['CFG']
        api = ContentApi(
            current_user=request.current_user,
            session=request.dbsession,
            config=app_config,
        )
        content = api.get_one(
            hapic_data.path.content_id,
            content_type=ContentType.Any
        )
        with new_revision(
                session=request.dbsession,
                tm=transaction.manager,
                content=content
        ):
            api.set_status(
                content,
                hapic_data.body.status,
            )
            api.save(content)
        return

    def bind(self, configurator: Configurator) -> None:
        # Get thread
        configurator.add_route(
            'thread',
            '/workspaces/{workspace_id}/threads/{content_id}',
            request_method='GET'
        )
        configurator.add_view(self.get_thread, route_name='thread')  # nopep8

        # update thread
        configurator.add_route(
            'update_thread',
            '/workspaces/{workspace_id}/threads/{content_id}',
            request_method='PUT'
        )  # nopep8
        configurator.add_view(self.update_thread, route_name='update_thread')  # nopep8

        # get thread revisions
        configurator.add_route(
            'thread_revisions',
            '/workspaces/{workspace_id}/threads/{content_id}/revisions',  # nopep8
            request_method='GET'
        )
        configurator.add_view(self.get_thread_revisions, route_name='thread_revisions')  # nopep8

        # get thread revisions
        configurator.add_route(
            'set_thread_status',
            '/workspaces/{workspace_id}/threads/{content_id}/status',  # nopep8
            request_method='PUT'
        )
        configurator.add_view(self.set_thread_status, route_name='set_thread_status')  # nopep8