Source code for flexget.api.core.cached

from __future__ import annotations

from typing import TYPE_CHECKING

from flask.helpers import send_file
from flask_restx import inputs
from requests import RequestException

from flexget.api import APIResource, api
from flexget.api.app import APIError, BadRequest
from flexget.utils.cache import cached_resource

if TYPE_CHECKING:
    from flask import Response
    from sqlalchemy.orm import Session

cached_api = api.namespace('cached', description='Cache remote resources')

cached_parser = api.parser()
cached_parser.add_argument('url', required=True, help='URL to cache')
cached_parser.add_argument(
    'force', type=inputs.boolean, default=False, help='Force fetching remote resource'
)




@cached_api.route('/')
@api.doc(description='Returns a cached copy of the requested resource, matching its mime type')
class CachedResource(APIResource):


    @api.response(200, description='Cached resource')
    @api.response(BadRequest)
    @api.response(APIError)
    @api.doc(expect=[cached_parser])
    def get(self, session: Session = None) -> Response:
        """Cache remote resources."""
        args = cached_parser.parse_args()
        url = args.get('url')
        force = args.get('force')
        try:
            file_path, mime_type = cached_resource(url, self.manager.config_base, force=force)
        except RequestException as e:
            raise BadRequest(f'Request Error: {e.args[0]}')
        except OSError as e:
            raise APIError(f'Error: {e}')
        return send_file(file_path, mimetype=mime_type)