CC-5888: Handle file deletion if the file is stored in the cloud

This commit is contained in:
drigato 2014-07-16 12:03:22 -04:00
parent aaee522ec6
commit 7ca6b91cdf
4 changed files with 17 additions and 17 deletions

View file

@ -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))

View file

@ -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)