CC-5888: Handle file deletion if the file is stored in the cloud
This commit is contained in:
parent
aaee522ec6
commit
7ca6b91cdf
4 changed files with 17 additions and 17 deletions
|
@ -17,7 +17,6 @@ class CloudStorageUploader:
|
|||
file_name, extension = os.path.splitext(file_base_name)
|
||||
object_name = "%s_%s%s" % (file_name, str(uuid.uuid4()), extension)
|
||||
|
||||
#cls = get_driver(getattr(Provider, self._provider))
|
||||
driver = self.get_cloud_driver()
|
||||
|
||||
try:
|
||||
|
@ -49,16 +48,14 @@ class CloudStorageUploader:
|
|||
def delete_obj(self, obj_name):
|
||||
driver = self.get_cloud_driver()
|
||||
|
||||
return_msg = dict()
|
||||
return_msg["success"] = False
|
||||
try:
|
||||
cloud_obj = driver.get_object(container_name=self._bucket,
|
||||
object_name=obj_name)
|
||||
return_msg["filesize"] = getattr(cloud_obj, 'size')
|
||||
return_msg["success"] = driver.delete_object(obj=cloud_obj)
|
||||
return return_msg
|
||||
filesize = getattr(cloud_obj, 'size')
|
||||
driver.delete_object(obj=cloud_obj)
|
||||
return filesize
|
||||
except ObjectDoesNotExistError:
|
||||
logging.info("Could not find object on %s" % self._provider)
|
||||
raise Exception("Could not find object on %s" % self._provider)
|
||||
|
||||
def get_cloud_driver(self):
|
||||
cls = get_driver(getattr(Provider, self._provider))
|
||||
|
|
|
@ -176,12 +176,11 @@ class MessageListener:
|
|||
elif message_type == "delete":
|
||||
object_name = msg_dict["object_name"]
|
||||
csu = CloudStorageUploader(self._provider, self._bucket, self._api_key, self._api_key_secret)
|
||||
response = csu.delete_obj(object_name)
|
||||
if response["success"]:
|
||||
audio_metadata = dict()
|
||||
audio_metadata["delete_success"] = True
|
||||
audio_metadata["filesize"] = response["filesize"]
|
||||
StatusReporter.report_success_to_callback_url(callback_url, api_key, audio_metadata)
|
||||
filesize = csu.delete_obj(object_name)
|
||||
return_data = dict()
|
||||
return_data["filesize"] = filesize
|
||||
return_data["import_status"] = 1
|
||||
StatusReporter.report_success_to_callback_url(callback_url, api_key, return_data)
|
||||
|
||||
except KeyError as e:
|
||||
# A field in msg_dict that we needed was missing (eg. audio_file_path)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue