Fix multiple watermark

This commit is contained in:
yoshiko2
2023-06-29 03:31:00 +08:00
parent 185589582f
commit 3a4fe159db

View File

@@ -528,10 +528,13 @@ def add_mark_thread(pic_path, cn_sub, leak, uncensored, hack, _4k, iso):
count = (count + 1) % 4 count = (count + 1) % 4
if uncensored: if uncensored:
add_to_pic(pic_path, img_pic, size, count, 3) add_to_pic(pic_path, img_pic, size, count, 3)
count = (count + 1) % 4
if hack: if hack:
add_to_pic(pic_path, img_pic, size, count, 4) add_to_pic(pic_path, img_pic, size, count, 4)
count = (count + 1) % 4
if _4k: if _4k:
add_to_pic(pic_path, img_pic, size, count, 5) add_to_pic(pic_path, img_pic, size, count, 5)
count = (count + 1) % 4
if iso: if iso:
add_to_pic(pic_path, img_pic, size, count, 6) add_to_pic(pic_path, img_pic, size, count, 6)
img_pic.close() img_pic.close()
@@ -550,6 +553,8 @@ def add_to_pic(pic_path, img_pic, size, count, mode):
pngpath = "Img/HACK.png" pngpath = "Img/HACK.png"
elif mode == 5: elif mode == 5:
pngpath = "Img/4K.png" pngpath = "Img/4K.png"
elif mode == 6:
pngpath = "Img/ISO.png"
else: else:
print('[-]Error: watermark image param mode invalid!') print('[-]Error: watermark image param mode invalid!')
return return