再将其它几个makedirs()一起修正,去掉错上加错的提升到admin建议信息

This commit is contained in:
lededev
2021-10-03 10:43:54 +08:00
parent f52db0011c
commit 8ef87c285f
2 changed files with 26 additions and 20 deletions

View File

@@ -214,13 +214,11 @@ def movie_lists(root, conf, regexstr):
def create_failed_folder(failed_folder):
if not os.path.isdir(failed_folder): # 新建failed文件夹
if not os.path.exists(failed_folder): # 新建failed文件夹
try:
os.makedirs(failed_folder)
if not os.path.isdir(failed_folder):
raise
except:
print("[-]failed!can not be make folder 'failed'\n[-](Please run as Administrator)")
print(f"[-]Fatal error! Can not make folder '{failed_folder}'")
sys.exit(0)

40
core.py
View File

@@ -83,17 +83,19 @@ def create_folder(json_data, conf: config.Config): # 创建文件夹
location_rule = location_rule.replace(title, shorttitle)
path = os.path.join(success_folder, location_rule).strip()
if not os.path.isdir(path):
if not os.path.exists(path):
path = escape_path(path, conf.escape_literals())
try:
os.makedirs(path)
if not os.path.isdir(path):
raise
except:
path = success_folder + '/' + location_rule.replace('/[' + number + ')-' + title, "/number")
path = escape_path(path, conf.escape_literals())
try:
os.makedirs(path)
except:
print(f"[-]Fatal error! Can not make folder '{path}'")
sys.exit(0)
os.makedirs(path)
return path
@@ -106,10 +108,12 @@ def download_file_with_filename(url, filename, path, conf: config.Config, filepa
for i in range(configProxy.retry):
try:
if configProxy.enable:
if not os.path.isdir(path):
os.makedirs(path)
if not os.path.isdir(path):
raise IOError
if not os.path.exists(path):
try:
os.makedirs(path)
except:
print(f"[-]Fatal error! Can not make folder '{path}'")
sys.exit(0)
proxies = configProxy.proxies()
headers = {
'User-Agent': G_USER_AGENT}
@@ -121,10 +125,12 @@ def download_file_with_filename(url, filename, path, conf: config.Config, filepa
code.write(r.content)
return
else:
if not os.path.isdir(path):
os.makedirs(path)
if not os.path.isdir(path):
raise IOError
if not os.path.exists(path):
try:
os.makedirs(path)
except:
print(f"[-]Fatal error! Can not make folder '{path}'")
sys.exit(0)
headers = {
'User-Agent': G_USER_AGENT}
r = requests.get(url, timeout=configProxy.timeout, headers=headers)
@@ -224,10 +230,12 @@ def print_files(path, leak_word, c_word, naming_rule, part, cn_sub, json_data, f
else:
nfo_path = os.path.join(path,f"{number}{part}{leak_word}{c_word}.nfo")
try:
if not os.path.isdir(path):
os.makedirs(path)
if not os.path.isdir(path):
raise IOError
if not os.path.exists(path):
try:
os.makedirs(path)
except:
print(f"[-]Fatal error! can not make folder '{path}'")
sys.exit(0)
with open(nfo_path, "wt", encoding='UTF-8') as code:
print('<?xml version="1.0" encoding="UTF-8" ?>', file=code)
print("<movie>", file=code)