+ deduplicate

This commit is contained in:
Volobuev Andrey
2025-07-09 17:30:48 +03:00
parent 617cb3fa94
commit 5ff7c2aca9
5 changed files with 79 additions and 22 deletions

View File

@@ -175,7 +175,7 @@ def backup_dashboards(client: SupersetClient,
if rotate_archive: if rotate_archive:
# [ANCHOR] ARCHIVE_OLD_BACKUPS # [ANCHOR] ARCHIVE_OLD_BACKUPS
try: try:
archive_exports(dashboard_dir, logger=logger) archive_exports(dashboard_dir, logger=logger, deduplicate=True)
logger.debug(f"[DEBUG] Старые экспорты для '{dashboard_title}' архивированы.") logger.debug(f"[DEBUG] Старые экспорты для '{dashboard_title}' архивированы.")
except Exception as cleanup_error: except Exception as cleanup_error:
logger.warning( logger.warning(
@@ -282,6 +282,7 @@ def main() -> int:
clients['dev'], clients['dev'],
"DEV", "DEV",
superset_backup_repo, superset_backup_repo,
rotate_archive=True,
logger=logger logger=logger
) )
@@ -290,6 +291,7 @@ def main() -> int:
clients['sbx'], clients['sbx'],
"SBX", "SBX",
superset_backup_repo, superset_backup_repo,
rotate_archive=True,
logger=logger logger=logger
) )
@@ -298,6 +300,7 @@ def main() -> int:
clients['prod'], clients['prod'],
"PROD", "PROD",
superset_backup_repo, superset_backup_repo,
rotate_archive=True,
logger=logger logger=logger
) )

View File

