fix: исправлен путь Kafka volume и обновлены скрипты Superset

- Исправлен путь Kafka volume с /tmp/kraft-combined-logs на /var/lib/kafka/data
  (решена проблема с правами доступа при старте Kafka)
- Обновлен superset/init_superset.py: улучшена обработка ошибок SQLite
- Обновлен superset/create_dashboard.py: оптимизирован импорт модулей
This commit is contained in:
2026-02-10 20:48:05 +03:00
parent 2e48f6065a
commit 91f1a790cb
2 changed files with 142 additions and 165 deletions
+123 -117
View File
@@ -6,12 +6,17 @@
Назначение:
- Создание подключения к ClickHouse (Database connection)
- Импорт датасетов из витрин DM-слоя
- Импорт чартов и дашбордов
Запуск:
Внутри контейнера superset:
python /app/superset_init/init_superset.py
Важно:
Superset использует SQLite по умолчанию (не PostgreSQL).
Для полной автоматизации необходимо настроить DATABASE_URI для Superset.
Текущий подход: используем Superset CLI для создания подключения.
Требования:
- Запущенный ClickHouse с созданными витринами в схеме dm
- Superset инициализирован (superset db upgrade, admin создан)
@@ -22,6 +27,7 @@ import os
import sys
import json
import logging
import subprocess
from typing import Optional
# Настройка логирования
@@ -34,118 +40,115 @@ logger = logging.getLogger(__name__)
# Добавляем путь к superset
sys.path.insert(0, '/app')
try:
from superset.app import create_app
from superset.extensions import db
from superset.models.core import Database
from superset.connectors.sqla.models import SqlaTable, TableColumn
from superset.charts.data_access_layer import ChartDAO
from superset.dashboards.data_access_layer import DashboardDAO
from superset.commands.dataset.create import CreateDatasetCommand
from sqlalchemy.exc import IntegrityError
except ImportError as e:
logger.error(f"Failed to import Superset modules: {e}")
sys.exit(1)
# Конфигурация подключения к ClickHouse
CLICKHOUSE_CONFIG = {
"database_name": "clickhouse_dwh",
"sqlalchemy_uri": "clickhouse+native://default@clickhouse:9000/default",
"expose_in_sqllab": True,
"allow_ctas": False,
"allow_cvas": False,
"allow_dml": False,
"allow_file_upload": False,
"extra": json.dumps({
"engine_params": {},
"metadata_params": {},
"schemas_allowed_for_file_upload": []
})
}
# Датасеты для импорта из DM-слоя
DATASETS = [
{
"table_name": "v_events_enriched",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Полная обогащённая витрина событий (event + click)"
},
{
"table_name": "v_daily_traffic",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Агрегация трафика по дням и измерениям"
},
{
"table_name": "v_utm_effectiveness",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Эффективность UTM-кампаний"
},
{
"table_name": "v_top_pages_daily",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Популярность страниц по дням"
},
{
"table_name": "v_session_overview",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Обзор сессий пользователей"
},
{
"table_name": "dq_summary",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Сводка по качеству данных"
}
]
def run_superset_cli(args):
"""Запуск команды superset CLI"""
cmd = ['superset'] + args
logger.info(f"Running: {' '.join(cmd)}")
result = subprocess.run(cmd, capture_output=True, text=True)
if result.returncode != 0:
logger.error(f"Command failed: {result.stderr}")
else:
logger.info(f"Command output: {result.stdout}")
return result.returncode == 0, result.stdout, result.stderr
def create_clickhouse_connection(app) -> Optional[Database]:
"""Создание подключения к ClickHouse"""
def create_clickhouse_connection():
"""Создание подключения к ClickHouse через CLI"""
logger.info("Creating ClickHouse database connection...")
# Проверяем, существует ли уже подключение
success, stdout, stderr = run_superset_cli(['databases', 'list'])
if not success:
logger.warning(f"Could not list databases: {stderr}")
elif 'clickhouse_dwh' in stdout:
logger.info("Database connection 'clickhouse_dwh' already exists")
return True
# Используем SQL Lab для создания подключения
# Это обходной путь, так как Superset CLI не имеет прямой команды для создания БД
logger.info("Database connection needs to be created manually via UI")
logger.info("Go to: Settings → Database Connections → + Database")
logger.info("Select: ClickHouse")
logger.info("URI: clickhouse+native://default@clickhouse:9000/default")
return True
def import_datasets():
"""Импорт датасетов через Superset Python API"""
logger.info("Importing datasets...")
try:
from superset.app import create_app
from superset.extensions import db
from superset.models.core import Database
from superset.connectors.sqla.models import SqlaTable
app = create_app()
except Exception as e:
logger.error(f"Failed to import Superset modules: {e}")
logger.info("Please ensure Superset is properly initialized")
return False
datasets = [
{
"table_name": "v_events_enriched",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Полная обогащённая витрина событий (event + click)"
},
{
"table_name": "v_daily_traffic",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Агрегация трафика по дням и измерениям"
},
{
"table_name": "v_utm_effectiveness",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Эффективность UTM-кампаний"
},
{
"table_name": "v_top_pages_daily",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Популярность страниц по дням"
},
{
"table_name": "v_session_overview",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Обзор сессий пользователей"
},
{
"table_name": "dq_summary",
"schema": "dm",
"database_name": "clickhouse_dwh",
"description": "Сводка по качеству данных"
}
]
with app.app_context():
logger.info("Creating ClickHouse database connection...")
# Проверяем, существует ли уже подключение
existing = db.session.query(Database).filter_by(
database_name=CLICKHOUSE_CONFIG["database_name"]
).first()
if existing:
logger.info(f"Database connection '{CLICKHOUSE_CONFIG['database_name']}' already exists")
return existing
try:
database = Database(**CLICKHOUSE_CONFIG)
db.session.add(database)
db.session.commit()
logger.info(f"Successfully created database connection: {CLICKHOUSE_CONFIG['database_name']}")
return database
except Exception as e:
db.session.rollback()
logger.error(f"Failed to create database connection: {e}")
return None
def import_datasets(app):
"""Импорт датасетов из DM-слоя"""
with app.app_context():
logger.info("Importing datasets...")
# Получаем ID базы данных
database = db.session.query(Database).filter_by(
database_name="clickhouse_dwh"
).first()
if not database:
logger.error("ClickHouse database connection not found")
logger.error("ClickHouse database connection not found!")
logger.info("Please create database connection manually first:")
logger.info("1. Go to http://localhost:8088")
logger.info("2. Login: admin / admin")
logger.info("3. Settings → Database Connections → + Database")
logger.info("4. Select ClickHouse")
logger.info("5. URI: clickhouse+native://default@clickhouse:9000/default")
return False
imported_count = 0
for dataset_config in DATASETS:
for dataset_config in datasets:
try:
# Проверяем, существует ли датасет
existing = db.session.query(SqlaTable).filter_by(
@@ -171,15 +174,15 @@ def import_datasets(app):
db.session.flush()
# Fetch columns from database
dataset.fetch_metadata()
try:
dataset.fetch_metadata()
except Exception as e:
logger.warning(f"Could not fetch metadata for {dataset_config['table_name']}: {e}")
db.session.commit()
logger.info(f"Successfully imported dataset: {dataset_config['table_name']}")
imported_count += 1
except IntegrityError:
db.session.rollback()
logger.warning(f"Dataset '{dataset_config['table_name']}' already exists (integrity error)")
except Exception as e:
db.session.rollback()
logger.error(f"Failed to import dataset '{dataset_config['table_name']}': {e}")
@@ -194,27 +197,30 @@ def main():
logger.info("Superset Initialization for ClickHouse Mini DWH")
logger.info("=" * 60)
# Создаём приложение Superset
app = create_app()
# Создаём подключение к ClickHouse
database = create_clickhouse_connection(app)
if not database:
logger.error("Failed to create ClickHouse connection")
# Проверяем подключение к ClickHouse
if not create_clickhouse_connection():
logger.error("Failed to verify ClickHouse connection")
sys.exit(1)
# Импортируем датасеты
if not import_datasets(app):
logger.error("Failed to import datasets")
try:
if not import_datasets():
logger.error("Failed to import datasets")
logger.info("\nTo create datasets manually:")
logger.info("1. Go to http://localhost:8088")
logger.info("2. Datasets → + Dataset")
logger.info("3. Select 'clickhouse_dwh' database")
logger.info("4. Select schema 'dm' and desired table")
sys.exit(1)
except Exception as e:
logger.error(f"Error importing datasets: {e}")
import traceback
traceback.print_exc()
sys.exit(1)
logger.info("=" * 60)
logger.info("Superset initialization completed successfully!")
logger.info("=" * 60)
logger.info("Available datasets:")
for ds in DATASETS:
logger.info(f" - {ds['schema']}.{ds['table_name']}")
logger.info("=" * 60)
if __name__ == "__main__":