Update consumer.py

This commit is contained in:
shamoon 2025-06-26 14:49:08 -07:00
parent ba91bb154a
commit 056b314a89
No known key found for this signature in database

View File

@ -468,7 +468,6 @@ class ConsumerPlugin(
# now that everything is done, we can start to store the document # now that everything is done, we can start to store the document
# in the system. This will be a transaction and reasonably fast. # in the system. This will be a transaction and reasonably fast.
success = False success = False
result = None
store_exception = None store_exception = None
try: try:
with transaction.atomic(): with transaction.atomic():
@ -572,8 +571,13 @@ class ConsumerPlugin(
# Return the most up to date fields # Return the most up to date fields
document.refresh_from_db() document.refresh_from_db()
result = f"Success. New document id {document.pk} created" document_parser.cleanup()
elif store_exception: tempdir.cleanup()
return f"Success. New document id {document.pk} created"
else:
document_parser.cleanup()
tempdir.cleanup()
if store_exception:
self._fail( self._fail(
str(store_exception), str(store_exception),
f"The following error occurred while storing document " f"The following error occurred while storing document "
@ -587,10 +591,6 @@ class ConsumerPlugin(
f"Error occurred while saving {self.filename}.", f"Error occurred while saving {self.filename}.",
) )
document_parser.cleanup()
tempdir.cleanup()
return result
def _parse_title_placeholders(self, title: str) -> str: def _parse_title_placeholders(self, title: str) -> str:
local_added = timezone.localtime(timezone.now()) local_added = timezone.localtime(timezone.now())