storing images in tempfolder for matching and storing

api/endpoints/findsalamanderinfo.py

@@ -2,17 +2,35 @@ from flask import request, jsonify
 from flask_restful import Resource
 from flask_jwt_extended import get_jwt_identity, jwt_required
 
+ALLOWED_EXTENSIONS = ['jpeg', 'png', 'jpg']
+
 
 # findSalamanderInfo endpoint
 class FindSalamanderInfo(Resource):
     @staticmethod
     @jwt_required
     def post():
+        if "image" not in request.files:
+            return jsonify({"message": "no file given"})
         image = request.files['image']
+        user_id = get_jwt_identity()
         if image.filename != '':
+            if image and allowed_image(image.filename):
                 # run AI on image
-            output = {"sex": "AI_sex", "species": "AI_species"}
+                # store processed and original image
+                image.filename = str(user_id) + "." + extension(image.filename)
+                image.save("temp_images/" + image.filename)
+                return jsonify({"sex": "AI_sex", "species": "AI_species"})
+            else:
+                return jsonify({"message": "file extension not allowed"})
         else:
-            output = {"message": "something wrong with image"}
+            return jsonify({"message": "something wrong with image"})
+
+
+def allowed_image(filename):
+    return '.' in filename and \
+           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
 
-        return output
+def extension(filename):
+    return filename.rsplit('.', 1)[1].lower()