init: docker-migrate universal migration tool

core/main.py

@@ -0,0 +1,106 @@
+# -*- coding: utf-8 -*-
+"""
+main.py — Точка входа. Меню выбора режима.
+"""
+
+import sys
+import os
+import argparse
+
+# Добавляем корень проекта в PATH
+_PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+if _PROJECT_ROOT not in sys.path:
+    sys.path.insert(0, _PROJECT_ROOT)
+
+from core.color import banner, menu, prompt, info, error, warn
+from core import state
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Docker Service Migration Tool")
+    parser.add_argument("--mode", choices=["source", "target"], help="Режим работы")
+    parser.add_argument("--resume", action="store_true", help="Продолжить после ошибки")
+    parser.add_argument("--status", action="store_true", help="Показать статус")
+    parser.add_argument("--logs", action="store_true", help="Показать логи ошибки")
+    parser.add_argument("--remote-dir", help="Директория архива на target")
+    parser.add_argument("--no-color", action="store_true", help="Отключить цвета")
+    args = parser.parse_args()
+
+    if args.no_color:
+        os.environ["NO_COLOR"] = "1"
+
+    if args.status:
+        state.show_status()
+        sys.exit(0)
+
+    if args.logs:
+        state.show_logs()
+        sys.exit(0)
+
+    if args.resume:
+        _do_resume()
+        sys.exit(0)
+
+    if args.mode == "source":
+        from source.source import run_source_mode
+        run_source_mode()
+        sys.exit(0)
+    elif args.mode == "target":
+        if args.remote_dir:
+            state.set_stage("INIT", mode="target", target_remote_dir=args.remote_dir)
+        from target.target import run_target_mode
+        run_target_mode()
+        sys.exit(0)
+
+    # Интерактивный режим
+    banner()
+    while True:
+        menu()
+        choice = prompt("Ваш выбор").strip()
+        if choice == "1":
+            from source.source import run_source_mode
+            run_source_mode()
+        elif choice == "2":
+            from target.target import run_target_mode
+            run_target_mode()
+        elif choice == "3":
+            _do_resume()
+        elif choice == "4":
+            _do_status_logs()
+        elif choice == "0":
+            info("Выход")
+            sys.exit(0)
+        else:
+            warn("Неверный выбор")
+
+
+def _do_resume():
+    from core.fsm import FSM
+    st = state.load_state()
+    stage = st.get("stage")
+    mode = st.get("mode")
+    if not mode or stage == "INIT":
+        info("Нет сохранённого состояния для продолжения")
+        return
+    fsm = FSM(mode=mode)
+    fsm.resume_from(stage)
+
+
+def _do_status_logs():
+    from core import state as stmod
+    while True:
+        print("\n")
+        print("  1  Показать статус")
+        print("  2  Показать лог последней ошибки")
+        print("  0  Назад")
+        c = prompt("Выбор").strip()
+        if c == "1":
+            stmod.show_status()
+        elif c == "2":
+            stmod.show_logs()
+        elif c == "0":
+            break
+
+
+if __name__ == "__main__":
+    main()