1 Star 0 Fork 39

Xking/DeepCreamPy

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
file.py 2.03 KB
一键复制 编辑 原始数据 按行查看 历史
deeppomf 提交于 2019-09-29 17:38 . Add variations feature
import os
def check_file(input_dir, output_dir, release_version = True):
file_list = []
output_file_list = []
files_removed = []
input_dir = os.listdir(input_dir)
output_dir = os.listdir(output_dir)
for file_in in input_dir:
if not file_in.startswith('.'):
file_list.append(file_in)
if release_version is True:
print("\nChecking valid files...")
for file_out in output_dir:
if file_out.lower().endswith('.png'):
output_file_list.append(file_out)
# solving https://github.com/deeppomf/DeepCreamPy/issues/25
# appending in list with reason as tuple (file name, reason)
for lhs in file_list:
lhs.lower()
if not lhs.lower().endswith('.png') :
files_removed.append((lhs, 0))
for rhs in output_file_list:
if(lhs == rhs):
files_removed.append((lhs, 1))
# seperated detecting same file names and deleting file name list
# just in case of index_error and show list of files which will not go though
# decensor process
print("\n### These files will not be decensored for following reason ###\n")
error_messages(file_list, files_removed)
input("\nPress anything to continue...")
print("\n###################################\n")
return file_list, files_removed
def error_messages(file_list, files_removed):
if files_removed is None:
return
for remove_this, reason in files_removed:
if file_list is not None:
file_list.remove(remove_this)
if reason == 0:
print(" REMOVED : (" + str(remove_this) +") is not PNG file format")
elif reason == 1:
print(" REMOVED : (" + str(remove_this) +") already exists")
elif reason == 2:
print(" REMOVED : (" + str(remove_this) +") file unreadable")
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/StarGod/DeepCreamPy.git
git@gitee.com:StarGod/DeepCreamPy.git
StarGod
DeepCreamPy
DeepCreamPy
master

搜索帮助

23e8dbc6 1850385 7e0993f3 1850385