177 lines
7.8 KiB
Python
Executable File
177 lines
7.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import glob
|
||
import os
|
||
import time
|
||
import re
|
||
import sys
|
||
from ADC_function import *
|
||
import json
|
||
import subprocess
|
||
import shutil
|
||
from configparser import ConfigParser
|
||
|
||
version='0.11.7'
|
||
os.chdir(os.getcwd())
|
||
|
||
input_dir='.' # 电影的读取与输出路径, 默认为当前路径
|
||
|
||
config = ConfigParser()
|
||
config.read(config_file, encoding='UTF-8')
|
||
|
||
def UpdateCheck():
|
||
if UpdateCheckSwitch() == '1':
|
||
html = json.loads(get_html('https://raw.githubusercontent.com/wenead99/AV_Data_Capture/master/update_check.json'))
|
||
if not version == html['version']:
|
||
print('[*] * New update ' + html['version'] + ' *')
|
||
print('[*] * Download *')
|
||
print('[*] ' + html['download'])
|
||
print('[*]=====================================')
|
||
else:
|
||
print('[+]Update Check disabled!')
|
||
|
||
def set_directory(): # 设置读取与存放路径
|
||
global input_dir
|
||
# 配置项switch为1且定义了新的路径时, 更改默认存取路径
|
||
if config['directory_capture']['switch'] == '1':
|
||
custom_input = config['directory_capture']['input_directory']
|
||
if custom_input != '': # 自定义了输入路径
|
||
input_dir = format_path(custom_input)
|
||
# 若自定义了输入路径, 输出路径默认在输入路径下
|
||
CreatFolder(input_dir)
|
||
#print('[+]Working directory is "' + os.getcwd() + '".')
|
||
#print('[+]Using "' + input_dir + '" as input directory.')
|
||
|
||
def format_path(path): # 使路径兼容Linux与MacOS
|
||
if path.find('\\'): # 是仅兼容Windows的路径格式
|
||
path_list=path.split('\\')
|
||
path='/'.join(path_list) # 转换为可移植的路径格式
|
||
return path
|
||
|
||
|
||
def movie_lists():
|
||
a2 = glob.glob( input_dir + "/*.mp4")
|
||
b2 = glob.glob( input_dir + "/*.avi")
|
||
c2 = glob.glob( input_dir + "/*.rmvb")
|
||
d2 = glob.glob( input_dir + "/*.wmv")
|
||
e2 = glob.glob( input_dir + "/*.mov")
|
||
f2 = glob.glob( input_dir + "/*.mkv")
|
||
g2 = glob.glob( input_dir + "/*.flv")
|
||
h2 = glob.glob( input_dir + "/*.ts")
|
||
total = a2 + b2 + c2 + d2 + e2 + f2 + g2 + h2
|
||
return total
|
||
def CreatFolder(folder_path):
|
||
if not os.path.exists(folder_path): # 新建文件夹
|
||
try:
|
||
print('[+]Creating ' + folder_path)
|
||
os.makedirs(folder_path)
|
||
except:
|
||
print("[-]failed!can not be make folder '"+folder_path+"'\n[-](Please run as Administrator)")
|
||
os._exit(0)
|
||
def lists_from_test(custom_nuber): #电影列表
|
||
a=[]
|
||
a.append(custom_nuber)
|
||
return a
|
||
def CEF(path):
|
||
files = os.listdir(path) # 获取路径下的子文件(夹)列表
|
||
for file in files:
|
||
try: #试图删除空目录,非空目录删除会报错
|
||
os.removedirs(path + '/' + file) # 删除这个空文件夹
|
||
print('[+]Deleting empty folder',path + '/' + file)
|
||
except:
|
||
a=''
|
||
def rreplace(self, old, new, *max):
|
||
#从右开始替换文件名中内容，源字符串，将被替换的子字符串， 新字符串，用于替换old子字符串，可选字符串, 替换不超过 max 次
|
||
count = len(self)
|
||
if max and str(max[0]).isdigit():
|
||
count = max[0]
|
||
return new.join(self.rsplit(old, count))
|
||
def getNumber(filepath):
|
||
try: # 试图提取番号
|
||
# ====番号获取主程序====
|
||
try: # 普通提取番号 主要处理包含减号-的番号
|
||
filepath1 = filepath.replace("_", "-")
|
||
filepath1.strip('22-sht.me').strip('-HD').strip('-hd')
|
||
filename = str(re.sub("\[\d{4}-\d{1,2}-\d{1,2}\] - ", "", filepath1)) # 去除文件名中时间
|
||
file_number = re.search('\w+-\d+', filename).group()
|
||
return file_number
|
||
except: # 提取不含减号-的番号
|
||
try: # 提取东京热番号格式 n1087
|
||
filename1 = str(re.sub("h26\d", "", filepath)).strip('Tokyo-hot').strip('tokyo-hot')
|
||
filename0 = str(re.sub(".*?\.com-\d+", "", filename1)).strip('_')
|
||
if '-C.' in filepath or '-c.' in filepath:
|
||
cn_sub = '1'
|
||
file_number = str(re.search('n\d{4}', filename0).group(0))
|
||
return file_number
|
||
except: # 提取无减号番号
|
||
filename1 = str(re.sub("h26\d", "", filepath)) # 去除h264/265
|
||
filename0 = str(re.sub(".*?\.com-\d+", "", filename1))
|
||
file_number2 = str(re.match('\w+', filename0).group())
|
||
if '-C.' in filepath or '-c.' in filepath:
|
||
cn_sub = '1'
|
||
file_number = str(file_number2.replace(re.match("^[A-Za-z]+", file_number2).group(),
|
||
re.match("^[A-Za-z]+", file_number2).group() + '-'))
|
||
return file_number
|
||
# if not re.search('\w-', file_number).group() == 'None':
|
||
# file_number = re.search('\w+-\w+', filename).group()
|
||
# 上面是插入减号-到番号中
|
||
# ====番号获取主程序=结束===
|
||
except Exception as e: # 番号提取异常
|
||
print('[-]' + str(os.path.basename(filepath)) + ' Cannot catch the number :')
|
||
print('[-]' + str(os.path.basename(filepath)) + ' :', e)
|
||
print('[-]Move ' + os.path.basename(filepath) + ' to failed folder')
|
||
#print('[-]' + filepath + ' -> ' + output_dir + '/failed/')
|
||
#shutil.move(filepath, output_dir + '/failed/')
|
||
except IOError as e2:
|
||
print('[-]' + str(os.path.basename(filepath)) + ' Cannot catch the number :')
|
||
print('[-]' + str(os.path.basename(filepath)) + ' :', e2)
|
||
#print('[-]' + filepath + ' -> ' + output_dir + '/failed/')
|
||
#shutil.move(filepath, output_dir + '/failed/')
|
||
|
||
def RunCore(movie):
|
||
# 异步调用core.py, core.py作为子线程执行, 本程序继续执行.
|
||
if os.path.exists('core.py'):
|
||
cmd_arg=[sys.executable,'core.py',movie,'--number',getNumber(movie)] #从py文件启动（用于源码py）
|
||
elif os.path.exists('core.exe'):
|
||
cmd_arg=['core.exe',movie,'--number',getNumber(movie)] #从exe启动（用于EXE版程序）
|
||
elif os.path.exists('core.py') and os.path.exists('core.exe'):
|
||
cmd_arg=[sys.executable,'core.py',movie,'--number',getNumber(movie)] #从py文件启动（用于源码py）
|
||
process=subprocess.Popen(cmd_arg)
|
||
return process
|
||
|
||
if __name__ =='__main__':
|
||
print('[*]===========AV Data Capture===========')
|
||
print('[*] Version '+version)
|
||
print('[*]=====================================')
|
||
UpdateCheck()
|
||
os.chdir(os.getcwd())
|
||
set_directory()
|
||
|
||
|
||
count = 0
|
||
movies = movie_lists()
|
||
count_all = str(len(movies))
|
||
print('[+]Find ' + str(len(movies)) + ' movies.')
|
||
process_list=[]
|
||
for movie in movies: #遍历电影列表 交给core处理
|
||
num=getNumber(movie) # 获取番号
|
||
if num is None:
|
||
movies.remove(movie) # 未获取到番号, 则将影片从列表移除
|
||
count_all=count_all-1
|
||
continue
|
||
print("[!]Making Data for [" + movie + "], the number is [" + num + "]")
|
||
process=RunCore(movie)
|
||
process_list.append(process)
|
||
print("[*]=====================================")
|
||
for i in range(len(movies)):
|
||
process_list[i].communicate()
|
||
percentage = str((i+1)/int(count_all)*100)[:4]+'%'
|
||
print('[!] - '+percentage+' ['+str(count)+'/'+count_all+'] -')
|
||
print("[!]The [" + getNumber(movies[i]) + "] process is done.")
|
||
print("[*]=====================================")
|
||
|
||
CEF(input_dir)
|
||
print("[+]All finished!!!")
|
||
input("[+][+]Press enter key exit, you can check the error messge before you exit.\n[+][+]按回车键结束，你可以在结束之前查看和错误信息。")
|