Created
April 29, 2017 08:50
-
-
Save yuqianma/dbd489106c749eb24d2030835b93b1e7 to your computer and use it in GitHub Desktop.
Flask crossdomain
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from datetime import timedelta | |
from flask import make_response, request, current_app | |
from functools import update_wrapper | |
# replaced 'basestring' width 'str' for python3 | |
def crossdomain(origin=None, methods=None, headers=None, | |
max_age=21600, attach_to_all=True, | |
automatic_options=True): | |
if methods is not None: | |
methods = ', '.join(sorted(x.upper() for x in methods)) | |
if headers is not None and not isinstance(headers, str): | |
headers = ', '.join(x.upper() for x in headers) | |
if not isinstance(origin, str): | |
origin = ', '.join(origin) | |
if isinstance(max_age, timedelta): | |
max_age = max_age.total_seconds() | |
def get_methods(): | |
if methods is not None: | |
return methods | |
options_resp = current_app.make_default_options_response() | |
return options_resp.headers['allow'] | |
def decorator(f): | |
def wrapped_function(*args, **kwargs): | |
if automatic_options and request.method == 'OPTIONS': | |
resp = current_app.make_default_options_response() | |
else: | |
resp = make_response(f(*args, **kwargs)) | |
if not attach_to_all and request.method != 'OPTIONS': | |
return resp | |
h = resp.headers | |
h['Access-Control-Allow-Origin'] = origin | |
h['Access-Control-Allow-Methods'] = get_methods() | |
h['Access-Control-Max-Age'] = str(max_age) | |
if headers is not None: | |
h['Access-Control-Allow-Headers'] = headers | |
return resp | |
f.provide_automatic_options = False | |
return update_wrapper(wrapped_function, f) | |
return decorator | |
@app.route('/my_service', methods=['GET', 'OPTIONS']) | |
@crossdomain(origin='*') | |
def my_service(): | |
return jsonify(foo='cross domain ftw') |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment