upadte cript

script/image_mark.py

@@ -3,11 +3,7 @@ import os
 import sys
 import random
 from PIL import Image,ImageDraw,ImageFont
-def add_text_to_image(image, text):
-    mode = 'RGB'
-    if image.endswith(".png"):
-        mode = "RGBA"
-
+def add_text_to_image(image, text , mode):
     #水印字体随机大小
     font_size = random.randint(30,60)
     font = ImageFont.truetype('Arial.ttf', font_size)
@@ -34,8 +30,12 @@ def add_text_to_image(image, text):
     return image_with_text
 
 if __name__ == '__main__':
+    mode = 'RGB'
+    if sys.argv[1].endswith(".png"):
+        mode = "RGBA"
+
     img = Image.open(sys.argv[1])
-    im_after = add_text_to_image(img, sys.argv[2])
+    im_after = add_text_to_image(img, sys.argv[2],mode)
     output = sys.argv[1].split(".");
     output = output[0] + "_mark" + "." + output[1]
     print(output)