init refactor

2025-07-29 17:56:15 +03:00
parent ca2357e2e2
commit f368f5ced9
3 changed files with 216 additions and 114 deletions
--- a/superset_tool/utils/init_clients.py
+++ b/superset_tool/utils/init_clients.py
@@ -0,0 +1,100 @@
+# [MODULE] Superset Init clients
+# @contract: Автоматизирует процесс инициализации клиентов для использования скриптами.
+# @semantic_layers:
+#   1. Инициализация логгера и клиентов Superset.
+# @coherence:
+#   - Использует `SupersetClient` для взаимодействия с API Superset.
+#   - Использует `SupersetLogger` для централизованного логирования.
+#   - Интегрируется с `keyring` для безопасного хранения паролей.
+
+# [IMPORTS] Стандартная библиотека
+import logging
+from datetime import datetime
+from pathlib import Path
+
+# [IMPORTS] Сторонние библиотеки
+import keyring
+
+# [IMPORTS] Локальные модули
+from superset_tool.models import SupersetConfig
+from superset_tool.client import SupersetClient
+from superset_tool.utils.logger import SupersetLogger
+
+
+# [FUNCTION] setup_clients
+# @contract: Инициализирует и возвращает SupersetClient для каждого заданного окружения.
+# @pre:
+#   - `keyring` должен содержать необходимые пароли для "dev migrate", "prod migrate", "sandbox migrate".
+#   - `logger` должен быть инициализирован.
+# @post:
+#   - Возвращает словарь {env_name: SupersetClient_instance}.
+#   - Логирует успешную инициализацию или ошибку.
+# @raise:
+#   - `Exception`: При любой ошибке в процессе инициализации клиентов (например, отсутствие пароля в keyring, проблемы с сетью при первой аутентификации).
+def setup_clients(logger: SupersetLogger):
+    """Инициализация клиентов для разных окружений"""
+    # [ANCHOR] CLIENTS_INITIALIZATION
+    clients = {}
+    try:
+        # [INFO] Инициализация конфигурации для Dev
+        dev_config = SupersetConfig(
+            base_url="https://devta.bi.dwh.rusal.com/api/v1",
+            auth={
+                "provider": "db",
+                "username": "migrate_user",
+                "password": keyring.get_password("system", "dev migrate"),
+                "refresh": True
+            },
+            verify_ssl=False
+        )
+        # [DEBUG] Dev config created: {dev_config.base_url}
+
+        # [INFO] Инициализация конфигурации для Prod
+        prod_config = SupersetConfig(
+            base_url="https://prodta.bi.dwh.rusal.com/api/v1",
+            auth={
+                "provider": "db",
+                "username": "migrate_user",
+                "password": keyring.get_password("system", "prod migrate"),
+                "refresh": True
+            },
+            verify_ssl=False
+        )
+        # [DEBUG] Prod config created: {prod_config.base_url}
+
+        # [INFO] Инициализация конфигурации для Sandbox
+        sandbox_config = SupersetConfig(
+            base_url="https://sandboxta.bi.dwh.rusal.com/api/v1",
+            auth={
+                "provider": "db",
+                "username": "migrate_user",
+                "password": keyring.get_password("system", "sandbox migrate"),
+                "refresh": True
+            },
+            verify_ssl=False
+        )
+        # [DEBUG] Sandbox config created: {sandbox_config.base_url}
+
+        # [INFO] Инициализация конфигурации для Preprod
+        preprod_config = SupersetConfig(
+            base_url="https://preprodta.bi.dwh.rusal.com/api/v1",
+            auth={
+                "provider": "db",
+                "username": "migrate_user",
+                "password": keyring.get_password("system", "preprod migrate"),
+                "refresh": True
+            },
+            verify_ssl=False
+        )
+        # [DEBUG] Sandbox config created: {sandbox_config.base_url}
+
+        # [INFO] Создание экземпляров SupersetClient
+        clients['dev'] = SupersetClient(dev_config, logger)
+        clients['sbx'] = SupersetClient(sandbox_config,logger)
+        clients['prod'] = SupersetClient(prod_config,logger)
+        clients['preprod'] = SupersetClient(preprod_config,logger)
+        logger.info("[COHERENCE_CHECK_PASSED] Клиенты для окружений успешно инициализированы", extra={"envs": list(clients.keys())})
+        return clients
+    except Exception as e:
+        logger.error(f"[ERROR] Ошибка инициализации клиентов: {str(e)}", exc_info=True)
+        raise