From 202964aea604bf9d71be5690c2d6cc3cc75cffe5 Mon Sep 17 00:00:00 2001
From: Egor fureunoir Gorbunov <contact@fureunoir.com>
Date: Tue, 21 Oct 2025 12:44:19 +0300
Subject: [PATCH] Features: 1) Add `_in_celery_task` utility function to detect
 task execution context; 2) Enhance vendor integration retrieval with
 context-specific logging (async/sync); 3) Improve error handling in vendor
 integration loading with exception logging;

Fixes: None;

Extra: 1) Refactor vendor filtering logic for clarity; 2) Add imports for logging, contextlib, and Celery utilities;
---
 core/utils/vendors.py | 42 +++++++++++++++++++++++++++---------------
 1 file changed, 27 insertions(+), 15 deletions(-)

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