Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
7c1f449bce | |||
0dbcc0e380 | |||
b8b5a2518c |
20
imgupload.py
20
imgupload.py
@ -4,7 +4,7 @@ imgupload.py
|
|||||||
|
|
||||||
Flask application for processing images uploaded through POST requests.
|
Flask application for processing images uploaded through POST requests.
|
||||||
"""
|
"""
|
||||||
from flask import Flask, request, jsonify, abort, Response
|
from flask import Flask, request, jsonify, Response
|
||||||
from flask_api import status
|
from flask_api import status
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
import os
|
import os
|
||||||
@ -17,7 +17,7 @@ app = Flask(__name__) # app is the app
|
|||||||
|
|
||||||
|
|
||||||
def allowed_extension(testext):
|
def allowed_extension(testext):
|
||||||
if testext in settings.ALLOWED_EXTENSIONS:
|
if testext.lower() in settings.ALLOWED_EXTENSIONS:
|
||||||
return True
|
return True
|
||||||
else:
|
else:
|
||||||
return False
|
return False
|
||||||
@ -49,6 +49,11 @@ def upload():
|
|||||||
if request.form["uploadKey"] in validkeys: # check if uploadKey is valid
|
if request.form["uploadKey"] in validkeys: # check if uploadKey is valid
|
||||||
print("Key is valid!")
|
print("Key is valid!")
|
||||||
|
|
||||||
|
if "verify" in request.form.keys():
|
||||||
|
if request.form["verify"] == "true":
|
||||||
|
print("Request is asking if key is valid (it is)")
|
||||||
|
return jsonify({'status': 'key_valid'})
|
||||||
|
|
||||||
if "imageUpload" in request.files: # check if image to upload was provided
|
if "imageUpload" in request.files: # check if image to upload was provided
|
||||||
f = request.files["imageUpload"] # f is the image to upload
|
f = request.files["imageUpload"] # f is the image to upload
|
||||||
else:
|
else:
|
||||||
@ -81,21 +86,16 @@ def upload():
|
|||||||
|
|
||||||
else: # if the extension was invalid
|
else: # if the extension was invalid
|
||||||
print("Uploaded extension is invalid!")
|
print("Uploaded extension is invalid!")
|
||||||
abort(415)
|
return jsonify({'status': 'error', 'error': 'INVALID_EXTENSION'}), status.HTTP_415_UNSUPPORTED_MEDIA_TYPE
|
||||||
|
|
||||||
else: # if the key was not valid
|
else: # if the key was not valid
|
||||||
print("Key is invalid!")
|
print("Key is invalid!")
|
||||||
print("Request key: {0}".format(request.form["uploadKey"]))
|
print("Request key: {0}".format(request.form["uploadKey"]))
|
||||||
abort(401)
|
return jsonify({'status': 'error', 'error': 'UNAUTHORIZED'}), status.HTTP_401_UNAUTHORIZED
|
||||||
|
|
||||||
else: # if uploadKey was not found in request body
|
else: # if uploadKey was not found in request body
|
||||||
print("No uploadKey found in request!")
|
print("No uploadKey found in request!")
|
||||||
abort(401)
|
return jsonify({'status': 'error', 'error': 'UNAUTHORIZED'}), status.HTTP_401_UNAUTHORIZED
|
||||||
|
|
||||||
|
|
||||||
else: # if the request method wasn't post
|
|
||||||
print("Request method was not POST!")
|
|
||||||
abort(405)
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
print("Run with `flask` or a WSGI server!")
|
print("Run with `flask` or a WSGI server!")
|
||||||
|
Reference in New Issue
Block a user