keannu125/DocManagerBackend
1
0
Fork 0
mirror of https://github.com/lemeow125/DocManagerBackend.git synced 2025-01-18 17:13:00 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix files not being read properly in watcher

Browse source
This commit is contained in:
Keannu Bernasol 2024-11-28 13:41:58 +08:00
parent 65662aeb45
commit 559baca38c
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -10,6 +10,7 @@ from config.settings import MEDIA_ROOT
from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler
from documents.models import Document
from django.core.files import File
import logging
import time
@ -67,16 +68,19 @@ class PDFHandler(FileSystemEventHandler):
metadata += text
document, created = Document.objects.get_or_create(
# Open the file for instance creation
DOCUMENT, created = Document.objects.get_or_create(
name=filename,
defaults={
'number_pages': num_pages,
'ocr_metadata': metadata,
'document_type': document_type
}
},
)
if created:
DOCUMENT.file.save(
name=filename, content=File(open(file_path, 'rb')))
self.logger.info(f"Document '{filename}' created successfully with type '{
document_type}'.")