fix update of images;

This commit is contained in:
Marco Zeisler 2021-01-16 19:24:07 +01:00
parent 8e24d0333e
commit 7207080b66
3 changed files with 16 additions and 11 deletions

View File

@ -70,7 +70,7 @@ class DropboxService(StorageServiceInterface):
return file_bytes
@staticmethod
def move_file(identifier_from: str, identifier_to: str, file: bytes) -> bool:
def move_file(identifier_from: str, identifier_to: str, file: bytes = None) -> bool:
"""Move/Rename a file on the dropbox storage.
:param identifier_from: Location of the file to be moved

View File

@ -74,20 +74,22 @@ class MongoDBService:
old = {"identifier": identifier} # Query for old version
metadata_orig = MongoDBService.getSingle(identifier)
metadata_new = metadata_orig
metadata_new['previous'] = identifier + '_' + str(metadata_orig['version'])
identifier_changed = identifier + '_' + str(metadata_orig['version']) # Set identifier to include version
metadata_new['version'] = str(int(metadata_new['version']) + 1) # Increment version by one
existing_meta_data_copy = MongoDBService.getSingle(identifier)
id_for_previous = identifier + '_' + str(meta['version'])
existing_meta_data_copy['previous'] = id_for_previous
identifier_changed = id_for_previous # Set identifier to include version
metadata_new.update(meta)
del meta['filename']
existing_meta_data_copy.update(meta)
existing_meta_data_copy['version'] = meta['version'] + 1 # Increment version by one
print("MongoDBService: identifier_changed: ", identifier_changed)
col.update_one(old, {"$set": {"identifier": identifier_changed, "filename": identifier_changed + '.jpg'}})
print("MongoDBService: Old object is ", col.find_one({"identifier": identifier_changed}))
MongoDBService.createSingle(metadata_new, identifier, decoded_image)
MongoDBService.createSingle(existing_meta_data_copy, identifier, decoded_image)
print("MongoDBService: New object is ", col.find_one({"identifier": identifier}))
return True

View File

@ -210,7 +210,6 @@ class ImageEndpoint:
def image_api_update(request, identifier):
logger.debug('Image UPDATE single call: {}'.format(request))
try:
payload = json.loads(request.body)
except json.decoder.JSONDecodeError:
@ -222,7 +221,7 @@ class ImageEndpoint:
'Error': str(e)}, status=500, safe=False)
b64encoded_image = payload['image_data']
identifier = payload['id']
# identifier = identifier
metadata = payload['metadata']
filename = payload['metadata']['filename']
@ -233,14 +232,18 @@ class ImageEndpoint:
'id': identifier}, status=404, safe=False)
metadata['version'] = mongodb_metadata['version']
filename = mongodb_metadata['filename']
decoded_image = WrapperService.unwrap_file(b64encoded_image)
MongoDBService.updateSingle(identifier, decoded_image, metadata)
# TODO TOBIAS get old image for MinIO
old_bytes = None
for service in ImageEndpoint.storageServiceList:
orig_new_name = identifier + '_' + str(metadata['version']) + '.jpg'
if not service.move_file(identifier, orig_new_name, decoded_image):
if not service.move_file(identifier, orig_new_name, old_bytes):
print("Could not move file from {} to {}".format(identifier, orig_new_name))
if not service.create_file(filename, decoded_image):
print("Could not save updated image to " + service.name)