mirror of
https://github.com/lemeow125/DocManagerBackend.git
synced 2025-01-19 09:33:01 +08:00
27 lines
843 B
Python
27 lines
843 B
Python
|
from django.db.models.signals import post_save
|
||
|
|
from django.dispatch import receiver
|
||
|
|
from config.settings import MEDIA_ROOT
|
||
|
|
import os
|
||
|
|
import fitz
|
||
|
|
import pytesseract
|
||
|
|
from PIL import Image
|
||
|
|
from .models import Document
|
||
|
|
|
||
|
|
|
||
|
|
@receiver(post_save, sender=Document)
|
||
|
|
def domain_post_save(sender, instance, **kwargs):
|
||
|
|
if not instance.ocr_metadata:
|
||
|
|
metadata = ""
|
||
|
|
with fitz.open(os.path.join(MEDIA_ROOT, instance.file.name)) as doc:
|
||
|
|
mat = fitz.Matrix(1.2, 1.2)
|
||
|
|
for page in doc:
|
||
|
|
pix = page.get_pixmap(matrix=mat)
|
||
|
|
output = f'{page.number}.jpg'
|
||
|
|
pix.save(output)
|
||
|
|
res = str(pytesseract.image_to_string(Image.open(output)))
|
||
|
|
os.remove(output)
|
||
|
|
metadata += res
|
||
|
|
|
||
|
|
instance.ocr_metadata = metadata
|
||
|
|
instance.save()
|