app.py

import os
import json

from flasgger import Swagger
from resource.celebrity_resource import Celebrity
from resource.movie_resource import Movie, MovieCelebrityList, MovieList
from resource.general_resource import PhotoList

from flask import Flask
from flask import request
from flask_caching import Cache
from flask_restful import Api
from flask_cors import CORS

from provider.httprequest_provider import HttpRequestProvider

headers = {'User-Agent': 'curl/7.64.1'}
for key in os.environ:
    if key.startswith("REQUEST_HEADERS_"):
        headers[key.replace("REQUEST_HEADERS_","").replace("_", "-")] = os.environ.get(key)
    
provider = HttpRequestProvider(headers)
app = Flask(__name__)
CORS(app)
app.config.update(RESTFUL_JSON=dict(ensure_ascii=False))
cache = Cache(config={'CACHE_TYPE': 'SimpleCache'})
cache.init_app(app, config={'CACHE_TYPE': 'simple'})

api = Api(app)
options = {
    'provider': provider,
    'logger': app.logger
}
api.add_resource(MovieList, '/movies', resource_class_kwargs=options)
api.add_resource(Movie, '/movies/<sid>', resource_class_kwargs=options)
api.add_resource(MovieCelebrityList, '/movies/<sid>/celebrities', resource_class_kwargs=options)
api.add_resource(Celebrity, '/celebrities/<cid>', resource_class_kwargs=options)
api.add_resource(PhotoList, '/photo/<sid>', resource_class_kwargs=options)

swagger_config = {
    "headers": [
    ],
    "specs": [
        {
            "endpoint": 'apispec',
            "route": '/apispec.json',
            "rule_filter": lambda rule: True,  # all in
            "model_filter": lambda tag: True,  # all in
        }
    ],
    "static_url_path": "/flasgger_static",
    # "static_folder": "static",  # must be set by user
    "swagger_ui": True,
    "specs_route": "/"
}
swagger = Swagger(app, config=swagger_config, template_file='docs/default.yml')

@app.route('/search', methods=['GET'])
# @cache.cached(timeout=30, query_string=True)
def api_search():
    return api_full_search()

@app.route('/fullsearch', methods=['GET'])
# @cache.cached(timeout=30, query_string=True)
def api_full_search():
    headers = {'content-type': 'application/json'}
    keyword = request.args.get("q")
    app.logger.info('the parameter is given: %s', keyword)
    result = ""
    try:
        obj = provider.search_full_list(str(keyword))
        result = json.dumps(obj, ensure_ascii=False)
    except Exception as ex:
        print(ex)
    if result:
        return result, 200, headers
    return f'Results Not Found: {keyword}', 404, headers

@app.route('/partialsearch', methods=['GET'])
# @cache.cached(timeout=30, query_string=True)
def api_partial_search():
    headers = {'content-type': 'application/json'}
    keyword = request.args.get("q")
    app.logger.info('the parameter is given: %s', keyword)
    result = ""
    try:
        obj = provider.search_partial_list(str(keyword))
        result = json.dumps(obj, ensure_ascii=False)
    except Exception as ex:
        print(ex)
    if result:
        return result, 200, headers
    return f'Results Not Found: {keyword}', 404, headers

@app.route('/fetchbysid', methods=['GET'])
# @cache.cached(timeout=30, query_string=True)
def api_fetch_by_sid():
    headers = {'content-type': 'application/json'}
    sid = request.args.get("sid")
    app.logger.info('the parameter is given: %s', sid)
    result = ""
    try:
        obj = provider.fetch_detail_info(str(sid))
        result = json.dumps(obj, ensure_ascii=False)
    except Exception as ex:
        print(ex)
    if result:
        return result, 200, headers
    return f'Results Not Found: {sid}', 404, headers

@app.route('/fetchceleritiesbysid', methods=['GET'])
# @cache.cached(timeout=30, query_string=True)
def api_fetch_celerities_by_sid():
    headers = {'content-type': 'application/json'}
    sid = request.args.get("sid")
    app.logger.info('the parameter is given: %s', sid)
    result = ""
    try:
        obj = provider.fetch_celebrities(str(sid))
        result = json.dumps(obj, ensure_ascii=False)
    except Exception as ex:
        print(ex)
    if result:
        return result, 200, headers
    return f'Results Not Found: {sid}', 404, headers

if __name__ == "__main__":
    app.run()