Initial Draft

lib/api/endpoints/MappingAPI.py

@@ -0,0 +1,35 @@
+from flask.views import MethodView
+from flask import request,Response
+
+import json
+import traceback
+import logging
+
+import lib.api.vector.VectorStore as cv
+
+class MappingAPI(MethodView):
+    lv_logger = logging.getLogger(__name__)
+
+    def get(self):
+        resp = { "test message": "working"}
+        status = 200
+        return Response(json.dumps(resp), status=status, mimetype='application/json')
+
+    def post(self):
+        try:
+            # Saving file
+            lv_file = request.files['file']
+            lv_source_domain = request.form['source_domain']
+            lv_saved_file_name = 'storage/' + lv_file.filename.split('/')[-1]
+            lv_file.save(lv_saved_file_name)
+
+            # Processing the file
+            lv_response = cv.fn_map_data(lv_saved_file_name,lv_file.filename.split('/')[-1],lv_source_domain)
+
+            status = 200
+            return Response(lv_response, status=status, mimetype='application/json')
+        except Exception as e:
+            self.lv_logger.error(e)
+            self.lv_logger.error(traceback.format_exc())
+            return Response(json.dumps({"error_message": str(e)}), status=500, mimetype='application/json')
+        

lib/api/endpoints/VectorStoreAPI.py

@@ -1,39 +1,39 @@
-from flask.views import MethodView
-from flask import request,Response
-
-import json
-import traceback
-import logging
-
-import lib.api.vector.VectorStore as cv
-
-class VectorStoreAPI(MethodView):
-    lv_logger = logging.getLogger(__name__)
-    
-    def get(self):
-        resp = { "test message": "working"}
-        status = 200
-        return Response(json.dumps(resp), status=status, mimetype='application/json')
-
-    def post(self):
-        try:
-            # Saving file
-            lv_file = request.files['file']
-            lv_domain = request.form['domain']
-            lv_file_name = 'storage/' + lv_domain + ".xlsx"
-            lv_file.save(lv_file_name)
-            
-            # Processing the file
-            lv_status = cv.fn_create_vector_store(lv_file_name, lv_domain)
-
-            return Response(
-                                json.dumps({"status":lv_status}), 
-                                status=200, 
-                                mimetype='application/json'
-                           )
-        except Exception as e:
-            self.lv_logger.error(e)
-            self.lv_logger.error(type(e))
-            self.lv_logger.error(traceback.format_exc())
-
+from flask.views import MethodView
+from flask import request,Response
+
+import json
+import traceback
+import logging
+
+import lib.api.vector.VectorStore as cv
+
+class VectorStoreAPI(MethodView):
+    lv_logger = logging.getLogger(__name__)
+    
+    def get(self):
+        resp = { "test message": "working"}
+        status = 200
+        return Response(json.dumps(resp), status=status, mimetype='application/json')
+
+    def post(self):
+        try:
+            # Saving file
+            lv_file = request.files['file']
+            lv_domain = request.form['domain']
+            lv_file_name = 'storage/' + lv_domain + ".xlsx"
+            lv_file.save(lv_file_name)
+            
+            # Processing the file
+            lv_status = cv.fn_create_vector_store(lv_file_name, lv_domain)
+
+            return Response(
+                                json.dumps({"status":lv_status}), 
+                                status=200, 
+                                mimetype='application/json'
+                           )
+        except Exception as e:
+            self.lv_logger.error(e)
+            self.lv_logger.error(type(e))
+            self.lv_logger.error(traceback.format_exc())
+
             return Response(json.dumps({"error_message":str(e)}), status=500, mimetype='application/json')