keannu125/DocManagerBackend
1
0
Fork 0
mirror of https://github.com/lemeow125/DocManagerBackend.git synced 2025-04-28 02:31:23 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add fix for file uploads with . symbols inside the filename

Browse source
This commit is contained in:
Keannu Christian Bernasol 2024-12-04 02:51:57 +08:00
parent e0eba6ca21
commit 7584f09b62
3 changed files with 25 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

12
docmanager_backend/config/management/commands/start_watcher.py
View file

@ -37,6 +37,7 @@ class PDFHandler(FileSystemEventHandler):
def process_pdf(self, file_path):
try:
filename = os.path.basename(file_path)
filename = str(filename).replace(" ", "")
metadata = ""
document_type = ""
@ -65,10 +66,7 @@ class PDFHandler(FileSystemEventHandler):
if line.strip():
document_type = line.strip().lower()
break
if (
not document_type
or document_type not in Document.DOCUMENT_TYPE_CHOICES
):
if not document_type:
document_type = "other"
metadata += text
@ -84,9 +82,11 @@ class PDFHandler(FileSystemEventHandler):
)
if created:
DOCUMENT.file.save(name=filename, content=File(open(file_path, "rb")))
DOCUMENT.file.save(
name=filename, content=File(open(file_path, "rb")))
self.logger.info(
f"Document '{filename}' created successfully with type '{document_type}'."
f"Document '{filename}' created successfully with type '{
document_type}'."
)
else: