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

Work around that error

Browse Source
This commit is contained in:
Trenton H
2023-12-11 17:41:03 -08:00
parent 5266bd1590
commit ad5c9ef208
2 changed files with 18 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/paperless_tesseract/setting_schema.py
View File

@@ -26,7 +26,11 @@ class OcrSetting:
def get_ocr_settings() -> OcrSetting:
db_settings = OcrSettingModel.objects.all().first()
# assert db_settings is not None
# Workaround for a test where the migration hasn't run to create the single model
if db_settings is None:
OcrSettingModel.objects.create()
db_settings = OcrSettingModel.objects.all().first()
assert db_settings is not None
user_args = None
if db_settings is not None and db_settings.user_args:
@@ -35,41 +39,30 @@ def get_ocr_settings() -> OcrSetting:
user_args = json.loads(settings.OCR_USER_ARGS)
return OcrSetting(
pages=db_settings.pages if db_settings is not None else settings.OCR_PAGES,
pages=db_settings.pages or settings.OCR_PAGES,
language=db_settings.language
if db_settings is not None and db_settings.language is not None
else settings.OCR_LANGUAGE,
output_type=db_settings.output_type
if db_settings is not None
else settings.OCR_OUTPUT_TYPE,
mode=db_settings.mode if db_settings is not None else settings.OCR_MODE,
output_type=db_settings.output_type or settings.OCR_OUTPUT_TYPE,
mode=db_settings.mode or settings.OCR_MODE,
skip_archive_file=(
db_settings.skip_archive_file
if db_settings is not None
else settings.OCR_SKIP_ARCHIVE_FILE
db_settings.skip_archive_file or settings.OCR_SKIP_ARCHIVE_FILE
),
image_dpi=db_settings.image_dpi
if db_settings is not None
else settings.OCR_IMAGE_DPI,
clean=db_settings.unpaper_clean
if db_settings is not None
else settings.OCR_CLEAN,
deskew=db_settings.deskew if db_settings is not None else settings.OCR_DESKEW,
rotate=db_settings.rotate_pages
if db_settings is not None
else settings.OCR_ROTATE_PAGES,
clean=db_settings.unpaper_clean or settings.OCR_CLEAN,
deskew=db_settings.deskew or settings.OCR_DESKEW,
rotate=db_settings.rotate_pages or settings.OCR_ROTATE_PAGES,
rotate_threshold=(
db_settings.rotate_pages_threshold
if db_settings is not None
else settings.OCR_ROTATE_PAGES_THRESHOLD
db_settings.rotate_pages_threshold or settings.OCR_ROTATE_PAGES_THRESHOLD
),
max_image_pixel=db_settings.max_image_pixels
if db_settings is not None
else settings.OCR_MAX_IMAGE_PIXELS,
color_conversion_strategy=(
db_settings.color_conversion_strategy
if db_settings is not None
else settings.OCR_COLOR_CONVERSION_STRATEGY
or settings.OCR_COLOR_CONVERSION_STRATEGY
),
user_args=user_args,
)