83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
from __future__ import annotations
|
||
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
from ad_user_creator.main import execute_command
|
||
|
||
|
||
def _ask_yes_no(prompt: str, default: bool) -> bool:
|
||
default_text = "Y/n" if default else "y/N"
|
||
value = input(f"{prompt} [{default_text}]: ").strip().lower()
|
||
if not value:
|
||
return default
|
||
return value in {"y", "yes", "1", "true"}
|
||
|
||
|
||
def _clean_path(path_text: str) -> str:
|
||
text = path_text.strip()
|
||
if len(text) >= 2 and ((text[0] == "'" and text[-1] == "'") or (text[0] == '"' and text[-1] == '"')):
|
||
text = text[1:-1]
|
||
return text
|
||
|
||
|
||
def _run_interactive(config_path: str) -> int:
|
||
dry_run = _ask_yes_no("是否启用 dry-run（不写入 LDAP）", default=False)
|
||
return execute_command(command="interactive", config_path=config_path, dry_run=dry_run)
|
||
|
||
|
||
def _run_batch(config_path: str) -> int:
|
||
input_path = _clean_path(input("请输入批量文件路径 (.csv/.xlsx): "))
|
||
if not input_path:
|
||
print("未提供文件路径。")
|
||
return 1
|
||
suffix = Path(input_path).suffix.lower()
|
||
if suffix not in {".csv", ".xlsx"}:
|
||
print(f"文件格式错误，仅支持 .csv 或 .xlsx: {input_path}")
|
||
return 1
|
||
|
||
dry_run = _ask_yes_no("是否启用 dry-run（不写入 LDAP）", default=False)
|
||
continue_on_error = _ask_yes_no("遇到错误是否继续处理后续记录", default=True)
|
||
return execute_command(
|
||
command="batch",
|
||
config_path=config_path,
|
||
dry_run=dry_run,
|
||
input_path=input_path,
|
||
continue_on_error=continue_on_error,
|
||
)
|
||
|
||
|
||
def menu() -> int:
|
||
config_path = "config/config.yaml"
|
||
while True:
|
||
print("\n=== AD User Creator ===")
|
||
print(f"当前配置文件: {config_path}")
|
||
print("1) 交互式创建")
|
||
print("2) 批量导入")
|
||
print("3) 修改配置文件路径")
|
||
print("q) 退出")
|
||
choice = input("请选择模式: ").strip().lower()
|
||
|
||
if choice == "1":
|
||
code = _run_interactive(config_path)
|
||
print(f"执行完成，返回码: {code}")
|
||
elif choice == "2":
|
||
code = _run_batch(config_path)
|
||
print(f"执行完成，返回码: {code}")
|
||
elif choice == "3":
|
||
new_path = _clean_path(input("请输入新的 config.yaml 路径: "))
|
||
if not new_path:
|
||
print("配置路径未变更。")
|
||
else:
|
||
config_path = new_path
|
||
print(f"已更新配置路径: {config_path}")
|
||
elif choice in {"q", "quit", "exit"}:
|
||
print("已退出。")
|
||
return 0
|
||
else:
|
||
print("无效选项，请输入 1/2/3/q。")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(menu())
|