diff --git a/core/utils/vendors.py b/core/utils/vendors.py index 61a210e9..462f2eab 100644 --- a/core/utils/vendors.py +++ b/core/utils/vendors.py @@ -1,27 +1,39 @@ +import logging +from contextlib import suppress from typing import Type +from celery import current_task +from celery.utils.log import get_task_logger + from core.models import Vendor from core.vendors import AbstractVendor from evibes.utils.misc import create_object +sync_logger = logging.getLogger("django") +async_logger = get_task_logger(__name__) + + +def _in_celery_task() -> bool: + task = current_task + with suppress(Exception): + return bool(task and getattr(task, "request", None) and getattr(task.request, "id", None)) + return False + def get_vendors_integrations(name: str | None = None) -> list[Type[AbstractVendor]]: vendors_integrations: list[Type[AbstractVendor]] = [] - vendors = ( - Vendor.objects.filter( - is_active=True, - integration_path__isnull=False, - name=name, - ) - if name - else Vendor.objects.filter( - is_active=True, - integration_path__isnull=False, - ) - ) + + vendors = Vendor.objects.filter(is_active=True, integration_path__isnull=False) + if name: + vendors = vendors.filter(name=name) + + logger = async_logger if _in_celery_task() else sync_logger + for vendor in vendors: - if vendor.integration_path: - module_name = ".".join(vendor.integration_path.split(".")[:-1]) - class_name = vendor.integration_path.split(".")[-1] + try: + module_name, class_name = vendor.integration_path.rsplit(".", 1) vendors_integrations.append(create_object(module_name, class_name)) + except Exception as e: + logger.warning("Couldn't load integration for vendor %s: %s", vendor.name, e) + return vendors_integrations