修复输出格式解析bug：.jpg和jpg能正确转换

main.py

@@ -1,7 +1,7 @@
 import os
 import sys
-from watermark import watermark_image, increment_count
-from config import OUTPUT_DIR, SUPPORTED_INPUT_EXTENSIONS, SUPPORTED_OUTPUT_EXTENSIONS, WORK_DIR
+from watermark import watermark_image_tiled, increment_count, _parse_output_spec
+from config import SUPPORTED_INPUT_EXTENSIONS, SUPPORTED_OUTPUT_EXTENSIONS, WORK_DIR
 
 
 def print_banner():
@@ -12,10 +12,12 @@ def print_banner():
     print("  输出: jpg / png / bmp / gif / tiff / webp")
     print("=" * 50)
     print("命令:")
-    print("  <图片路径> \"水印内容\"      加水印，保持原格式输出")
+    print("  <图片路径> \"水印内容\"       加水印，输出到原目录")
     print("  <图片路径>>.png \"内容\"     加水印并转换格式")
-    print("  /exit 或 exit               退出程序")
-    print("  /get 或 get                 显示已处理图片统计")
+    print("  <图片路径>>./output.png      输出到指定目录")
+    print("  <图片路径>>../output.png     输出到上级目录")
+    print("  /exit 或 exit                 退出程序")
+    print("  /get 或 get                   显示已处理图片统计")
     print()
 
 
@@ -31,35 +33,53 @@ def print_stats():
 
 
 def process_single(image_arg, text):
-    """处理单张图片：可选格式转换，添加水印，计数+1。"""
-    # 去掉首尾引号（兼容命令行传入带引号的路径）
+    """处理单张图片：可选格式转换/输出路径，添加平铺水印，计数+1。"""
+    # 去掉首尾引号
     image_arg = image_arg.strip().strip('"').strip("'")
     text = text.strip().strip('"').strip("'")
 
-    if ">" not in image_arg:
-        output_ext = None
-    else:
+    output_spec = None
+    if ">" in image_arg:
         parts = image_arg.rsplit(">", 1)
-        image_arg, out_ext = parts[0].strip(), parts[1].strip().lstrip(".")
-        if out_ext.lower() not in SUPPORTED_INPUT_EXTENSIONS:
-            print(f"不支持的输出格式: .{out_ext}，支持的格式: {', '.join(sorted(SUPPORTED_OUTPUT_EXTENSIONS))}")
-            return
-        output_ext = f".{out_ext.lower()}"
+        image_arg = parts[0].strip()
+        output_spec = parts[1].strip()
 
+    # 验证输入路径
     input_path = image_arg
     if not os.path.isfile(input_path):
-        print(f"找不到输入图片: {input_path}")
+        print(f"❌ 找不到输入图片: {input_path}")
         return
 
     _, ext = os.path.splitext(input_path)
     if ext.lower() not in SUPPORTED_INPUT_EXTENSIONS:
-        print(f"不支持的输入格式: {ext}，支持的格式: {', '.join(sorted(SUPPORTED_INPUT_EXTENSIONS))}")
+        print(f"❌ 不支持的输入格式: {ext}，支持的格式: {', '.join(sorted(SUPPORTED_INPUT_EXTENSIONS))}")
         return
 
-    output_path = watermark_image(input_path, text, output_ext=output_ext)
-    total = increment_count()
-    print(f"✅ 已保存: {output_path}")
-    print(f"📊 累计处理图片: {total} 张")
+    # 处理输出规格
+    output_path = None
+    output_ext = None
+    if output_spec:
+        try:
+            output_path, output_ext = _parse_output_spec(output_spec, input_path)
+        except ValueError as e:
+            print(f"❌ {e}")
+            return
+    else:
+        # 无输出规格，保持原格式，输出到原目录
+        _, output_ext = os.path.splitext(input_path)
+        input_dir = os.path.dirname(os.path.abspath(input_path))
+        base_name = os.path.basename(input_path)
+        name_no_ext = os.path.splitext(base_name)[0]
+        output_path = os.path.join(input_dir, f"{name_no_ext}_watermarked{output_ext.lower()}")
+
+    # 添加平铺水印
+    try:
+        final_path = watermark_image_tiled(input_path, text, output_path=output_path, output_ext=output_ext)
+        total = increment_count()
+        print(f"✅ 已保存: {final_path}")
+        print(f"📊 累计处理图片: {total} 张")
+    except Exception as e:
+        print(f"❌ 处理失败: {e}")
 
 
 def main():
@@ -88,26 +108,42 @@ def main():
         elif cmd in ("/help", "help"):
             print_banner()
         elif ">" in cmd:
-            # 格式转换模式: "图片路径>.png 水印内容"  或  "图片路径>PNG 水印内容"
+            # 格式转换模式: "图片路径>.png 水印内容" 或 "图片路径>./output.png 水印内容"
             parts = cmd.split(">", 1)
             image_part = parts[0].strip()
             rest = parts[1].strip()
             if not rest:
-                print("缺少水印内容，请重试。输入 /help 查看帮助。")
+                print("❌ 缺少输出规格或水印内容，请重试。")
                 continue
-            # rest 中第一个空格前是格式，后面是水印文本
+
+            # 分析 rest 部分：可能是 "格式 文本" 或 "路径 文本"
             tokens = rest.split(None, 1)
-            if len(tokens) < 2:
-                print("格式参数后缺少水印内容，请重试。输入 /help 查看帮助。")
+            if len(tokens) == 2:
+                output_spec, text = tokens
+                full_arg = f"{image_part}>{output_spec}"
+            elif len(tokens) == 1:
+                # 只有一个 token，判断是否是路径（包含 . 或 / 或 \）
+                candidate = tokens[0]
+                if "." in candidate or "/" in candidate or "\\" in candidate:
+                    # 可能是路径，但缺少水印内容
+                    print("❌ 缺少水印内容，请重试。")
+                    continue
+                else:
+                    # 是格式名
+                    output_spec = candidate
+                    # 继续读取后续内容作为水印文本（如果有）
+                    print("❌ 缺少水印内容，请重试。")
+                    continue
+            else:
+                print("❌ 格式错误，请重试。")
                 continue
-            final_format, text = tokens
-            full_arg = f"{image_part}>{final_format}"
+
             process_single(full_arg, text)
         else:
             # 普通模式: "图片路径 水印内容"
             parts = cmd.split(None, 1)
             if len(parts) < 2:
-                print("格式错误，需要提供图片路径和水印内容。输入 /help 查看帮助。")
+                print("❌ 格式错误，需要提供图片路径和水印内容。")
                 continue
             image_arg, text = parts
             process_single(image_arg, text)