en2zmax / mig_s.py

#!/usr/bin/env python3 """ Скрипт миграции tmux сессий в новую структуру инвентарей """ import os import re import shutil from pathlib import Path class SessionMigrator: def __init__(self): self.source_dir = Path.home() / '.config' / 'tmux' / 'sessions' self.target_dir = Path.home() / '.config' / 'tmux-sessions' / 'inventory' def parse_session_file(self, file_path): """Парсит файл сессии и извлекает данные""" with open(file_path, 'r', encoding='utf-8') as f: content = f.read() # Базовые данные data = { 'filename': file_path.name, 'session_name': None, 'sshpass_path': None, 'servers': [] } # Парсим имя сессии session_match = re.search(r'session_name=(["\']?)([^"\'\n]+)\1', content) if session_match: data['session_name'] = session_match.group(2).strip() # Парсим путь к паролю sshpass_match = re.search(r'sshpass=\$\(pass ([^)\n]+)\)', content) if sshpass_match: data['sshpass_path'] = sshpass_match.group(1).strip() # Парсим серверы из массива servers servers_match = re.search(r'servers=\(([^)]+)\)', content, re.DOTALL) if servers_match: servers_content = servers_match.group(1) # Извлекаем строки серверов server_lines = re.findall(r'"([^"]+)"', servers_content) for line in server_lines: parts = line.strip().split(' ', 1) if len(parts) == 2: data['servers'].append(f'"{parts[0]} {parts[1]}"') # Обработка специального формата для Приморье if 'Приморье' in file_path.name: data = self._parse_primorye_format(content, data) return data def _parse_primorye_format(self, content, data): """Специальный парсинг для файла Приморье с другим форматом""" # Имя сессии session_match = re.search(r'SESSION=([^\n]+)', content) if session_match: data['session_name'] = session_match.group(1).strip() # Путь к паролю sshpass_match = re.search(r'SSHPASS=\$\(pass ([^)\n]+)\)', content) if sshpass_match: data['sshpass_path'] = sshpass_match.group(1).strip() # Извлекаем серверы из tmux new-window команд servers = [] window_matches = re.findall(r'tmux new-window[^>]+-n ([^\s]+)[^>]+"sshpass[^>]+ssh ([^"\s]+)"', content) for server_name, server_connect in window_matches: servers.append(f'"{server_name} {server_connect}"') data['servers'] = servers return data def generate_config_content(self, data): """Генерирует содержимое нового конфигурационного файла""" lines = [ f"# Конфигурация для проекта {data['filename'].replace('.session', '')}", f"SESSION_NAME=\"{data['session_name']}\"", f"SSHPASS_PATH=\"{data['sshpass_path']}\"", "", "# Список серверов в формате \"имя_вкладки подключение\"", "SERVERS=(" ] # Добавляем серверы for server in data['servers']: lines.append(f" {server}") lines.extend([ ")", "" ]) return '\n'.join(lines) def sanitize_filename(self, filename): """Очищает имя файла от недопустимых символов""" # Убираем расширение .session clean_name = filename.replace('.session', '') # Заменяем пробелы и другие проблемные символы clean_name = re.sub(r'[^\w\-_.]', '_', clean_name) return clean_name + '.conf' def migrate(self): """Основная функция миграции""" print("🚀 Начинаем миграцию tmux сессий...") print(f"📁 Исходный каталог: {self.source_dir}") print(f"📁 Целевой каталог: {self.target_dir}") print() # Создаем целевой каталог если не существует self.target_dir.mkdir(parents=True, exist_ok=True) # Получаем список файлов сессий session_files = list(self.source_dir.glob('*.session')) if not session_files: print("❌ Не найдено файлов сессий для миграции") return migrated_count = 0 errors = [] for session_file in session_files: try: print(f"📖 Обрабатываем: {session_file.name}") # Парсим файл data = self.parse_session_file(session_file) # Проверяем что все необходимые данные есть if not data['session_name']: print(f" ⚠️ Предупреждение: не найдено имя сессии в {session_file.name}") if not data['sshpass_path']: print(f" ⚠️ Предупреждение: не найден путь к паролю в {session_file.name}") if not data['servers']: print(f" ⚠️ Предупреждение: не найдены серверы в {session_file.name}") # Генерируем новое имя файла new_filename = self.sanitize_filename(session_file.name) new_file_path = self.target_dir / new_filename # Генерируем содержимое config_content = self.generate_config_content(data) # Сохраняем файл with open(new_file_path, 'w', encoding='utf-8') as f: f.write(config_content) print(f" ✅ Создан: {new_filename}") print(f" Session: {data['session_name']}") print(f" Servers: {len(data['servers'])}") print(f" Pass path: {data['sshpass_path']}") print() migrated_count += 1 except Exception as e: error_msg = f"Ошибка обработки {session_file.name}: {str(e)}" print(f" ❌ {error_msg}") errors.append(error_msg) # Выводим итоги print("=" * 50) print(f"📊 Миграция завершена!") print(f"✅ Успешно мигрировано: {migrated_count}/{len(session_files)}") if errors: print(f"❌ Ошибки ({len(errors)}):") for error in errors: print(f" - {error}") # Показываем созданные файлы print(f"\n📁 Созданные файлы в {self.target_dir}:") for conf_file in sorted(self.target_dir.glob('*.conf')): print(f" 📄 {conf_file.name}") print(f"\n🎯 Теперь вы можете использовать новый лаунчер:") print(f" tmux-sessions # Меню выбора") print(f" tmux-sessions --list # Список проектов") def main(): migrator = SessionMigrator() # Проверяем существует ли исходный каталог if not migrator.source_dir.exists(): print(f"❌ Исходный каталог не существует: {migrator.source_dir}") return # Запускаем миграцию migrator.migrate() if __name__ == '__main__': main()