origin control disabled

faustomilletari · Jul 25, 2018 · bf9f6e1 · bf9f6e1
1 parent 7ffa989
commit bf9f6e1
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/tomaat/server/service.py b/tomaat/server/service.py
@@ -112,12 +112,22 @@ def __init__(self, config, app, input_interface, output_interface):
         self.output_interface = output_interface
 
     @klein_app.route('/interface', methods=['GET'])
-    def interface(self, _):
+    def interface(self, request):
+        request.setHeader('Access-Control-Allow-Origin', '*')
+        request.setHeader('Access-Control-Allow-Methods', 'GET')
+        request.setHeader('Access-Control-Allow-Headers', 'x-prototype-version,x-requested-with')
+        request.setHeader('Access-Control-Max-Age', 2520)  # 42 hours
+
         return json.dumps(self.input_interface)
 
     @klein_app.route('/predict', methods=['POST'])
     @inlineCallbacks
     def predict(self, request):
+        request.setHeader('Access-Control-Allow-Origin', '*')
+        request.setHeader('Access-Control-Allow-Methods', 'POST')
+        request.setHeader('Access-Control-Allow-Headers', 'x-prototype-version,x-requested-with')
+        request.setHeader('Access-Control-Max-Age', 2520)  # 42 hours
+
         logger.info('predicting...')
 
         result = yield threads.deferToThread(self.received_data_handler, request)