diff --git a/core.py b/core.py
index e1b48a7..3ad271b 100755
--- a/core.py
+++ b/core.py
@@ -1,696 +1,691 @@
-# -*- 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 mgstage
-import avsox
-import javbus
-import javdb
-import fanza
-import requests
-import random
-
-
-# =====================本地文件处理===========================
-
-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 + '/')
- return
-
-
-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)")
- return
-
-
-def getDataFromJSON(file_number, filepath, failed_folder): # 从JSON返回元数据
- """
- iterate through all services and fetch the data
- """
-
- func_mapping = {
- "avsox": avsox.main,
- "fc2": fc2fans_club.main,
- "fanza": fanza.main,
- "javdb": javdb.main,
- "javbus": javbus.main,
- "mgstage": mgstage.main,
- }
-
- # default fetch order list, from the begining to the end
- sources = ["javbus", "javdb", "fanza", "mgstage", "fc2", "avsox"]
-
- # if the input file name matches centain rules,
- # move some web service to the begining of the list
- if re.match(r"^\d{5,}", file_number) or (
- "HEYZO" in file_number or "heyzo" in file_number or "Heyzo" in file_number
- ):
- sources.insert(0, sources.pop(sources.index("avsox")))
- elif re.match(r"\d+\D+", file_number) or (
- "siro" in file_number or "SIRO" in file_number or "Siro" in file_number
- ):
- sources.insert(0, sources.pop(sources.index("mgstage")))
- elif "fc2" in file_number or "FC2" in file_number:
- sources.insert(0, sources.pop(sources.index("fc2")))
-
- for source in sources:
- json_data = json.loads(func_mapping[source](file_number))
- # if any service return a valid return, break
- if getDataState(json_data) != 0:
- break
-
- # ================================================网站规则添加结束================================================
-
- title = json_data['title']
- actor_list = str(json_data['actor']).strip("[ ]").replace("'", '').split(',') # 字符串转列表
- release = json_data['release']
- number = json_data['number']
- studio = json_data['studio']
- source = json_data['source']
- runtime = json_data['runtime']
- outline = json_data['runtime']
- label = json_data['label']
- year = json_data['year']
- 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)
- return
-
- # 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'])
- if 'actor' in config['Name_Rule']['location_rule'] and len(actor) > 100:
- print(config['Name_Rule']['location_rule'])
- location_rule = eval(config['Name_Rule']['location_rule'].replace("actor","'多人作品'"))
- if 'title' in config['Name_Rule']['location_rule'] and len(title) > 100:
- location_rule = eval(config['Name_Rule']['location_rule'].replace("title",'number'))
-
- # 返回处理后的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
- json_data['year'] = year
- 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.jpg')
- 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!
- proxy, timeout, retry_count = get_network_settings()
- 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!')
- return
- 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!')
- return
- 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)
- return
-
-
-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)
- return
- 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)
- return
- 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)
- return
- 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.jpg", 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(" " + website + "", 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(" " + website + "", 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)
- return
- except Exception as e1:
- print(e1)
- print("[-]Write Failed!")
- moveFailedFolder(filepath, failed_folder)
- return
-
-
-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.jpg')
- except:
- print('[-]Cover cut failed!')
- elif imagecut == 0:
- img = Image.open(path + '/fanart.jpg')
- w = img.width
- h = img.height
- img.save(path + '/poster.jpg')
- 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|WEBM|avi|rmvb|wmv|mov|mp4|mkv|flv|ts|webm)$', 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.')
- return
- except PermissionError:
- print('[-]Error! Please run as administrator!')
- return
-
-
-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())
- 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.')
- return
- except PermissionError:
- print('[-]Error! Please run as administrator!')
- return
-
-
-def copyRenameJpgToBackdrop(option, path, number, c_word):
- if option == 'plex':
- shutil.copy(path + '/fanart.jpg', path + '/Backdrop.jpg')
- shutil.copy(path + '/poster.jpg', 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)
- return
-
-
-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 = ''
- 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) # 定义番号
- if json_data["number"] != number:
- # fix issue #119
- # the root cause is we normalize the search id
- # PrintFiles() will use the normalized id from website,
- # but pasteFileToFolder() still use the input raw search id
- # so the solution is: use the normalized search id
- number = json_data["number"]
- 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) # 创建文件夹
- # =======================================================================刮削模式
- 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) # 移动文件
+# -*- 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 mgstage
+import avsox
+import javbus
+import javdb
+import fanza
+import requests
+import random
+
+
+# =====================本地文件处理===========================
+
+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 + '/')
+ return
+
+
+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)")
+ return
+
+
+def getDataFromJSON(file_number, filepath, failed_folder): # 从JSON返回元数据
+ """
+ iterate through all services and fetch the data
+ """
+
+ func_mapping = {
+ "avsox": avsox.main,
+ "fc2": fc2fans_club.main,
+ "fanza": fanza.main,
+ "javdb": javdb.main,
+ "javbus": javbus.main,
+ "mgstage": mgstage.main,
+ }
+
+ # default fetch order list, from the begining to the end
+ sources = ["javbus", "javdb", "fanza", "mgstage", "fc2", "avsox"]
+
+ # if the input file name matches centain rules,
+ # move some web service to the begining of the list
+ if re.match(r"^\d{5,}", file_number) or (
+ "HEYZO" in file_number or "heyzo" in file_number or "Heyzo" in file_number
+ ):
+ sources.insert(0, sources.pop(sources.index("avsox")))
+ elif re.match(r"\d+\D+", file_number) or (
+ "siro" in file_number or "SIRO" in file_number or "Siro" in file_number
+ ):
+ sources.insert(0, sources.pop(sources.index("fanza")))
+ elif "fc2" in file_number or "FC2" in file_number:
+ sources.insert(0, sources.pop(sources.index("fc2")))
+
+ for source in sources:
+ json_data = json.loads(func_mapping[source](file_number))
+ # if any service return a valid return, break
+ if getDataState(json_data) != 0:
+ break
+
+ # ================================================网站规则添加结束================================================
+
+ title = json_data['title']
+ actor_list = str(json_data['actor']).strip("[ ]").replace("'", '').split(',') # 字符串转列表
+ release = json_data['release']
+ number = json_data['number']
+ studio = json_data['studio']
+ source = json_data['source']
+ runtime = json_data['runtime']
+ outline = json_data['runtime']
+ label = json_data['label']
+ year = json_data['year']
+ 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)
+ return
+
+ # 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'])
+ if 'actor' in config['Name_Rule']['location_rule'] and len(actor) > 100:
+ print(config['Name_Rule']['location_rule'])
+ location_rule = eval(config['Name_Rule']['location_rule'].replace("actor","'多人作品'"))
+ if 'title' in config['Name_Rule']['location_rule'] and len(title) > 100:
+ location_rule = eval(config['Name_Rule']['location_rule'].replace("title",'number'))
+
+ # 返回处理后的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
+ json_data['year'] = year
+ json_data['actor_list'] = actor_list
+ 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.jpg')
+ 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!
+ proxy, timeout, retry_count = get_network_settings()
+ 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!')
+ return
+ 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!')
+ return
+ 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)
+ return
+
+
+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)
+ return
+ 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)
+ return
+ 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)
+ return
+ 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, actor_list):
+ 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.jpg", file=code)
+ print(" thumb.png", file=code)
+ print(" fanart.jpg", file=code)
+ try:
+ for key in actor_list:
+ print(" ", file=code)
+ print(" " + key + "", 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 in actor_list:
+ print(" ", file=code)
+ print(" " + key + "", 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(" " + website + "", 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 in actor_list:
+ print(" ", file=code)
+ print(" " + key + "", 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(" " + website + "", 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)
+ return
+ except Exception as e1:
+ print(e1)
+ print("[-]Write Failed!")
+ moveFailedFolder(filepath, failed_folder)
+ return
+
+
+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.jpg')
+ except:
+ print('[-]Cover cut failed!')
+ elif imagecut == 0:
+ img = Image.open(path + '/fanart.jpg')
+ w = img.width
+ h = img.height
+ img.save(path + '/poster.jpg')
+ 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|WEBM|avi|rmvb|wmv|mov|mp4|mkv|flv|ts|webm)$', 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.')
+ return
+ except PermissionError:
+ print('[-]Error! Please run as administrator!')
+ return
+
+
+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())
+ 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.')
+ return
+ except PermissionError:
+ print('[-]Error! Please run as administrator!')
+ return
+
+
+def copyRenameJpgToBackdrop(option, path, number, c_word):
+ if option == 'plex':
+ shutil.copy(path + '/fanart.jpg', path + '/Backdrop.jpg')
+ shutil.copy(path + '/poster.jpg', 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)
+ return
+
+
+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 = ''
+ 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) # 定义番号
+ if json_data["number"] != number:
+ # fix issue #119
+ # the root cause is we normalize the search id
+ # PrintFiles() will use the normalized id from website,
+ # but pasteFileToFolder() still use the input raw search id
+ # so the solution is: use the normalized search id
+ number = json_data["number"]
+ 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) # 创建文件夹
+ # =======================================================================刮削模式
+ 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, json_data['actor_list']) # 打印文件
+ pasteFileToFolder(filepath, path, number, c_word) # 移动文件
+ # =======================================================================整理模式
+ elif program_mode == '2':
+ pasteFileToFolder_mode2(filepath, path, multi_part, number, part, c_word) # 移动文件