from http import HTTPStatus from sanic import response, Request, Blueprint from sanic_ext import validate from sanic_ext.extensions.openapi import openapi from wrolpi.api_utils import json_response from wrolpi.common import logger, wrol_mode_check, api_param_limiter from wrolpi.errors import ValidationError from wrolpi.schema import JSONErrorResponse from . import schema from .lib import get_statistics, _doc_response, _get_doc, _delete_docs, _search_docs NAME = 'docs' docs_bp = Blueprint('Docs', '/api/docs') logger = logger.getChild(__name__) @docs_bp.get('/statistics') @openapi.response(HTTPStatus.OK, schema.DocStatisticsResponse) async def statistics(_: Request): ret = get_statistics() return json_response(ret, HTTPStatus.OK) @docs_bp.get('/') @openapi.description('Get a doc by FileGroup ID') @openapi.response(HTTPStatus.NOT_FOUND, JSONErrorResponse) async def get_doc(request: Request, file_group_id: int): session = request.ctx.session doc = _get_doc(session, file_group_id) return json_response(_doc_response(doc)) @docs_bp.delete('/', name='doc_delete_one') @docs_bp.delete('/', name='doc_delete_many') @openapi.description('Delete Docs by FileGroup IDs') @openapi.response(HTTPStatus.NOT_FOUND, JSONErrorResponse) @wrol_mode_check async def delete_docs(_: Request, file_group_ids: str): try: file_group_ids = [int(i) for i in str(file_group_ids).split(',')] except ValueError: raise ValidationError('Could not parse file group ids') _delete_docs(*file_group_ids) return response.empty() doc_limit_limiter = api_param_limiter(100) @docs_bp.post('/search') @openapi.definition( summary='Search docs with filtering', body=schema.DocSearchRequest, ) @openapi.response(HTTPStatus.OK, schema.DocSearchResponse) @validate(schema.DocSearchRequest) async def search_docs(_: Request, body: schema.DocSearchRequest): limit = doc_limit_limiter(body.limit) offset = body.offset or 0 file_groups, total = _search_docs( search_str=body.search_str, author=body.author, subject=body.subject, language=body.language, mimetype=body.mimetype, limit=limit, offset=offset, order_by=body.order_by, tag_names=body.tag_names, ) ret = dict(file_groups=file_groups, totals=dict(file_groups=total)) return json_response(ret)