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