2024-03-08 21:19:15 -08:00

40 lines
1.1 KiB
Python

from pathlib import Path
from django.conf import settings
from django.core.checks import Error
from django.core.checks import register
@register()
def check_remote_parser_configured(app_configs, **kwargs):
if (
settings.REMOTE_OCR_ENGINE == "azureaivision"
and not settings.REMOTE_OCR_ENDPOINT
):
return [
Error(
"Azure AI Vision remote parser requires endpoint to be configured.",
),
]
if settings.REMOTE_OCR_ENGINE == "awstextract" and (
not settings.REMOTE_OCR_API_KEY_ID or not settings.REMOTE_OCR_REGION
):
return [
Error(
"AWS Textract remote parser requires access key ID and region to be configured.",
),
]
if settings.REMOTE_OCR_ENGINE == "googlecloudvision" and (
not settings.REMOTE_OCR_CREDENTIALS_FILE
or not Path(settings.REMOTE_OCR_CREDENTIALS_FILE).exists()
):
return [
Error(
"Google Cloud Vision remote parser requires a valid credentials file to be configured.",
),
]
return []