Skip to content

Commit 00b7662

Browse files
committed
Add "Access-Control-Allow-Origin: *" to http headers in some function.
1 parent d0aeeb9 commit 00b7662

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

cloud_dfs/__init__.py

+10-5
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import os
2-
from flask import Flask, g, render_template, request, jsonify, url_for, send_file
2+
from flask import Flask, g, render_template, request, jsonify, url_for, send_file, Response, make_response
33
import sqlalchemy
44
from cloud_dfs.token import TokenManager, NotAvailableTokenError
55
from cloud_dfs.database.models import Data, DataGroup
@@ -150,18 +150,23 @@ def get_data(hex_token):
150150
print("Got Data :", data_obj)
151151

152152
if data_obj.data_type == 'binary':
153-
return send_file(data_obj.path, mimetype='application/octet-stream',
154-
as_attachment=True, attachment_filename=data_obj.name), 200
153+
response = make_response(send_file(data_obj.path, mimetype='application/octet-stream',
154+
as_attachment=True, attachment_filename=data_obj.name))
155155
elif data_obj.data_type == 'text':
156156
with open(data_obj.path, 'rt') as f:
157157
data = f.read()
158-
return jsonify({
158+
159+
response = make_response(jsonify({
159160
'name': data_obj.name,
160161
'data': data
161-
}), 200
162+
}))
162163
else:
163164
return '', 500
164165

166+
response.status_code = 200
167+
response.headers['Access-Control-Allow-Origin'] = '*'
168+
return response
169+
165170
@app.route('/data/<hex_token>', methods=['DELETE'])
166171
def del_data(hex_token):
167172
token = bytes.fromhex(hex_token)

0 commit comments

Comments
 (0)