def get_lastest_images(self):
try:
list_of_files = glob.iglob(self.path_to_upload_thumbnail + '/*') # * means all if need specific format then *.csv
files = sorted(list_of_files, key=os.path.getctime, reverse=True)
if len(files):
latest_images = []
for file in files:
image_name = os.path.basename(file)
latest_images.append(ObjectDetectionService.get_image_url(
image_name,
self.user_id,
self.app_name,
os.path.join(config["storage"]["upload_dir"], config["storage"]["thumbnail_dir"]),
self.url_root
))
if len(latest_images) == config["storage"]["retrieval_limit"]:
break
ApiResponse.set_body(0, "Records found", {"latest_images": latest_images})
else:
ApiResponse.set_msg(32, "Record not found")
except Exception as e:
ApiResponse.set_msg(33, "Internal server error")
Logger.error(ObjectDetectionService.LOG_TAG, "Exception: {error}".format(error=str(e)))
return ApiResponse.get(config["api"]["base_code_object_detection"])
def set_prerequisites(self, params):
try:
Logger.info(ObjectDetectionService.LOG_TAG, params)
self.url_root = params.get("url_root")
self.user_id = params.get("user_id")
self.app_name = params.get("app_name")
self.path_to_image = os.path.join(ObjectDetectionService.CWD_PATH, 'storage', self.user_id,
self.app_name)
self.path_to_upload_original = os.path.join(self.path_to_image, config["storage"]["upload_dir"],
config["storage"]["original_dir"])
self.path_to_upload_thumbnail = os.path.join(self.path_to_image, config["storage"]["upload_dir"],
config["storage"]["thumbnail_dir"])
self.path_to_output = os.path.join(self.path_to_image, config["storage"]["output_dir"])
if not os.path.isdir(self.path_to_image):
mkdir_p(self.path_to_image)
if not os.path.isdir(self.path_to_upload_original):
mkdir_p(self.path_to_upload_original)
if not os.path.isdir(self.path_to_upload_thumbnail):
mkdir_p(self.path_to_upload_thumbnail)
if not os.path.isdir(self.path_to_output):
mkdir_p(self.path_to_output)
except Exception as e:
ApiResponse.set_msg(1, "Internal server error")
Logger.error(ObjectDetectionService.LOG_TAG, "Exception: {error}".format(error=str(e)))
return ApiResponse.get(config["api"]["base_code_object_detection"])
def resize_image(self, image_name):
try:
im = Image.open(os.path.join(self.path_to_upload_original, image_name))
im.thumbnail(ObjectDetectionService.SIZE, Image.ANTIALIAS)
im.save(os.path.join(self.path_to_upload_thumbnail , image_name) , "JPEG")
ApiResponse.set_body(0, "Image resized successfully", {
"image_name": image_name
})
except IOError:
ApiResponse.set_msg(46, "Cannot create thumbnail for '%s'" % image_name)
return ApiResponse.get(config["api"]["base_code_object_detection"])
def save_object_detected_image(self, image_name):
try:
Logger.info(ObjectDetectionService.LOG_TAG, "Image name received: {image_name}".format(image_name=image_name))
if(image_name):
image_path = os.path.join(self.path_to_upload_original, image_name)
image_file = imread(image_path)
self.input_q.put(image_file)
scipy.misc.imsave(os.path.join(self.path_to_output, image_name), self.output_q.get())
ApiResponse.set_body(0, "Success", { "image_url": ObjectDetectionService.get_image_url(
image_name,
self.user_id,
self.app_name,
config["storage"]["output_dir"],
self.url_root
)})
else:
ApiResponse.set_msg(2,"Image name is required")
except FileNotFoundError as e:
ApiResponse.set_msg(3, "Image not found. Please try it after uploading the image.")
Logger.error(ObjectDetectionService.LOG_TAG, "Exception: {error}".format(error=str(e)))
except Exception as e:
ApiResponse.set_msg(4, "Internal server error")
Logger.error(ObjectDetectionService.LOG_TAG, "Exception: {error}".format(error=str(e)))
return ApiResponse.get(config["api"]["base_code_object_detection"])
def save_image(self, image_file):
try:
# submit a empty part without filename
if image_file.filename == '':
ApiResponse.set_msg(16, "Image file is required")
elif image_file and ObjectDetectionService.allowed_file(image_file.filename):
filename = secure_filename(image_file.filename)
unique_filename = ObjectDetectionService.get_unique_filename(filename)
image_file.save(os.path.join(self.path_to_upload_original, unique_filename))
response = self.resize_image(unique_filename)
if(response['header']['code'] == 0):
ApiResponse.set_body(0, "Image uploaded successfully", {
"image_name": response['body']['image_name']
})
else:
return response
else:
ApiResponse.set_msg(17, "Allowed extensions are: " + ObjectDetectionService.ALLOWED_EXTENSIONS)
except Exception as e:
ApiResponse.set_msg(18, "Internal server error")
Logger.error(ObjectDetectionService.LOG_TAG, "Exception: {error}".format(error=str(e)))
return ApiResponse.get(config["api"]["base_code_object_detection"])
def decorated_function(*args, **kwargs):
if 'image_file' not in request.files:
ApiResponse.set_msg(RESPONSE['image']['offset'],
RESPONSE['image']['message'])
return ApiResponse.get(config["api"]["base_code_object_detection"])
return f(*args, **kwargs)
def prerequisite(key):
if key not in request.params:
ApiResponse.set_msg(RESPONSE[key]['offset'], RESPONSE[key]['message'])
else:
ApiResponse.set_msg(0, 'Success')
return ApiResponse.get(config["api"]["base_code_decorators"])