lgcosta/paperless-ngx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #33 from tienthienhd/bugfix/fix-ocr-custom

Browse Source 
fix: ocr key
This commit is contained in:
Trần Minh Đức 2024-06-04 20:32:36 +07:00 committed by GitHub
commit 7fa2c66f4a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/documents/parsers.py
View File

@ -149,15 +149,17 @@ def custom_get_parser_class_for_mime_type(mime_type: str) -> Optional[type["Docu
if not options: if not options:
return None return None
k = ApplicationConfiguration.objects.filter().first() k = ApplicationConfiguration.objects.filter().first()
best_parser = sorted(options, key=lambda _: _["weight"], reverse=True)[0]
if len(best_parser)>1:
best_parser = sorted(options, key=lambda _: _["weight"], reverse=True)[1] best_parser = sorted(options, key=lambda _: _["weight"], reverse=True)[1]
if k.ocr_key!='': if k.ocr_key!='':
headers = { headers = {
'Authorization': f'Bearer {k.ocr_key}' 'Authorization': f'Bearer {k.ocr_key}'
} }
url_ocr_pdf_by_fileid = settings.TCGROUP_OCR_CUSTOM["URL"]["URL_OCR_BY_FILEID"] url_ocr_pdf_by_fileid = settings.TCGROUP_OCR_CUSTOM["URL"]["URL_OCR_BY_FILEID"]
response_ocr = requests.post(url_ocr_pdf_by_fileid, headers=headers) response_ocr = requests.post(url_ocr_pdf_by_fileid, params={"file_id":0}, headers=headers)
logger.debug(f'status code: {response_ocr.status_code}') logger.debug(f'status code: {response_ocr.status_code}')
if response_ocr.status_code != 401: if response_ocr.status_code == 404:
best_parser = sorted(options, key=lambda _: _["weight"], reverse=True)[0] best_parser = sorted(options, key=lambda _: _["weight"], reverse=True)[0]
logger.debug('Successful key authentication ...') logger.debug('Successful key authentication ...')
logger.debug('Fail key authentication ...', best_parser["parser"]) logger.debug('Fail key authentication ...', best_parser["parser"])