diff --git a/main.py b/main.py index f6b2b22..e8e69fd 100644 --- a/main.py +++ b/main.py @@ -4,27 +4,24 @@ from watermark import watermark_image, increment_count from config import OUTPUT_DIR, SUPPORTED_INPUT_EXTENSIONS, SUPPORTED_OUTPUT_EXTENSIONS, WORK_DIR +def print_banner(): + """启动时显示完整帮助菜单。""" + print("=" * 50) + print(" imgwatermark - 图像水印工具 v1.0") + print(" 支持: .jpg .png .bmp .gif .tiff .webp") + print(" 输出: jpg / png / bmp / gif / tiff / webp") + print("=" * 50) + print("命令:") + print(" <图片路径> \"水印内容\" 加水印,保持原格式输出") + print(" <图片路径>>.png \"内容\" 加水印并转换格式(格式可选:jpg/png/bmp/gif/tiff/webp)") + print(" /exit 或 exit 退出程序") + print(" /get 或 get 显示已处理图片统计") + print() + + def print_help(): - help_text = """ -======================================== - imgwatermark - 图像水印工具 v1.0 - 支持: .jpg .png .bmp .gif .tiff .webp - 输出: jpg / png / bmp / gif / tiff / webp -======================================== - -命令: - <图片路径> \"水印内容\" 加水印,保持原格式输出 - <图片路径>>.png \"内容\" 加水印并转换格式(格式可选:jpg/png/bmp/gif/tiff/webp) - /exit 或 exit 退出程序 - /get 或 get 显示已处理图片统计 - /help 或 help 显示本帮助信息 - -示例: - D:\\img\\photo.jpg \"版权所有\" - D:\\img\\photo.jpg>.png \"2026\" - /get -""" - print(help_text.strip()) + """交互式模式下重复显示帮助。""" + print_banner() def print_stats(): @@ -76,10 +73,7 @@ def main(): process_single(sys.argv[1], sys.argv[2]) return - print("=" * 40) - print(" imgwatermark - 图像水印工具 v1.0") - print(" 输入 /help 查看帮助") - print("=" * 40) + print_banner() while True: try: cmd = input("\n> ").strip() @@ -97,7 +91,7 @@ def main(): elif cmd in ("/get", "get"): print_stats() elif cmd in ("/help", "help"): - print_help() + print_banner() elif ">" in cmd: # 格式转换模式: "图片路径>.png 水印内容" 或 "图片路径>PNG 水印内容" parts = cmd.split(">", 1)