From e0d2058fa0cf7cfe5828f3a1c5f6290b0c6a53e5 Mon Sep 17 00:00:00 2001
From: Yoshiko <42309414+yoshiko2@users.noreply.github.com>
Date: Fri, 14 Feb 2020 18:46:11 +0800
Subject: [PATCH] Update 2.6
---
AV_Data_Capture.py | 5 +-
core.py | 6 +-
core2.py | 678 +++++++++++++++++++++++++++++++++++++++++++++
update_check.json | 4 +-
4 files changed, 688 insertions(+), 5 deletions(-)
create mode 100644 core2.py
diff --git a/AV_Data_Capture.py b/AV_Data_Capture.py
index 9968590..dad39f2 100755
--- a/AV_Data_Capture.py
+++ b/AV_Data_Capture.py
@@ -96,7 +96,7 @@ def getNumber(filepath,absolute_path = False):
if __name__ == '__main__':
- version = '2.5'
+ version = '2.6'
config_file = 'config.ini'
config = ConfigParser()
config.read(config_file, encoding='UTF-8')
@@ -134,6 +134,9 @@ if __name__ == '__main__':
count = count + 1
percentage = str(count / int(count_all) * 100)[:4] + '%'
print('[!] - ' + percentage + ' [' + str(count) + '/' + count_all + '] -')
+ # print("[!]Making Data for [" + i + "], the number is [" + getNumber(i) + "]")
+ # core_main(i, getNumber(i))
+ # print("[*]======================================================")
try:
print("[!]Making Data for [" + i + "], the number is [" + getNumber(i) + "]")
core_main(i, getNumber(i))
diff --git a/core.py b/core.py
index e8ab087..ce11094 100755
--- a/core.py
+++ b/core.py
@@ -90,6 +90,7 @@ def getDataFromJSON(file_number, filepath, failed_folder): # 从JSON返回元
runtime = json_data['runtime']
outline = json_data['runtime']
label = json_data['label']
+ year = json_data['year']
try:
cover_small = json_data['cover_small']
except:
@@ -134,6 +135,7 @@ def getDataFromJSON(file_number, filepath, failed_folder): # 从JSON返回元
json_data['tag'] = tag
json_data['naming_rule'] = naming_rule
json_data['location_rule'] = location_rule
+ json_data['year'] = year
return json_data
@@ -548,7 +550,7 @@ def cutImage(option, imagecut, path, number, c_word):
def pasteFileToFolder(filepath, path, number, c_word): # 文件路径，番号，后缀，要移动至的位置
- houzhui = str(re.search('[.](AVI|RMVB|WMV|MOV|MP4|MKV|FLV|TS|WEBM|avi|rmvb|wmv|mov|mp4|mkv|flv|ts|webm)$', filepath).group())
+ houzhui = str(re.search('[.](AVI|RMVB|WMV|MOV|MP4|MKV|FLV|TS|avi|rmvb|wmv|mov|mp4|mkv|flv|ts)$', filepath).group())
try:
if config['common']['soft_link'] == '1': # 如果soft_link=1 使用软链接
os.symlink(filepath, path + '/' + number + c_word + houzhui)
@@ -575,7 +577,7 @@ def pasteFileToFolder(filepath, path, number, c_word): # 文件路径，番号
def pasteFileToFolder_mode2(filepath, path, multi_part, number, part, c_word): # 文件路径，番号，后缀，要移动至的位置
if multi_part == 1:
number += part # 这时number会被附加上CD1后缀
- houzhui = str(re.search('[.](AVI|RMVB|WMV|MOV|MP4|MKV|FLV|TS|WEBM|avi|rmvb|wmv|mov|mp4|mkv|flv|ts|webm)$', filepath).group())
+ houzhui = str(re.search('[.](AVI|RMVB|WMV|MOV|MP4|MKV|FLV|TS|avi|rmvb|wmv|mov|mp4|mkv|flv|ts)$', filepath).group())
try:
if config['common']['soft_link'] == '1':
os.symlink(filepath, path + '/' + number + part + c_word + houzhui)
diff --git a/core2.py b/core2.py
new file mode 100644
index 0000000..cdfd637
--- /dev/null
+++ b/core2.py
@@ -0,0 +1,678 @@
+# -*- coding: utf-8 -*-
+
+import re
+import os
+import os.path
+import shutil
+from PIL import Image
+import time
+import json
+from ADC_function import *
+from configparser import ConfigParser
+import argparse
+# =========website========
+import fc2fans_club
+import siro
+import avsox
+import javbus
+import javdb
+import fanza
+import requests
+
+
+# =====================本地文件处理===========================
+
+def escapePath(path, Config): # Remove escape literals
+ escapeLiterals = Config['escape']['literals']
+ backslash = '\\'
+ for literal in escapeLiterals:
+ path = path.replace(backslash + literal, '')
+ return path
+
+
+def moveFailedFolder(filepath, failed_folder):
+ print('[-]Move to Failed output folder')
+ shutil.move(filepath, str(os.getcwd()) + '/' + failed_folder + '/')
+ os._exit(0)
+
+
+def CreatFailedFolder(failed_folder):
+ if not os.path.exists(failed_folder + '/'): # 新建failed文件夹
+ try:
+ os.makedirs(failed_folder + '/')
+ except:
+ print("[-]failed!can not be make Failed output folder\n[-](Please run as Administrator)")
+ os._exit(0)
+
+
+def getDataFromJSON(file_number, filepath, failed_folder): # 从JSON返回元数据
+ # ================================================网站规则添加开始================================================
+
+ if re.match('^\d{5,}', file_number):
+ json_data = json.loads(avsox.main(file_number))
+ if getDataState(json_data) == 0: # 如果元数据获取失败，请求番号至其他网站抓取
+ json_data = json.loads(javdb.main(file_number))
+ # ==
+ elif re.match('\d+\D+', file_number):
+ json_data = json.loads(siro.main(file_number))
+ if getDataState(json_data) == 0: # 如果元数据获取失败，请求番号至其他网站抓取
+ json_data = json.loads(javbus.main(file_number))
+ if getDataState(json_data) == 0: # 如果元数据获取失败，请求番号至其他网站抓取
+ json_data = json.loads(javdb.main(file_number))
+ # ==
+ elif 'fc2' in file_number or 'FC2' in file_number:
+ json_data = json.loads(fc2fans_club.main(
+ file_number.replace('fc2-', '').replace('fc2_', '').replace('FC2-', '').replace('fc2_', '')))
+ # ==
+ elif 'HEYZO' in file_number or 'heyzo' in file_number or 'Heyzo' in file_number:
+ json_data = json.loads(avsox.main(file_number))
+ # ==
+ elif 'siro' in file_number or 'SIRO' in file_number or 'Siro' in file_number:
+ json_data = json.loads(siro.main(file_number))
+ elif not '-' in file_number or '_' in file_number:
+ json_data = json.loads(fanza.main(file_number))
+ if getDataState(json_data) == 0: # 如果元数据获取失败，请求番号至其他网站抓取
+ json_data = json.loads(javbus.main(file_number))
+ if getDataState(json_data) == 0: # 如果元数据获取失败，请求番号至其他网站抓取
+ json_data = json.loads(avsox.main(file_number))
+ if getDataState(json_data) == 0: # 如果元数据获取失败，请求番号至其他网站抓取
+ json_data = json.loads(javdb.main(file_number))
+ # ==
+ else:
+ json_data = json.loads(javbus.main(file_number))
+ if getDataState(json_data) == 0: # 如果元数据获取失败，请求番号至其他网站抓取
+ json_data = json.loads(avsox.main(file_number))
+ if getDataState(json_data) == 0: # 如果元数据获取失败，请求番号至其他网站抓取
+ json_data = json.loads(javdb.main(file_number))
+
+ # ================================================网站规则添加结束================================================
+
+ title = json_data['title']
+ actor_list = str(json_data['actor']).strip("[ ]").replace("'", '').split(',') # 字符串转列表
+ release = json_data['release']
+ number = json_data['number']
+ try:
+ cover_small = json_data['cover_small']
+ except:
+ cover_small = ''
+ imagecut = json_data['imagecut']
+ tag = str(json_data['tag']).strip("[ ]").replace("'", '').replace(" ", '').split(',') # 字符串转列表 @
+ actor = str(actor_list).strip("[ ]").replace("'", '').replace(" ", '')
+
+ if title == '' or number == '':
+ print('[-]Movie Data not found!')
+ moveFailedFolder(filepath, failed_folder)
+
+ # if imagecut == '3':
+ # DownloadFileWithFilename()
+
+ # ====================处理异常字符====================== #\/:*?"<>|
+ title = title.replace('\\', '')
+ title = title.replace('/', '')
+ title = title.replace(':', '')
+ title = title.replace('*', '')
+ title = title.replace('?', '')
+ title = title.replace('"', '')
+ title = title.replace('<', '')
+ title = title.replace('>', '')
+ title = title.replace('|', '')
+ release = release.replace('/', '-')
+ tmpArr = cover_small.split(',')
+ if len(tmpArr) > 0:
+ cover_small = tmpArr[0].strip('\"').strip('\'')
+ # ====================处理异常字符 END================== #\/:*?"<>|
+
+ naming_rule = eval(config['Name_Rule']['naming_rule'])
+ location_rule = eval(config['Name_Rule']['location_rule'])
+
+ # 返回处理后的json_data
+ json_data['title'] = title
+ json_data['actor'] = actor
+ json_data['release'] = release
+ json_data['cover_small'] = cover_small
+ json_data['tag'] = tag
+ json_data['naming_rule'] = naming_rule
+ json_data['location_rule'] = location_rule
+ return json_data
+
+
+def get_info(json_data): # 返回json里的数据
+ title = json_data['title']
+ studio = json_data['studio']
+ year = json_data['year']
+ outline = json_data['outline']
+ runtime = json_data['runtime']
+ director = json_data['director']
+ actor_photo = json_data['actor_photo']
+ release = json_data['release']
+ number = json_data['number']
+ cover = json_data['cover']
+ website = json_data['website']
+ return title, studio, year, outline, runtime, director, actor_photo, release, number, cover, website
+
+
+def smallCoverCheck(path, number, imagecut, cover_small, c_word, option, Config, filepath, failed_folder):
+ if imagecut == 3:
+ if option == 'emby':
+ DownloadFileWithFilename(cover_small, '1.jpg', path, Config, filepath, failed_folder)
+ try:
+ img = Image.open(path + '/1.jpg')
+ except Exception:
+ img = Image.open('1.jpg')
+ w = img.width
+ h = img.height
+ img.save(path + '/' + number + c_word + '.png')
+ time.sleep(1)
+ os.remove(path + '/1.jpg')
+ if option == 'kodi':
+ DownloadFileWithFilename(cover_small, '1.jpg', path, Config, filepath, failed_folder)
+ try:
+ img = Image.open(path + '/1.jpg')
+ except Exception:
+ img = Image.open('1.jpg')
+ w = img.width
+ h = img.height
+ img.save(path + '/' + number + c_word + '-poster.jpg')
+ time.sleep(1)
+ os.remove(path + '/1.jpg')
+ if option == 'plex':
+ DownloadFileWithFilename(cover_small, '1.jpg', path, Config, filepath, failed_folder)
+ try:
+ img = Image.open(path + '/1.jpg')
+ except Exception:
+ img = Image.open('1.jpg')
+ w = img.width
+ h = img.height
+ img.save(path + '/poster.png')
+ os.remove(path + '/1.jpg')
+
+
+def creatFolder(success_folder, location_rule, json_data, Config): # 创建文件夹
+ title, studio, year, outline, runtime, director, actor_photo, release, number, cover, website = get_info(json_data)
+ if len(location_rule) > 240: # 新建成功输出文件夹
+ path = success_folder + '/' + location_rule.replace("'actor'", "'manypeople'", 3).replace("actor","'manypeople'",3) # path为影片+元数据所在目录
+ else:
+ path = success_folder + '/' + location_rule
+ # print(path)
+ if not os.path.exists(path):
+ path = escapePath(path, Config)
+ try:
+ os.makedirs(path)
+ except:
+ path = success_folder + '/' + location_rule.replace('/[' + number + ']-' + title, "/number")
+ path = escapePath(path, Config)
+ os.makedirs(path)
+ return path
+
+
+# =====================资源下载部分===========================
+def DownloadFileWithFilename(url, filename, path, Config, filepath, failed_folder): # path = examle:photo , video.in the Project Folder!
+ retry_count = 0
+ proxy = ''
+ timeout = 0
+ try:
+ proxy = Config['proxy']['proxy']
+ timeout = int(Config['proxy']['timeout'])
+ retry_count = int(Config['proxy']['retry'])
+ except:
+ print('[-]Proxy config error! Please check the config.')
+ i = 0
+
+ while i < retry_count:
+ try:
+ if not proxy == '':
+ if not os.path.exists(path):
+ os.makedirs(path)
+ headers = {
+ 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/68.0.3440.106 Safari/537.36'}
+ r = requests.get(url, headers=headers, timeout=timeout,
+ proxies={"http": "http://" + str(proxy), "https": "https://" + str(proxy)})
+ if r == '':
+ print('[-]Movie Data not found!')
+ os._exit(0)
+ with open(str(path) + "/" + filename, "wb") as code:
+ code.write(r.content)
+ return
+ else:
+ if not os.path.exists(path):
+ os.makedirs(path)
+ headers = {
+ 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/68.0.3440.106 Safari/537.36'}
+ r = requests.get(url, timeout=timeout, headers=headers)
+ if r == '':
+ print('[-]Movie Data not found!')
+ os._exit(0)
+ with open(str(path) + "/" + filename, "wb") as code:
+ code.write(r.content)
+ return
+ except requests.exceptions.RequestException:
+ i += 1
+ print('[-]Image Download : Connect retry ' + str(i) + '/' + str(retry_count))
+ except requests.exceptions.ConnectionError:
+ i += 1
+ print('[-]Image Download : Connect retry ' + str(i) + '/' + str(retry_count))
+ except requests.exceptions.ProxyError:
+ i += 1
+ print('[-]Image Download : Connect retry ' + str(i) + '/' + str(retry_count))
+ except requests.exceptions.ConnectTimeout:
+ i += 1
+ print('[-]Image Download : Connect retry ' + str(i) + '/' + str(retry_count))
+ print('[-]Connect Failed! Please check your Proxy or Network!')
+ moveFailedFolder(filepath, failed_folder)
+
+
+def imageDownload(option, cover, number, c_word, path, multi_part, Config, filepath, failed_folder): # 封面是否下载成功，否则移动到failed
+ if option == 'emby':
+ if DownloadFileWithFilename(cover, number + c_word + '.jpg', path, Config, filepath, failed_folder) == 'failed':
+ moveFailedFolder(filepath, failed_folder)
+ DownloadFileWithFilename(cover, number + c_word + '.jpg', path, Config, filepath, failed_folder)
+ if not os.path.getsize(path + '/' + number + c_word + '.jpg') == 0:
+ print('[+]Image Downloaded!', path + '/' + number + c_word + '.jpg')
+ return
+ i = 1
+ while i <= int(config['proxy']['retry']):
+ if os.path.getsize(path + '/' + number + c_word + '.jpg') == 0:
+ print('[!]Image Download Failed! Trying again. [' + config['proxy']['retry'] + '/3]')
+ DownloadFileWithFilename(cover, number + c_word + '.jpg', path, Config, filepath, failed_folder)
+ i = i + 1
+ continue
+ else:
+ break
+ if multi_part == 1:
+ old_name = os.path.join(path, number + c_word + '.jpg')
+ new_name = os.path.join(path, number + c_word + '.jpg')
+ os.rename(old_name, new_name)
+ print('[+]Image Downloaded!', path + '/' + number + c_word + '.jpg')
+ else:
+ print('[+]Image Downloaded!', path + '/' + number + c_word + '.jpg')
+ elif option == 'plex':
+ if DownloadFileWithFilename(cover, 'fanart.jpg', path, Config, filepath, failed_folder) == 'failed':
+ moveFailedFolder(filepath, failed_folder)
+ DownloadFileWithFilename(cover, 'fanart.jpg', path, Config, filepath, failed_folder)
+ if not os.path.getsize(path + '/fanart.jpg') == 0:
+ print('[+]Image Downloaded!', path + '/fanart.jpg')
+ return
+ i = 1
+ while i <= int(config['proxy']['retry']):
+ if os.path.getsize(path + '/fanart.jpg') == 0:
+ print('[!]Image Download Failed! Trying again. [' + config['proxy']['retry'] + '/3]')
+ DownloadFileWithFilename(cover, 'fanart.jpg', path, Config, filepath, failed_folder)
+ i = i + 1
+ continue
+ else:
+ break
+ if not os.path.getsize(path + '/' + number + c_word + '.jpg') == 0:
+ print('[!]Image Download Failed! Trying again.')
+ DownloadFileWithFilename(cover, number + c_word + '.jpg', path, Config, filepath, failed_folder)
+ print('[+]Image Downloaded!', path + '/fanart.jpg')
+ elif option == 'kodi':
+ if DownloadFileWithFilename(cover, number + c_word + '-fanart.jpg', path, Config, filepath, failed_folder) == 'failed':
+ moveFailedFolder(filepath, failed_folder)
+ DownloadFileWithFilename(cover, number + c_word + '-fanart.jpg', path, Config, filepath, failed_folder)
+ if not os.path.getsize(path + '/' + number + c_word + '-fanart.jpg') == 0:
+ print('[+]Image Downloaded!', path + '/' + number + c_word + '-fanart.jpg')
+ return
+ i = 1
+ while i <= int(config['proxy']['retry']):
+ if os.path.getsize(path + '/' + number + c_word + '-fanart.jpg') == 0:
+ print('[!]Image Download Failed! Trying again. [' + config['proxy']['retry'] + '/3]')
+ DownloadFileWithFilename(cover, number + c_word + '-fanart.jpg', path, Config, filepath, failed_folder)
+ i = i + 1
+ continue
+ else:
+ break
+ print('[+]Image Downloaded!', path + '/' + number + c_word + '-fanart.jpg')
+
+
+def PrintFiles(option, path, c_word, naming_rule, part, cn_sub, json_data, filepath, failed_folder, tag):
+ title, studio, year, outline, runtime, director, actor_photo, release, number, cover, website = get_info(json_data)
+ try:
+ if not os.path.exists(path):
+ os.makedirs(path)
+ if option == 'plex':
+ with open(path + "/" + number + c_word + ".nfo", "wt", encoding='UTF-8') as code:
+ print('', file=code)
+ print("", file=code)
+ print(" " + naming_rule + part + "", file=code)
+ print(" ", file=code)
+ print(" ", file=code)
+ print(" " + studio + "+", file=code)
+ print(" " + year + "", file=code)
+ print(" " + outline + "", file=code)
+ print(" " + outline + "", file=code)
+ print(" " + str(runtime).replace(" ", "") + "", file=code)
+ print(" " + director + "", file=code)
+ print(" poster.png", file=code)
+ print(" thumb.png", file=code)
+ print(" fanart.jpg", file=code)
+ try:
+ for key, value in actor_photo.items():
+ print(" ", file=code)
+ print(" " + key + "", file=code)
+ if not value == '': # or actor_photo == []:
+ print(" " + value + "", file=code)
+ print(" ", file=code)
+ except:
+ aaaa = ''
+ print(" " + studio + "", file=code)
+ print(" ", file=code)
+ if cn_sub == '1':
+ print(" 中文字幕", file=code)
+ try:
+ for i in str(json_data['tag']).strip("[ ]").replace("'", '').replace(" ", '').split(','):
+ print(" " + i + "", file=code)
+ except:
+ aaaaa = ''
+ try:
+ for i in str(json_data['tag']).strip("[ ]").replace("'", '').replace(" ", '').split(','):
+ print(" " + i + "", file=code)
+ except:
+ aaaaaaaa = ''
+ if cn_sub == '1':
+ print(" 中文字幕", file=code)
+ print(" " + number + "", file=code)
+ print(" " + release + "", file=code)
+ print(" " + cover + "", file=code)
+ print(" " + website + "", file=code)
+ print("", file=code)
+ print("[+]Writeed! " + path + "/" + number + ".nfo")
+ elif option == 'emby':
+ with open(path + "/" + number + c_word + ".nfo", "wt", encoding='UTF-8') as code:
+ print('', file=code)
+ print("", file=code)
+ print(" " + naming_rule + part + "", file=code)
+ print(" ", file=code)
+ print(" ", file=code)
+ print(" " + studio + "+", file=code)
+ print(" " + year + "", file=code)
+ print(" " + outline + "", file=code)
+ print(" " + outline + "", file=code)
+ print(" " + str(runtime).replace(" ", "") + "", file=code)
+ print(" " + director + "", file=code)
+ print(" " + number + c_word + ".png", file=code)
+ print(" " + number + c_word + ".png", file=code)
+ print(" " + number + c_word + '.jpg' + "", file=code)
+ try:
+ for key, value in actor_photo.items():
+ print(" ", file=code)
+ print(" " + key + "", file=code)
+ if not value == '': # or actor_photo == []:
+ print(" " + value + "", file=code)
+ print(" ", file=code)
+ except:
+ aaaa = ''
+ print(" " + studio + "", file=code)
+ print(" ", file=code)
+ if cn_sub == '1':
+ print(" 中文字幕", file=code)
+ try:
+ for i in tag:
+ print(" " + i + "", file=code)
+ except:
+ aaaaa = ''
+ try:
+ for i in tag:
+ print(" " + i + "", file=code)
+ except:
+ aaaaaaaa = ''
+ if cn_sub == '1':
+ print(" 中文字幕", file=code)
+ print(" " + number + "", file=code)
+ print(" " + release + "", file=code)
+ print(" " + cover + "", file=code)
+ print(" " + "https://www.javbus.com/" + number + "", file=code)
+ print("", file=code)
+ print("[+]Writeed! " + path + "/" + number + c_word + ".nfo")
+ elif option == 'kodi':
+ with open(path + "/" + number + c_word + ".nfo", "wt", encoding='UTF-8') as code:
+ print('', file=code)
+ print("", file=code)
+ print(" " + naming_rule + part + "", file=code)
+ print(" ", file=code)
+ print(" ", file=code)
+ print(" " + studio + "+", file=code)
+ print(" " + year + "", file=code)
+ print(" " + outline + "", file=code)
+ print(" " + outline + "", file=code)
+ print(" " + str(runtime).replace(" ", "") + "", file=code)
+ print(" " + director + "", file=code)
+ print(" " + number + c_word + "-poster.jpg", file=code)
+ print(" " + number + c_word + '-fanart.jpg' + "", file=code)
+ try:
+ for key, value in actor_photo.items():
+ print(" ", file=code)
+ print(" " + key + "", file=code)
+ if not value == '': # or actor_photo == []:
+ print(" " + value + "", file=code)
+ print(" ", file=code)
+ except:
+ aaaa = ''
+ print(" " + studio + "", file=code)
+ print(" ", file=code)
+ if cn_sub == '1':
+ print(" 中文字幕", file=code)
+ try:
+ for i in tag:
+ print(" " + i + "", file=code)
+ except:
+ aaaaa = ''
+ try:
+ for i in tag:
+ print(" " + i + "", file=code)
+ except:
+ aaaaaaaa = ''
+ if cn_sub == '1':
+ print(" 中文字幕", file=code)
+ print(" " + number + "", file=code)
+ print(" " + release + "", file=code)
+ print(" " + cover + "", file=code)
+ print(" " + "https://www.javbus.com/" + number + "", file=code)
+ print("", file=code)
+ print("[+]Writeed! " + path + "/" + number + c_word + ".nfo")
+ except IOError as e:
+ print("[-]Write Failed!")
+ print(e)
+ moveFailedFolder(filepath, failed_folder)
+ except Exception as e1:
+ print(e1)
+ print("[-]Write Failed!")
+ moveFailedFolder(filepath, failed_folder)
+
+
+def cutImage(option, imagecut, path, number, c_word):
+ if option == 'plex':
+ if imagecut == 1:
+ try:
+ img = Image.open(path + '/fanart.jpg')
+ imgSize = img.size
+ w = img.width
+ h = img.height
+ img2 = img.crop((w / 1.9, 0, w, h))
+ img2.save(path + '/poster.png')
+ except:
+ print('[-]Cover cut failed!')
+ elif imagecut == 0:
+ img = Image.open(path + '/fanart.jpg')
+ w = img.width
+ h = img.height
+ img.save(path + '/poster.png')
+ elif option == 'emby':
+ if imagecut == 1:
+ try:
+ img = Image.open(path + '/' + number + c_word + '.jpg')
+ imgSize = img.size
+ w = img.width
+ h = img.height
+ img2 = img.crop((w / 1.9, 0, w, h))
+ img2.save(path + '/' + number + c_word + '.png')
+ except:
+ print('[-]Cover cut failed!')
+ elif imagecut == 0:
+ img = Image.open(path + '/' + number + c_word + '.jpg')
+ w = img.width
+ h = img.height
+ img.save(path + '/' + number + c_word + '.png')
+ elif option == 'kodi':
+ if imagecut == 1:
+ try:
+ img = Image.open(path + '/' + number + c_word + '-fanart.jpg')
+ imgSize = img.size
+ w = img.width
+ h = img.height
+ img2 = img.crop((w / 1.9, 0, w, h))
+ img2.save(path + '/' + number + c_word + '-poster.jpg')
+ except:
+ print('[-]Cover cut failed!')
+ elif imagecut == 0:
+ img = Image.open(path + '/' + number + c_word + '-fanart.jpg')
+ w = img.width
+ h = img.height
+ try:
+ img = img.convert('RGB')
+ img.save(path + '/' + number + c_word + '-poster.jpg')
+ except:
+ img = img.convert('RGB')
+ img.save(path + '/' + number + c_word + '-poster.jpg')
+
+
+def pasteFileToFolder(filepath, path, number, c_word): # 文件路径，番号，后缀，要移动至的位置
+ houzhui = str(re.search('[.](AVI|RMVB|WMV|MOV|MP4|MKV|FLV|TS|avi|rmvb|wmv|mov|mp4|mkv|flv|ts)$', filepath).group())
+ try:
+ if config['common']['soft_link'] == '1': # 如果soft_link=1 使用软链接
+ os.symlink(filepath, path + '/' + number + c_word + houzhui)
+ else:
+ os.rename(filepath, path + '/' + number + c_word + houzhui)
+ if os.path.exists(os.getcwd() + '/' + number + c_word + '.srt'): # 字幕移动
+ os.rename(os.getcwd() + '/' + number + c_word + '.srt', path + '/' + number + c_word + '.srt')
+ print('[+]Sub moved!')
+ elif os.path.exists(os.getcwd() + '/' + number + c_word + '.ssa'):
+ os.rename(os.getcwd() + '/' + number + c_word + '.ssa', path + '/' + number + c_word + '.ssa')
+ print('[+]Sub moved!')
+ elif os.path.exists(os.getcwd() + '/' + number + c_word + '.sub'):
+ os.rename(os.getcwd() + '/' + number + c_word + '.sub', path + '/' + number + c_word + '.sub')
+ print('[+]Sub moved!')
+ except FileExistsError:
+ print('[-]File Exists! Please check your movie!')
+ print('[-]move to the root folder of the program.')
+ os._exit(0)
+ except PermissionError:
+ print('[-]Error! Please run as administrator!')
+ os._exit(0)
+
+
+def pasteFileToFolder_mode2(filepath, path, multi_part, number, part, c_word): # 文件路径，番号，后缀，要移动至的位置
+ if multi_part == 1:
+ number += part # 这时number会被附加上CD1后缀
+ houzhui = str(re.search('[.](AVI|RMVB|WMV|MOV|MP4|MKV|FLV|TS|avi|rmvb|wmv|mov|mp4|mkv|flv|ts)$', filepath).group())
+ try:
+ if config['common']['soft_link'] == '1':
+ os.symlink(filepath, path + '/' + number + part + c_word + houzhui)
+ else:
+ os.rename(filepath, path + '/' + number + part + c_word + houzhui)
+ if os.path.exists(number + '.srt'): # 字幕移动
+ os.rename(number + part + c_word + '.srt', path + '/' + number + part + c_word + '.srt')
+ print('[+]Sub moved!')
+ elif os.path.exists(number + part + c_word + '.ass'):
+ os.rename(number + part + c_word + '.ass', path + '/' + number + part + c_word + '.ass')
+ print('[+]Sub moved!')
+ elif os.path.exists(number + part + c_word + '.sub'):
+ os.rename(number + part + c_word + '.sub', path + '/' + number + part + c_word + '.sub')
+ print('[+]Sub moved!')
+ print('[!]Success')
+ except FileExistsError:
+ print('[-]File Exists! Please check your movie!')
+ print('[-]move to the root folder of the program.')
+ os._exit(0)
+ except PermissionError:
+ print('[-]Error! Please run as administrator!')
+ os._exit(0)
+
+
+def copyRenameJpgToBackdrop(option, path, number, c_word):
+ if option == 'plex':
+ shutil.copy(path + '/fanart.jpg', path + '/Backdrop.jpg')
+ shutil.copy(path + '/poster.png', path + '/thumb.png')
+ if option == 'emby':
+ shutil.copy(path + '/' + number + c_word + '.jpg', path + '/Backdrop.jpg')
+ if option == 'kodi':
+ shutil.copy(path + '/' + number + c_word + '-fanart.jpg', path + '/Backdrop.jpg')
+
+
+def get_part(filepath, failed_folder):
+ try:
+ if re.search('-CD\d+', filepath):
+ return re.findall('-CD\d+', filepath)[0]
+ if re.search('-cd\d+', filepath):
+ return re.findall('-cd\d+', filepath)[0]
+ except:
+ print("[-]failed!Please rename the filename again!")
+ moveFailedFolder(filepath, failed_folder)
+
+
+def debug_mode(json_data):
+ try:
+ if config['debug_mode']['switch'] == '1':
+ print('[+] ---Debug info---')
+ for i, v in json_data.items():
+ if i == 'outline':
+ print('[+] -', i, ' :', len(v), 'characters')
+ continue
+ if i == 'actor_photo' or i == 'year':
+ continue
+ print('[+] -', "%-11s" % i, ':', v)
+ print('[+] ---Debug info---')
+ except:
+ aaa = ''
+
+
+def core_main(file_path, number_th):
+ # =======================================================================初始化所需变量
+ multi_part = 0
+ part = ''
+ c_word = ''
+ option = ''
+ cn_sub = ''
+ path = ''
+ config_file = 'config.ini'
+ Config = ConfigParser()
+ Config.read(config_file, encoding='UTF-8')
+ try:
+ option = ReadMediaWarehouse()
+ except:
+ print('[-]Config media_warehouse read failed!')
+ program_mode = Config['common']['main_mode'] # 运行模式
+ failed_folder = Config['common']['failed_output_folder'] # 失败输出目录
+ success_folder = Config['common']['success_output_folder'] # 成功输出目录
+ filepath = file_path # 影片的路径
+ number = number_th
+ json_data = getDataFromJSON(number, filepath, failed_folder) # 定义番号
+ imagecut = json_data['imagecut']
+ tag = json_data['tag']
+ # =======================================================================判断-C,-CD后缀
+ if '-CD' in filepath or '-cd' in filepath:
+ multi_part = 1
+ part = get_part(filepath, failed_folder)
+ if '-c.' in filepath or '-C.' in filepath or '中文' in filepath or '字幕' in filepath:
+ cn_sub = '1'
+ c_word = '-C' # 中文字幕影片后缀
+
+ CreatFailedFolder(failed_folder) # 创建输出失败目录
+ debug_mode(json_data) # 调试模式检测
+ path = creatFolder(success_folder, json_data['location_rule'], json_data, Config) # 创建文件夹
+ print(path)
+ # =======================================================================刮削模式
+ if program_mode == '1':
+ if multi_part == 1:
+ number += part # 这时number会被附加上CD1后缀
+ smallCoverCheck(path, number, imagecut, json_data['cover_small'], c_word, option, Config, filepath, failed_folder) # 检查小封面
+ imageDownload(option, json_data['cover'], number, c_word, path, multi_part, Config, filepath, failed_folder) # creatFoder会返回番号路径
+ cutImage(option, imagecut, path, number, c_word) # 裁剪图
+ copyRenameJpgToBackdrop(option, path, number, c_word)
+ PrintFiles(option, path, c_word, json_data['naming_rule'], part, cn_sub, json_data, filepath, failed_folder, tag) # 打印文件
+ pasteFileToFolder(filepath, path, number, c_word) # 移动文件
+ # =======================================================================整理模式
+ elif program_mode == '2':
+ pasteFileToFolder_mode2(filepath, path, multi_part, number, part, c_word) # 移动文件
diff --git a/update_check.json b/update_check.json
index c671921..b44830a 100644
--- a/update_check.json
+++ b/update_check.json
@@ -1,5 +1,5 @@
{
- "version": "2.5",
- "version_show":"2.5",
+ "version": "2.6",
+ "version_show":"2.6",
"download": "https://github.com/yoshiko2/AV_Data_Capture/releases"
}