rewritten for mules

classification_service/views/classify_view.py

@@ -1,70 +1,15 @@
 #!/usr/bin/env python3
-import os
 from flask import request, jsonify
 from flask_classful import FlaskView
 from utils import json_required
-import requests
-import tempfile
-import numpy
-
-from pyAudioAnalysis.audioTrainTest import load_model, load_model_knn, classifier_wrapper
+import json
+import uwsgi
 
 
 class ClassifyView(FlaskView):
 
     @json_required
     def post(self):
-
         task = request.json  # tag, features, model
-
-        _, temp_model_name = tempfile.mkstemp()
-        temp_means_name = temp_model_name + "MEANS"
-
-        r = requests.get(f"http://model-service/model/{task['model']}/details")
-        r.raise_for_status()
-
-        model_details = r.json()
-
-        try:
-
-            r = requests.get(f"http://model-service/model/{task['model']}")
-            r.raise_for_status()
-
-            with open(temp_model_name, 'wb') as f:
-                f.write(r.content)
-
-            r = requests.get(f"http://model-service/model/{task['model']}?means")
-            r.raise_for_status()
-
-            with open(temp_means_name, 'wb') as f:
-                f.write(r.content)
-
-            if model_details['type'] == 'knn':
-                classifier, mean, std, classes, mid_window, mid_step, short_window, short_step, compute_beat \
-                    = load_model_knn(temp_model_name)
-
-            else:
-                classifier, mean, std, classes, mid_window, mid_step, short_window, short_step, compute_beat \
-                    = load_model(temp_model_name)
-
-            feature_vector = (numpy.array(task['features']) - mean) / std
-            class_id, probability = classifier_wrapper(classifier, model_details['type'], feature_vector)
-
-        finally:  # bruuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuh
-            try:
-                os.remove(temp_model_name)
-            except FileNotFoundError:
-                pass
-
-            try:
-                os.remove(temp_means_name)
-            except FileNotFoundError:
-                pass
-
-        # TODO: Publish to message queue
-        results = {
-            "class_id": class_id,
-            "probability": probability
-        }
-
+        uwsgi.mule_msg(json.dumps(task))
         return jsonify({"status": "OK", "msg": "enqueued"})