mapper + lint

superset_tool/utils/init_clients.py

@@ -1,36 +1,33 @@
-# [MODULE] Superset Clients Initializer
-# PURPOSE: Централизованно инициализирует клиенты Superset для различных окружений (DEV, PROD, SBX, PREPROD).
-# COHERENCE:
-#   - Использует `SupersetClient` для создания экземпляров клиентов.
-#   - Использует `SupersetLogger` для логирования процесса.
-#   - Интегрируется с `keyring` для безопасного получения паролей.
+# <GRACE_MODULE id="superset_tool.utils.init_clients" name="init_clients.py">
+#   @SEMANTICS: utility, factory, client, initialization, configuration
+#   @PURPOSE: Централизованно инициализирует клиенты Superset для различных окружений (DEV, PROD, SBX, PREPROD), используя `keyring` для безопасного доступа к паролям.
+#   @DEPENDS_ON: superset_tool.models -> Использует SupersetConfig для создания конфигураций.
+#   @DEPENDS_ON: superset_tool.client -> Создает экземпляры SupersetClient.
+#   @DEPENDS_ON: keyring -> Для безопасного получения паролей.
 
-# [IMPORTS] Сторонние библиотеки
+# <IMPORTS>
 import keyring
 from typing import Dict
-
-# [IMPORTS] Локальные модули
 from superset_tool.models import SupersetConfig
 from superset_tool.client import SupersetClient
 from superset_tool.utils.logger import SupersetLogger
+# </IMPORTS>
 
-# CONTRACT:
-#   PURPOSE: Инициализирует и возвращает словарь клиентов `SupersetClient` для всех предопределенных окружений.
-#   PRECONDITIONS:
-#     - `keyring` должен содержать пароли для систем "dev migrate", "prod migrate", "sandbox migrate", "preprod migrate".
-#     - `logger` должен быть инициализированным экземпляром `SupersetLogger`.
-#   POSTCONDITIONS:
-#     - Возвращает словарь, где ключи - это имена окружений ('dev', 'sbx', 'prod', 'preprod'),
-#       а значения - соответствующие экземпляры `SupersetClient`.
-#   PARAMETERS:
-#     - logger: SupersetLogger - Экземпляр логгера для записи процесса инициализации.
-#   RETURN: Dict[str, SupersetClient] - Словарь с инициализированными клиентами.
-#   EXCEPTIONS:
-#     - Логирует и выбрасывает `Exception` при любой ошибке (например, отсутствие пароля, ошибка подключения).
+# --- Начало кода модуля ---
+
+# <ANCHOR id="setup_clients" type="Function">
+#   @PURPOSE: Инициализирует и возвращает словарь клиентов `SupersetClient` для всех предопределенных окружений.
+#   @PRE: `keyring` должен содержать пароли для систем "dev migrate", "prod migrate", "sbx migrate", "preprod migrate".
+#   @PRE: `logger` должен быть валидным экземпляром `SupersetLogger`.
+#   @POST: Возвращает словарь с инициализированными клиентами.
+#   @PARAM: logger: SupersetLogger - Экземпляр логгера для записи процесса.
+#   @RETURN: Dict[str, SupersetClient] - Словарь, где ключ - имя окружения, значение - `SupersetClient`.
+#   @THROW: ValueError - Если пароль для окружения не найден в `keyring`.
+#   @THROW: Exception - При любых других ошибках инициализации.
+#   @RELATION: CREATES_INSTANCE_OF -> SupersetConfig
+#   @RELATION: CREATES_INSTANCE_OF -> SupersetClient
 def setup_clients(logger: SupersetLogger) -> Dict[str, SupersetClient]:
-    """Инициализирует и настраивает клиенты для всех окружений Superset."""
-    # [ANCHOR] CLIENTS_INITIALIZATION
-    logger.info("[INFO][INIT_CLIENTS_START] Запуск инициализации клиентов Superset.")
+    logger.info("[setup_clients][Enter] Starting Superset clients initialization.")
     clients = {}
     
     environments = {
@@ -42,7 +39,7 @@ def setup_clients(logger: SupersetLogger) -> Dict[str, SupersetClient]:
     
     try:
         for env_name, base_url in environments.items():
-            logger.debug(f"[DEBUG][CONFIG_CREATE] Создание конфигурации для окружения: {env_name.upper()}")
+            logger.debug("[setup_clients][State] Creating config for environment: %s", env_name.upper())
             password = keyring.get_password("system", f"{env_name} migrate")
             if not password:
                 raise ValueError(f"Пароль для '{env_name} migrate' не найден в keyring.")
@@ -50,23 +47,21 @@ def setup_clients(logger: SupersetLogger) -> Dict[str, SupersetClient]:
             config = SupersetConfig(
                 env=env_name,
                 base_url=base_url,
-                auth={
-                    "provider": "db",
-                    "username": "migrate_user",
-                    "password": password,
-                    "refresh": True
-                },
+                auth={"provider": "db", "username": "migrate_user", "password": password, "refresh": True},
                 verify_ssl=False
             )
             
             clients[env_name] = SupersetClient(config, logger)
-            logger.debug(f"[DEBUG][CLIENT_SUCCESS] Клиент для {env_name.upper()} успешно создан.")
+            logger.debug("[setup_clients][State] Client for %s created successfully.", env_name.upper())
 
-        logger.info(f"[COHERENCE_CHECK_PASSED][INIT_CLIENTS_SUCCESS] Все клиенты ({', '.join(clients.keys())}) успешно инициализированы.")
+        logger.info("[setup_clients][Exit] All clients (%s) initialized successfully.", ', '.join(clients.keys()))
         return clients
         
     except Exception as e:
-        logger.error(f"[CRITICAL][INIT_CLIENTS_FAILED] Ошибка при инициализации клиентов: {str(e)}", exc_info=True)
+        logger.critical("[setup_clients][Failure] Critical error during client initialization: %s", e, exc_info=True)
         raise
-# END_FUNCTION_setup_clients
-# END_MODULE_init_clients
+# </ANCHOR id="setup_clients">
+
+# --- Конец кода модуля ---
+
+# </GRACE_MODULE id="superset_tool.utils.init_clients">