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

Update core.py

Browse Source 
支持分集
This commit is contained in:
mo_yy 2019-08-19 10:18:32 +08:00
parent 1e407ef962
commit fd05706636
1 changed files with 32 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
core.py
View File

@ -41,6 +41,7 @@ imagecut=''
tag=[] tag=[]
cn_sub='' cn_sub=''
multi_part=0 multi_part=0
part=''
path='' path=''
houzhui='' houzhui=''
website='' website=''
@ -273,7 +274,13 @@ def imageDownload(filepath): #封面是否下载成功，否则移动到failed
if DownloadFileWithFilename(cover, number + '.jpg', path) == 'failed': if DownloadFileWithFilename(cover, number + '.jpg', path) == 'failed':
moveFailedFolder() moveFailedFolder()
DownloadFileWithFilename(cover, number + '.jpg', path) DownloadFileWithFilename(cover, number + '.jpg', path)
print('[+]Image Downloaded!', path + '/' + number + '.jpg') if multi_part == 1:
old_name = os.path.join(path, number + '.jpg')
new_name = os.path.join(path, number + part + '.jpg')
os.rename(old_name, new_name)
print('[+]Image Downloaded!', path + '/' + number + part + '.jpg')
else:
print('[+]Image Downloaded!', path + '/' + number + '.jpg')
elif option == 'plex': elif option == 'plex':
if DownloadFileWithFilename(cover, 'fanart.jpg', path) == 'failed': if DownloadFileWithFilename(cover, 'fanart.jpg', path) == 'failed':
moveFailedFolder() moveFailedFolder()
@ -448,9 +455,22 @@ def renameJpgToBackdrop_copy():
if option == 'emby': if option == 'emby':
shutil.copy(path + '/' + number + '.jpg', path + '/Backdrop.jpg') shutil.copy(path + '/' + number + '.jpg', path + '/Backdrop.jpg')
def renameBackdropToJpg_copy():
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 + '/Backdrop.jpg', path + '/' + number + '.jpg')
print('[+]Image Downloaded!', path + '/' + number + '.jpg')
def get_part(filepath):
if re.search('-CD\d+', filepath):
return re.findall('-CD\d+', filepath)[0]
if __name__ == '__main__': if __name__ == '__main__':
filepath=argparse_get_file()[0] #影片的路径 filepath=argparse_get_file()[0] #影片的路径
if '-CD' in filepath or '-cd' in filepath:
multi_part = 1
part = get_part(filepath)
if '-c.' in filepath or '-C.' in filepath or '中文' in filepath or '字幕' in filepath: if '-c.' in filepath or '-C.' in filepath or '中文' in filepath or '字幕' in filepath:
cn_sub='1' cn_sub='1'
@ -467,11 +487,17 @@ if __name__ == '__main__':
getDataFromJSON(number) # 定义番号 getDataFromJSON(number) # 定义番号
creatFolder() # 创建文件夹 creatFolder() # 创建文件夹
if program_mode == '1': if program_mode == '1':
imageDownload(filepath) # creatFoder会返回番号路径 if part == '-CD1' or multi_part == 0:
PrintFiles(filepath) # 打印文件 imageDownload(filepath) # creatFoder会返回番号路径
smallCoverCheck() if multi_part == 1:
cutImage() # 裁剪图 number += part
PrintFiles(filepath) # 打印文件
smallCoverCheck()
cutImage() # 裁剪图
renameJpgToBackdrop_copy()
else:
number += part
renameBackdropToJpg_copy()
pasteFileToFolder(filepath, path) # 移动文件 pasteFileToFolder(filepath, path) # 移动文件
renameJpgToBackdrop_copy()
elif program_mode == '2': elif program_mode == '2':
pasteFileToFolder_mode2(filepath, path) # 移动文件 pasteFileToFolder_mode2(filepath, path) # 移动文件