demo_api.py

import sys
from flask import Flask, request, send_from_directory
from flask_restful import Resource, Api
from toolz import merge
from flask_cors import CORS

from api.algorithm.result import get_result

app = Flask(__name__)
cors = CORS(app, resources={r"/*": {"origins": "*"}})
api = Api(app)


class PageExtractor(Resource):
    def get(self):
        url = request.args.get('url')
        result = get_result(url)
        return [merge(result,
                     {'url': url})]

    def post(self):
        url = request.form['url']
        result = get_result(url)
        return [merge(result,
                     {'url': url})]


@app.route('/js/<path:path>')
def send_js(path):
    return send_from_directory('js', path)


api.add_resource(PageExtractor, '/extract')

if __name__ == '__main__':
    debug = True if len(sys.argv) > 1 else False
    print(f"\n ==== Starting page extractor demo API. Debug={debug} ====\n")
    app.run(debug=True)