138 lines
4.2 KiB
Python
138 lines
4.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
main.py — Точка входа. Меню выбора режима.
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
import argparse
|
|
|
|
_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 as cerror, 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:
|
|
st = state.load_state()
|
|
has_resume = bool(st.get("mode") and st.get("stage") and st.get("stage") != "INIT")
|
|
menu(has_resume=has_resume)
|
|
try:
|
|
choice = prompt("Ваш выбор", default="0").strip()
|
|
except SystemExit:
|
|
break
|
|
if choice == "1":
|
|
from source.source import run_source_mode
|
|
try:
|
|
run_source_mode()
|
|
except KeyboardInterrupt:
|
|
warn("Прервано")
|
|
except Exception as e:
|
|
cerror(f"Ошибка: {e}")
|
|
sys.exit(1)
|
|
break
|
|
elif choice == "2":
|
|
from target.target import run_target_mode
|
|
try:
|
|
run_target_mode()
|
|
except KeyboardInterrupt:
|
|
warn("Прервано")
|
|
except Exception as e:
|
|
cerror(f"Ошибка: {e}")
|
|
sys.exit(1)
|
|
break
|
|
elif choice == "3":
|
|
if not has_resume:
|
|
warn("Нет сохранённого состояния для продолжения")
|
|
continue
|
|
try:
|
|
_do_resume()
|
|
except KeyboardInterrupt:
|
|
warn("Прервано")
|
|
break
|
|
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)
|
|
try:
|
|
fsm.resume_from(stage)
|
|
except Exception as e:
|
|
cerror(f"Ошибка при resume: {e}")
|
|
|
|
|
|
def _do_status_logs():
|
|
from core import state as stmod
|
|
while True:
|
|
print("\n")
|
|
print(" 1 Показать статус")
|
|
print(" 2 Показать лог последней ошибки")
|
|
print(" 0 Назад")
|
|
try:
|
|
c = prompt("Выбор", default="0").strip()
|
|
except SystemExit:
|
|
break
|
|
if c == "1":
|
|
stmod.show_status()
|
|
elif c == "2":
|
|
stmod.show_logs()
|
|
elif c == "0":
|
|
break
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|