bacow/AV_Data_Capture
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

simplify again

Browse Source
This commit is contained in:
lededev
2021-06-14 13:29:58 +08:00
parent 8020d24226
commit fa49a14a0f
1 changed files with 17 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
core.py
View File

@@ -588,28 +588,27 @@ def add_mark_thread(pic_path, cn_sub, leak, uncensored, conf):
def add_to_pic(pic_path, img_pic, size, count, mode): def add_to_pic(pic_path, img_pic, size, count, mode):
mark_pic_path = '' mark_pic_path = ''
pngpath = ''
if mode == 1:
pngpath = "Img/SUB.png"
elif mode == 2:
pngpath = "Img/LEAK.png"
elif mode == 3:
pngpath = "Img/UNCENSORED.png"
else:
print('[-]Error: watermark image param mode invalid!')
return
# 先找pyinstaller打包的图片 # 先找pyinstaller打包的图片
base_path = '' if hasattr(sys, '_MEIPASS') and os.path.isfile(os.path.join(getattr(sys, '_MEIPASS'), pngpath)):
if hasattr(sys, '_MEIPASS') and os.path.isfile(os.path.join(getattr(sys, '_MEIPASS'),"Img/SUB.png")): mark_pic_path = os.path.join(getattr(sys, '_MEIPASS'), pngpath)
base_path = getattr(sys, '_MEIPASS')
# 再找py脚本所在路径的图片 # 再找py脚本所在路径的图片
elif os.path.isfile(os.path.join(os.path.dirname(os.path.realpath(__file__)),"Img/SUB.png")): elif os.path.isfile(os.path.join(os.path.dirname(os.path.realpath(__file__)), pngpath)):
base_path = os.path.dirname(os.path.realpath(__file__)) mark_pic_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), pngpath)
if len(base_path) > 0 and os.path.isdir(base_path):
if mode == 1:
mark_pic_path = os.path.join(base_path, "Img/SUB.png")
elif mode == 2:
mark_pic_path = os.path.join(base_path, "Img/LEAK.png")
elif mode == 3:
mark_pic_path = os.path.join(base_path, "Img/UNCENSORED.png")
# 如果没有本地图片才通过网络下载 # 如果没有本地图片才通过网络下载
else: else:
if mode == 1: mark_pic_path = BytesIO(
mark_pic_path = BytesIO(get_html("https://raw.githubusercontent.com/yoshiko2/AV_Data_Capture/master/Img/SUB.png",return_type="content")) get_html("https://raw.githubusercontent.com/yoshiko2/AV_Data_Capture/master/" + pngpath,
elif mode == 2: return_type="content"))
mark_pic_path = BytesIO(get_html("https://raw.githubusercontent.com/yoshiko2/AV_Data_Capture/master/Img/LEAK.png",return_type="content"))
elif mode == 3:
mark_pic_path = BytesIO(get_html("https://raw.githubusercontent.com/yoshiko2/AV_Data_Capture/master/Img/UNCENSORED.png",return_type="content"))
img_subt = Image.open(mark_pic_path) img_subt = Image.open(mark_pic_path)
scroll_high = int(img_pic.height / size) scroll_high = int(img_pic.height / size)
scroll_wide = int(scroll_high * img_subt.width / img_subt.height) scroll_wide = int(scroll_high * img_subt.width / img_subt.height)