@@ -38,7 +38,7 @@ logger.info("[COHERENCE_CHECK_PASSED] Логгер инициализирова
# @semantic: Определяет, как UUID и URI базы данных Clickhouse должны быть изменены. # @semantic: Определяет, как UUID и URI базы данных Clickhouse должны быть изменены.
# @invariant: 'old' и 'new' должны содержать полные конфигурации. # @invariant: 'old' и 'new' должны содержать полные конфигурации.
database_config_click = { database_config_click = {
"new": { "old": {
"database_name": "Prod Clickhouse", "database_name": "Prod Clickhouse",
"sqlalchemy_uri": "clickhousedb+connect://clicketl:XXXXXXXXXX@rgm-s-khclk.hq.root.ad:443/dm", "sqlalchemy_uri": "clickhousedb+connect://clicketl:XXXXXXXXXX@rgm-s-khclk.hq.root.ad:443/dm",
"uuid": "b9b67cb5-9874-4dc6-87bd-354fc33be6f9", "uuid": "b9b67cb5-9874-4dc6-87bd-354fc33be6f9",
@@ -47,7 +47,7 @@ database_config_click = {
"allow_cvas": "false", "allow_cvas": "false",
"allow_dml": "false" "allow_dml": "false"
}, },
"old": { "new": {
"database_name": "Dev Clickhouse", "database_name": "Dev Clickhouse",
"sqlalchemy_uri": "clickhousedb+connect://dwhuser:XXXXXXXXXX@10.66.229.179:8123/dm", "sqlalchemy_uri": "clickhousedb+connect://dwhuser:XXXXXXXXXX@10.66.229.179:8123/dm",
"uuid": "e9fd8feb-cb77-4e82-bc1d-44768b8d2fc2", "uuid": "e9fd8feb-cb77-4e82-bc1d-44768b8d2fc2",
@@ -63,7 +63,7 @@ logger.debug("[CONFIG] Конфигурация Clickhouse загружена.")
# @semantic: Определяет, как UUID и URI базы данных Greenplum должны быть изменены. # @semantic: Определяет, как UUID и URI базы данных Greenplum должны быть изменены.
# @invariant: 'old' и 'new' должны содержать полные конфигурации. # @invariant: 'old' и 'new' должны содержать полные конфигурации.
database_config_gp = { database_config_gp = {
"new": { "old": {
"database_name": "Prod Greenplum", "database_name": "Prod Greenplum",
"sqlalchemy_uri": "postgresql+psycopg2://viz_powerbi_gp_prod:XXXXXXXXXX@10.66.229.201:5432/dwh", "sqlalchemy_uri": "postgresql+psycopg2://viz_powerbi_gp_prod:XXXXXXXXXX@10.66.229.201:5432/dwh",
"uuid": "805132a3-e942-40ce-99c7-bee8f82f8aa8", "uuid": "805132a3-e942-40ce-99c7-bee8f82f8aa8",
@@ -72,7 +72,7 @@ database_config_gp = {
"allow_cvas": "true", "allow_cvas": "true",
"allow_dml": "true" "allow_dml": "true"
}, },
"old": { "new": {
"database_name": "DEV Greenplum", "database_name": "DEV Greenplum",
"sqlalchemy_uri": "postgresql+psycopg2://viz_superset_gp_dev:XXXXXXXXXX@10.66.229.171:5432/dwh", "sqlalchemy_uri": "postgresql+psycopg2://viz_superset_gp_dev:XXXXXXXXXX@10.66.229.171:5432/dwh",
"uuid": "97b97481-43c3-4181-94c5-b69eaaa1e11f", "uuid": "97b97481-43c3-4181-94c5-b69eaaa1e11f",
@@ -140,7 +140,7 @@ except Exception as e:
# [CONFIG] Определение исходного и целевого клиентов для миграции # [CONFIG] Определение исходного и целевого клиентов для миграции
# [COHERENCE_NOTE] Эти переменные задают конкретную миграцию. Для параметризации можно использовать аргументы командной строки. # [COHERENCE_NOTE] Эти переменные задают конкретную миграцию. Для параметризации можно использовать аргументы командной строки.
from_c = dev_client # Источник миграции from_c = dev_client # Источник миграции
to_c = sandbox_client # Цель миграции to_c = dev_client # Цель миграции
dashboard_slug = "FI0060" # Идентификатор дашборда для миграции dashboard_slug = "FI0060" # Идентификатор дашборда для миграции
# dashboard_id = 53 # ID не нужен, если есть slug # dashboard_id = 53 # ID не нужен, если есть slug
@@ -161,12 +161,12 @@ try:
# Экспорт дашборда во временную директорию ИЛИ чтение с диска # Экспорт дашборда во временную директорию ИЛИ чтение с диска
# [COHERENCE_NOTE] В текущем коде закомментирован экспорт и используется локальный файл. # [COHERENCE_NOTE] В текущем коде закомментирован экспорт и используется локальный файл.
# Для полноценной миграции следует использовать export_dashboard(). # Для полноценной миграции следует использовать export_dashboard().
zip_content, filename = from_c.export_dashboard(dashboard_id) # Предпочтительный путь для реальной миграции #zip_content, filename = from_c.export_dashboard(dashboard_id) # Предпочтительный путь для реальной миграции
# [DEBUG] Использование файла с диска для тестирования миграции # [DEBUG] Использование файла с диска для тестирования миграции
#zip_db_path = r"C:\Users\VolobuevAA\Downloads\dashboard_export_20250616T174203.zip" zip_db_path = r"C:\Users\VolobuevAA\Downloads\dashboard_export_20250704T082538.zip"
#logger.warning(f"[WARN] Используется ЛОКАЛЬНЫЙ файл дашборда для миграции: {zip_db_path}. Это может привести к некогерентности, если файл устарел.") logger.warning(f"[WARN] Используется ЛОКАЛЬНЫЙ файл дашборда для миграции: {zip_db_path}. Это может привести к некогерентности, если файл устарел.")
#zip_content, filename = read_dashboard_from_disk(zip_db_path, logger=logger) zip_content, filename = read_dashboard_from_disk(zip_db_path, logger=logger)
# [ANCHOR] SAVE_AND_UNPACK # [ANCHOR] SAVE_AND_UNPACK
# Сохранение и распаковка во временную директорию # Сохранение и распаковка во временную директорию

View File

@@ -179,9 +179,9 @@ def inspect_datasets(client: SupersetClient):
logger = SupersetLogger( level=logging.DEBUG,console=True) logger = SupersetLogger( level=logging.DEBUG,console=True)
clients = setup_clients(logger) clients = setup_clients(logger)
# Поиск всех таблиц с 'select' в датасете # Поиск всех таблиц в датасете
results = search_datasets( results = search_datasets(
client=clients['sbx'], client=clients['dev'],
search_pattern=r'dm_view\.counterparty', search_pattern=r'dm_view\.counterparty',
search_fields=["sql"], search_fields=["sql"],
logger=logger logger=logger

View File

@@ -414,7 +414,7 @@ class SupersetClient:
else: else:
self.logger.debug( self.logger.debug(
"[DEBUG] Имя файла экспорта получено из заголовков.", "[DEBUG] Имя файла экспорта получено из заголовков.",
extra={"filename": filename, "dashboard_id": dashboard_id} extra={"header_filename": filename, "dashboard_id": dashboard_id}
) )
return filename return filename

View File

@@ -23,6 +23,7 @@ from contextlib import contextmanager
# [IMPORTS] Third-party # [IMPORTS] Third-party
import yaml import yaml
import shutil import shutil
import zlib
import tempfile import tempfile
from datetime import datetime from datetime import datetime
@@ -164,11 +165,31 @@ def read_dashboard_from_disk(
# [SECTION] Archive Management # [SECTION] Archive Management
def calculate_crc32(file_path: Path) -> str:
"""[HELPER] Calculates the CRC32 checksum of a file.
@pre:
- file_path must be a valid path to a file.
@post:
- Returns the CRC32 checksum as a hexadecimal string.
@raise:
- FileNotFoundError: If the file does not exist.
- Exception: For any other file I/O errors.
"""
try:
with open(file_path, 'rb') as f:
crc32_value = zlib.crc32(f.read())
return hex(crc32_value)[2:].zfill(8) # Convert to hex string, remove "0x", and pad with zeros
except FileNotFoundError:
raise FileNotFoundError(f"File not found: {file_path}")
except Exception as e:
raise Exception(f"Error calculating CRC32 for {file_path}: {str(e)}")
def archive_exports( def archive_exports(
output_dir: str, output_dir: str,
daily_retention: int = 7, daily_retention: int = 7,
weekly_retention: int = 4, weekly_retention: int = 4,
monthly_retention: int = 12, monthly_retention: int = 12,
deduplicate: bool = False,
logger: Optional[SupersetLogger] = None logger: Optional[SupersetLogger] = None
) -> None: ) -> None:
"""[CONTRACT] Управление архивом экспортированных дашбордов """[CONTRACT] Управление архивом экспортированных дашбордов
@@ -178,30 +199,56 @@ def archive_exports(
@post: @post:
- Сохраняет файлы согласно политике хранения - Сохраняет файлы согласно политике хранения
- Удаляет устаревшие архивы - Удаляет устаревшие архивы
- Сохраняет логическую структуру каталогов - Логирует все действия
@raise:
- ValueError: Если retention параметры некорректны
- Exception: При любых других ошибках
""" """
logger = logger or SupersetLogger(name="fileio", console=False) logger = logger or SupersetLogger(name="fileio", console=False)
logger.info(f"[ARCHIVE] Старт очистки архивов в {output_dir}") logger.info(f"[ARCHIVE] Starting archive cleanup in {output_dir}. Deduplication: {deduplicate}")
# [VALIDATION] Проверка параметров # [VALIDATION] Проверка параметров
if not all(isinstance(x, int) and x >= 0 for x in [daily_retention, weekly_retention, monthly_retention]): if not all(isinstance(x, int) and x >= 0 for x in [daily_retention, weekly_retention, monthly_retention]):
raise ValueError("[CONFIG_ERROR] Значения retention должны быть положительными") raise ValueError("[CONFIG_ERROR] Retention values must be positive integers.")
checksums = {} # Dictionary to store checksums and file paths
try: try:
export_dir = Path(output_dir) export_dir = Path(output_dir)
files_with_dates = []
# [PROCESSING] Сбор данных о файлах if not export_dir.exists():
logger.error(f"[ARCHIVE_ERROR] Directory does not exist: {export_dir}")
raise FileNotFoundError(f"Directory not found: {export_dir}")
# [PROCESSING] Сбор информации о файлах
files_with_dates = []
for file in export_dir.glob("*.zip"): for file in export_dir.glob("*.zip"):
try: try:
timestamp_str = file.stem.split('_')[-1].split('T')[0] timestamp_str = file.stem.split('_')[-1].split('T')[0]
file_date = datetime.strptime(timestamp_str, "%Y%m%d").date() file_date = datetime.strptime(timestamp_str, "%Y%m%d").date()
logger.debug(f"[DATE_PARSE] Файл {file.name} добавлен к анализу очистки (массив files_with_dates)")
except (ValueError, IndexError): except (ValueError, IndexError):
file_date = datetime.fromtimestamp(file.stat().st_mtime).date() file_date = datetime.fromtimestamp(file.stat().st_mtime).date()
logger.warning(f"[DATE_PARSE] Используется дата модификации для {file.name}") logger.warning(f"[DATE_PARSE] Using modification date for {file.name}")
files_with_dates.append((file, file_date)) files_with_dates.append((file, file_date))
# [DEDUPLICATION]
if deduplicate:
logger.info("[DEDUPLICATION] Starting checksum-based deduplication.")
for file in files_with_dates:
file_path = file[0]
try:
crc32_checksum = calculate_crc32(file_path)
if crc32_checksum in checksums:
# Duplicate found, delete the older file
logger.warning(f"[DEDUPLICATION] Duplicate found: {file_path}. Deleting.")
file_path.unlink()
else:
checksums[crc32_checksum] = file_path
except Exception as e:
logger.error(f"[DEDUPLICATION_ERROR] Error processing {file_path}: {str(e)}", exc_info=True)
# [PROCESSING] Применение политик хранения # [PROCESSING] Применение политик хранения
keep_files = apply_retention_policy( keep_files = apply_retention_policy(
files_with_dates, files_with_dates,
@@ -212,13 +259,20 @@ def archive_exports(
) )
# [CLEANUP] Удаление устаревших файлов # [CLEANUP] Удаление устаревших файлов
deleted_count = 0
for file, _ in files_with_dates: for file, _ in files_with_dates:
if file not in keep_files: if file not in keep_files:
file.unlink(missing_ok=True) try:
logger.info(f"[FILE_REMOVED] Удален архив: {file.name}") file.unlink()
deleted_count += 1
logger.info(f"[FILE_REMOVED] Deleted archive: {file.name}")
except OSError as e:
logger.error(f"[FILE_ERROR] Error deleting {file.name}: {str(e)}", exc_info=True)
logger.info(f"[ARCHIVE_RESULT] Cleanup completed. Deleted {deleted_count} archives.")
except Exception as e: except Exception as e:
logger.error(f"[ARCHIVE_ERROR] Ошибка обработки архивов: {str(e)}", exc_info=True) logger.error(f"[ARCHIVE_ERROR] Critical error during archive cleanup: {str(e)}", exc_info=True)
raise raise
def apply_retention_policy( def apply_retention_policy(