From a8f39201b9f074af43a4ec45d19ec019ec2cf285 Mon Sep 17 00:00:00 2001 From: unknown Date: Mon, 22 Nov 2021 21:59:28 +0800 Subject: [PATCH] =?UTF-8?q?Add=20=E7=A0=B4=E8=A7=A3=20watermark=20&=20Bigg?= =?UTF-8?q?er=20watermark=20image?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- core.py | 119 ++++++++++++++++++++++++++++++++------------------------ 1 file changed, 68 insertions(+), 51 deletions(-) diff --git a/core.py b/core.py index 00f0aa7..fd0e6d6 100644 --- a/core.py +++ b/core.py @@ -66,8 +66,8 @@ def get_info(json_data): # 返回json里的数据 return title, studio, year, outline, runtime, director, actor_photo, release, number, cover, trailer, website, series, label -def small_cover_check(path, number, cover_small, leak_word, c_word, filepath): - filename = f"{number}{leak_word}{c_word}-poster.jpg" +def small_cover_check(path, number, cover_small, leak_word, c_word, hack_word, filepath): + filename = f"{number}{leak_word}{c_word}{hack_word}-poster.jpg" download_file_with_filename(cover_small, filename, path, filepath) print('[+]Image Downloaded! ' + os.path.join(path, filename)) @@ -165,20 +165,20 @@ def download_file_with_filename(url, filename, path, filepath): moveFailedFolder(filepath) return -def trailer_download(trailer, leak_word, c_word, number, path, filepath): - if download_file_with_filename(trailer, number + leak_word + c_word + '-trailer.mp4', path, filepath) == 'failed': +def trailer_download(trailer, leak_word, c_word, hack_word, number, path, filepath): + if download_file_with_filename(trailer, number + leak_word + c_word + hack_word + '-trailer.mp4', path, filepath) == 'failed': return configProxy = config.getInstance().proxy() for i in range(configProxy.retry): - if file_not_exist_or_empty(path+'/' + number + leak_word + c_word + '-trailer.mp4'): + if file_not_exist_or_empty(path+'/' + number + leak_word + c_word + hack_word + '-trailer.mp4'): print('[!]Video Download Failed! Trying again. [{}/3]', i + 1) - download_file_with_filename(trailer, number + leak_word + c_word + '-trailer.mp4', path, filepath) + download_file_with_filename(trailer, number + leak_word + c_word + hack_word + '-trailer.mp4', path, filepath) continue else: break - if file_not_exist_or_empty(path + '/' + number + leak_word + c_word + '-trailer.mp4'): + if file_not_exist_or_empty(path + '/' + number + leak_word + c_word + hack_word + '-trailer.mp4'): return - print('[+]Video Downloaded!', path + '/' + number + leak_word + c_word + '-trailer.mp4') + print('[+]Video Downloaded!', path + '/' + number + leak_word + c_word + hack_word + '-trailer.mp4') # 剧照下载成功，否则移动到failed def extrafanart_download(data, path, number, filepath): @@ -247,8 +247,8 @@ def extrafanart_download_threadpool(url_list, save_dir, number): # 封面是否下载成功，否则移动到failed -def image_download(cover, number, leak_word, c_word, path, filepath): - filename = f"{number}{leak_word}{c_word}-fanart.jpg" +def image_download(cover, number, leak_word, c_word, hack_word, path, filepath): + filename = f"{number}{leak_word}{c_word}{hack_word}-fanart.jpg" full_filepath = os.path.join(path, filename) if config.getInstance().download_only_missing_images() and not file_not_exist_or_empty(full_filepath): return @@ -267,15 +267,15 @@ def image_download(cover, number, leak_word, c_word, path, filepath): if file_not_exist_or_empty(full_filepath): return print('[+]Image Downloaded!', full_filepath) - shutil.copyfile(full_filepath, os.path.join(path, f"{number}{leak_word}{c_word}-thumb.jpg")) + shutil.copyfile(full_filepath, os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}-thumb.jpg")) -def print_files(path, leak_word, c_word, naming_rule, part, cn_sub, json_data, filepath, tag, actor_list, liuchu, uncensored): +def print_files(path, leak_word, c_word, naming_rule, part, cn_sub, json_data, filepath, tag, actor_list, liuchu, uncensored, hack_word): title, studio, year, outline, runtime, director, actor_photo, release, number, cover, trailer, website, series, label = get_info(json_data) if config.getInstance().main_mode() == 3: # 模式3下，由于视频文件不做任何改变，.nfo文件必须和视频文件名称除后缀外完全一致，KODI等软件方可支持 nfo_path = str(Path(filepath).with_suffix('.nfo')) else: - nfo_path = os.path.join(path,f"{number}{part}{leak_word}{c_word}.nfo") + nfo_path = os.path.join(path,f"{number}{part}{leak_word}{c_word}{hack_word}.nfo") try: if not os.path.exists(path): try: @@ -303,9 +303,9 @@ def print_files(path, leak_word, c_word, naming_rule, part, cn_sub, json_data, f print(" " + outline + "", file=code) print(" " + str(runtime).replace(" ", "") + "", file=code) print(" " + director + "", file=code) - print(" " + number + leak_word + c_word + "-poster.jpg", file=code) - print(" " + number + leak_word + c_word + "-thumb.jpg", file=code) - print(" " + number + leak_word + c_word + '-fanart.jpg' + "", file=code) + print(" " + number + leak_word + c_word + hack_word + "-poster.jpg", file=code) + print(" " + number + leak_word + c_word + hack_word + "-thumb.jpg", file=code) + print(" " + number + leak_word + c_word + hack_word + '-fanart.jpg' + "", file=code) try: for key in actor_list: print(" ", file=code) @@ -321,6 +321,8 @@ def print_files(path, leak_word, c_word, naming_rule, part, cn_sub, json_data, f print(" 流出", file=code) if uncensored == 1: print(" 无码", file=code) + if hack_word != '': + print(" 破解", file=code) try: for i in tag: print(" " + i + "", file=code) @@ -333,6 +335,8 @@ def print_files(path, leak_word, c_word, naming_rule, part, cn_sub, json_data, f print(" 流出", file=code) if uncensored == 1: print(" 无码", file=code) + if hack_word != '': + print(" 破解", file=code) try: for i in tag: print(" " + i + "", file=code) @@ -361,8 +365,8 @@ def print_files(path, leak_word, c_word, naming_rule, part, cn_sub, json_data, f return -def cutImage(imagecut, path, number, leak_word, c_word): - fullpath_noext = os.path.join(path, f"{number}{leak_word}{c_word}") +def cutImage(imagecut, path, number, leak_word, c_word, hack_word): + fullpath_noext = os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}") if imagecut == 1: # 剪裁大封面 try: img = Image.open(fullpath_noext + '-fanart.jpg') @@ -372,7 +376,8 @@ def cutImage(imagecut, path, number, leak_word, c_word): img2 = img.crop((w / 1.9, 0, w, h)) img2.save(fullpath_noext + '-poster.jpg') print('[+]Image Cutted! ' + fullpath_noext + '-poster.jpg') - except: + except Exception as e: + print(e) print('[-]Cover cut failed!') elif imagecut == 0: # 复制封面 shutil.copyfile(fullpath_noext + '-fanart.jpg', fullpath_noext + '-poster.jpg') @@ -386,7 +391,7 @@ def cutImage(imagecut, path, number, leak_word, c_word): # leak 流出 参数值为 1 0 # uncensored 无码 参数值为 1 0 # ========================================================================加水印 -def add_mark(poster_path, thumb_path, cn_sub, leak, uncensored): +def add_mark(poster_path, thumb_path, cn_sub, leak, uncensored, hack): mark_type = '' if cn_sub: mark_type += ',字幕' @@ -394,15 +399,17 @@ def add_mark(poster_path, thumb_path, cn_sub, leak, uncensored): mark_type += ',流出' if uncensored: mark_type += ',无码' + if hack: + mark_type += ',破解' if mark_type == '': return - add_mark_thread(thumb_path, cn_sub, leak, uncensored) + add_mark_thread(thumb_path, cn_sub, leak, uncensored, hack) print('[+]Thumb Add Mark: ' + mark_type.strip(',')) - add_mark_thread(poster_path, cn_sub, leak, uncensored) + add_mark_thread(poster_path, cn_sub, leak, uncensored, hack) print('[+]Poster Add Mark: ' + mark_type.strip(',')) -def add_mark_thread(pic_path, cn_sub, leak, uncensored): - size = 14 +def add_mark_thread(pic_path, cn_sub, leak, uncensored, hack): + size = 9 img_pic = Image.open(pic_path) # 获取自定义位置，取余配合pos达到顺时针添加的效果 # 左上 0, 右上 1, 右下 2， 左下 3 @@ -415,6 +422,8 @@ def add_mark_thread(pic_path, cn_sub, leak, uncensored): count = (count + 1) % 4 if uncensored == 1 or uncensored == '1': add_to_pic(pic_path, img_pic, size, count, 3) + if hack == 1 or hack == '1': + add_to_pic(pic_path, img_pic, size, count, 4) img_pic.close() def add_to_pic(pic_path, img_pic, size, count, mode): @@ -426,6 +435,8 @@ def add_to_pic(pic_path, img_pic, size, count, mode): pngpath = "Img/LEAK.png" elif mode == 3: pngpath = "Img/UNCENSORED.png" + elif mode == 4: + pngpath = "Img/HACK.png" else: print('[-]Error: watermark image param mode invalid!') return @@ -456,12 +467,12 @@ def add_to_pic(pic_path, img_pic, size, count, mode): img_pic.save(pic_path, quality=95) # ========================结束================================= -def paste_file_to_folder(filepath, path, number, leak_word, c_word): # 文件路径，番号，后缀，要移动至的位置 +def paste_file_to_folder(filepath, path, number, leak_word, c_word, hack_word): # 文件路径，番号，后缀，要移动至的位置 filepath_obj = pathlib.Path(filepath) houzhui = filepath_obj.suffix file_parent_origin_path = str(filepath_obj.parent) try: - targetpath = os.path.join(path, f"{number}{leak_word}{c_word}{houzhui}") + targetpath = os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}{houzhui}") # 任何情况下都不要覆盖，以免遭遇数据源或者引擎错误导致所有文件得到同一个number，逐一 # 同名覆盖致使全部文件损失且不可追回的最坏情况 if os.path.exists(targetpath): @@ -498,7 +509,7 @@ def paste_file_to_folder(filepath, path, number, leak_word, c_word): # 文件 sub_filepath = sub_filepath.replace(subname, ".cht" + subname) subname = ".cht" + subname if os.path.isfile(sub_filepath): - shutil.move(sub_filepath, os.path.join(path, f"{number}{leak_word}{c_word}{subname}")) + shutil.move(sub_filepath, os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}{subname}")) print('[+]Sub moved!') return True @@ -514,13 +525,13 @@ def paste_file_to_folder(filepath, path, number, leak_word, c_word): # 文件 return -def paste_file_to_folder_mode2(filepath, path, multi_part, number, part, leak_word, c_word): # 文件路径，番号，后缀，要移动至的位置 +def paste_file_to_folder_mode2(filepath, path, multi_part, number, part, leak_word, c_word, hack_word): # 文件路径，番号，后缀，要移动至的位置 if multi_part == 1: number += part # 这时number会被附加上CD1后缀 filepath_obj = pathlib.Path(filepath) houzhui = filepath_obj.suffix file_parent_origin_path = str(filepath_obj.parent) - targetpath = os.path.join(path, f"{number}{part}{leak_word}{c_word}{houzhui}") + targetpath = os.path.join(path, f"{number}{part}{leak_word}{c_word}{hack_word}{houzhui}") if os.path.exists(targetpath): raise FileExistsError('File Exists on destination path, we will never overwriting.') try: @@ -533,7 +544,7 @@ def paste_file_to_folder_mode2(filepath, path, multi_part, number, part, leak_wo for subname in sub_res: sub_filepath = str(filepath_obj.with_suffix(subname)) if os.path.isfile(sub_filepath): # 字幕移动 - shutil.move(sub_filepath, os.path.join(path, f"{number}{part}{leak_word}{c_word}{subname}")) + shutil.move(sub_filepath, os.path.join(path, f"{number}{part}{leak_word}{c_word}{hack_word}{subname}")) print('[+]Sub moved!') print('[!]Success') return True @@ -587,6 +598,8 @@ def core_main(file_path, number_th, oCC): c_word = '' cn_sub = '' liuchu = '' + hack = '' + hack_word = '' filepath = file_path # 影片的路径 绝对路径 @@ -628,6 +641,10 @@ def core_main(file_path, number_th, oCC): else: leak = 0 + if 'hack'.upper() in str(filepath).upper() or '破解' in filepath: + hack = 1 + hack_word = "-hack" + # 调试模式检测 if conf.debug(): debug_print(json_data) @@ -647,16 +664,16 @@ def core_main(file_path, number_th, oCC): # 检查小封面, 如果image cut为3，则下载小封面 if imagecut == 3: - small_cover_check(path, number, json_data.get('cover_small'), leak_word, c_word, filepath) + small_cover_check(path, number, json_data.get('cover_small'), leak_word, c_word, hack_word, filepath) # creatFolder会返回番号路径 - image_download( json_data.get('cover'), number, leak_word, c_word, path, filepath) + image_download( json_data.get('cover'), number, leak_word, c_word, hack_word, path, filepath) if not multi_part or part.lower() == '-cd1': try: # 下载预告片 if conf.is_trailer() and json_data.get('trailer'): - trailer_download(json_data.get('trailer'), leak_word, c_word, number, path, filepath) + trailer_download(json_data.get('trailer'), leak_word, c_word, hack_word, number, path, filepath) except: pass try: @@ -667,29 +684,29 @@ def core_main(file_path, number_th, oCC): pass # 裁剪图 - cutImage(imagecut, path, number, leak_word, c_word) + cutImage(imagecut, path, number, leak_word, c_word, hack_word) # 添加水印 - poster_path = os.path.join(path, f"{number}{leak_word}{c_word}-poster.jpg") - thumb_path = os.path.join(path, f"{number}{leak_word}{c_word}-thumb.jpg") + poster_path = os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}-poster.jpg") + thumb_path = os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}-thumb.jpg") if conf.is_watermark(): - add_mark(poster_path, thumb_path, cn_sub, leak, uncensored) + add_mark(poster_path, thumb_path, cn_sub, leak, uncensored, hack) # 移动电影 - paste_file_to_folder(filepath, path, number, leak_word, c_word) + paste_file_to_folder(filepath, path, number, leak_word, c_word, hack_word) # 最后输出.nfo元数据文件，以完成.nfo文件创建作为任务成功标志 - print_files(path, leak_word, c_word, json_data.get('naming_rule'), part, cn_sub, json_data, filepath, tag, json_data.get('actor_list'), liuchu, uncensored) + print_files(path, leak_word, c_word, json_data.get('naming_rule'), part, cn_sub, json_data, filepath, tag, json_data.get('actor_list'), liuchu, uncensored, hack_word) elif conf.main_mode() == 2: # 创建文件夹 path = create_folder(json_data) # 移动文件 - paste_file_to_folder_mode2(filepath, path, multi_part, number, part, leak_word, c_word) - poster_path = os.path.join(path, f"{number}{leak_word}{c_word}-poster.jpg") - thumb_path = os.path.join(path, f"{number}{leak_word}{c_word}-thumb.jpg") + paste_file_to_folder_mode2(filepath, path, multi_part, number, part, leak_word, c_word, hack_word) + poster_path = os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}-poster.jpg") + thumb_path = os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}-thumb.jpg") if conf.is_watermark(): - add_mark(poster_path, thumb_path, cn_sub, leak, uncensored) + add_mark(poster_path, thumb_path, cn_sub, leak, uncensored, hack) elif conf.main_mode() == 3: path = str(Path(file_path).parent) @@ -698,29 +715,29 @@ def core_main(file_path, number_th, oCC): # 检查小封面, 如果image cut为3，则下载小封面 if imagecut == 3: - small_cover_check(path, number, json_data.get('cover_small'), leak_word, c_word, filepath) + small_cover_check(path, number, json_data.get('cover_small'), leak_word, c_word, hack_word, filepath) # creatFolder会返回番号路径 - image_download(json_data.get('cover'), number, leak_word, c_word, path, filepath) + image_download(json_data.get('cover'), number, leak_word, c_word, hack_word, path, filepath) if not multi_part or part.lower() == '-cd1': # 下载预告片 if conf.is_trailer() and json_data.get('trailer'): - trailer_download(json_data.get('trailer'), leak_word, c_word, number, path, filepath) + trailer_download(json_data.get('trailer'), leak_word, c_word, hack_word, number, path, filepath) # 下载剧照 data, path, filepath if conf.is_extrafanart() and json_data.get('extrafanart'): extrafanart_download(json_data.get('extrafanart'), path, number, filepath) # 裁剪图 - cutImage(imagecut, path, number, leak_word, c_word) + cutImage(imagecut, path, number, leak_word, c_word, hack_word) # 添加水印 - poster_path = os.path.join(path, f"{number}{leak_word}{c_word}-poster.jpg") - thumb_path = os.path.join(path, f"{number}{leak_word}{c_word}-thumb.jpg") + poster_path = os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}-poster.jpg") + thumb_path = os.path.join(path, f"{number}{leak_word}{c_word}{hack_word}-thumb.jpg") if conf.is_watermark(): - add_mark(poster_path, thumb_path, cn_sub, leak, uncensored) + add_mark(poster_path, thumb_path, cn_sub, leak, uncensored, hack) # 最后输出.nfo元数据文件，以完成.nfo文件创建作为任务成功标志 print_files(path, leak_word, c_word, json_data.get('naming_rule'), part, cn_sub, json_data, filepath, - tag, json_data.get('actor_list'), liuchu, uncensored) + tag, json_data.get('actor_list'), liuchu, uncensored, hack_word)