433 lines
18 KiB
Python
433 lines
18 KiB
Python
###-----检测psd是否合格，通过的psd压缩-----
|
||
|
||
import os
|
||
import time
|
||
import config
|
||
from psd_tools import PSDImage
|
||
from PIL import Image
|
||
import json
|
||
import utils
|
||
import sys
|
||
|
||
curr = os.path.dirname(sys.argv[0])
|
||
if curr != "":
|
||
os.system("cd " + curr)
|
||
|
||
|
||
def check_psd(psd_path):
|
||
psd_name = os.path.split(psd_path)[1]
|
||
names = psd_name.split('_')
|
||
log(f'开始检测：{psd_name}')
|
||
error_log = []
|
||
if len(names) != 3 and len(names) != 4 and len(names) != 5 and len(names) != 6:
|
||
error_log.append(config.error_log.psd_name_error.value)
|
||
# elif not names[0][0].islower():
|
||
# error_log.append(f'{psd_name}:{config.error_log.psd_name_error.value}')
|
||
# elif not names[1][0].islower():
|
||
# error_log.append(f'{psd_name}:{config.error_log.psd_name_error.value}')
|
||
# elif not names[2][0].isdigit():
|
||
# error_log.append(f'{psd_name}:{config.error_log.psd_name_error.value}')
|
||
|
||
try:
|
||
psd_point_num = int(names[1][:2])
|
||
except Exception as e:
|
||
print(e)
|
||
error_log.append(config.error_log.psd_name_error.value)
|
||
return
|
||
|
||
psd = PSDImage.open(psd_path)
|
||
|
||
color_mode = get_psd_color_mode(psd_path)
|
||
if color_mode == 'CMYK':
|
||
error_log.append(f'{psd_name}:{config.error_log.psd_cmyk.value}')
|
||
print("PSD 文件的色彩空间是 CMYK")
|
||
# elif color_mode == 'RGB':
|
||
# print("PSD 文件的色彩空间是 RGB")
|
||
|
||
psd_width = psd.width
|
||
psd_height = psd.height
|
||
|
||
# print(f'{psd.size}')
|
||
|
||
if (psd_width != config.psd_standard_width and psd_width != config.psd_standard_width2) or (psd_height != config.psd_standard_height and psd_height != config.psd_standard_height2):
|
||
error_log.append(f'{psd_name}:{config.error_log.psd_size_error.value}')
|
||
|
||
|
||
titem_name = 'titem'
|
||
tfull_name = 'tfull'
|
||
tmask_name = 'tmask'
|
||
tlight_name = 'tlight'
|
||
tshadow_name = 'tshadow'
|
||
group_name = 'group_'
|
||
|
||
all_items = []
|
||
all_full = {}
|
||
all_mask = []
|
||
all_tlight = []
|
||
all_tshadow = []
|
||
all_group = []
|
||
all_group_num = []
|
||
# all_color_it_full = []
|
||
# all_color_it_item = []
|
||
is_exist_base = False
|
||
is_exist_titem_group = False
|
||
all_occlusion_degree = []
|
||
|
||
for layer in psd:
|
||
if ' ' in layer.name:
|
||
error_log.append(f'{psd_name}:{layer.name}:{config.error_log.name_contains_spaces.value}')
|
||
if layer.name == 'nouse' and layer.is_group():
|
||
continue
|
||
if layer.name == titem_name and layer.is_group():
|
||
is_exist_titem_group = True
|
||
for item_layer in layer:
|
||
if ' ' in item_layer.name:
|
||
error_log.append(f'{psd_name}:{item_layer.name}:{config.error_log.name_contains_spaces.value}')
|
||
# item_names = item_layer.name.split('_')
|
||
if titem_name in item_layer.name:
|
||
if item_layer.name in all_items: ##or item_layer.name in all_color_it_item:
|
||
error_log.append(f'{psd_name}:{item_layer.name}:{config.error_log.exit_repeat_layer.value}')
|
||
else:
|
||
# if 'titemcolor' in item_layer.name:
|
||
# all_color_it_item.append(item_layer.name)
|
||
# else:
|
||
all_items.append(item_layer.name)
|
||
else:
|
||
error_log.append(f'{psd_name}:{item_layer.name}:{config.error_log.psd_item_name_error.value}')
|
||
|
||
elif group_name in layer.name:
|
||
try:
|
||
group_num_str = layer.name.replace(group_name, "")
|
||
group_num = int(group_num_str)
|
||
except Exception as e:
|
||
error_log.append(f'{psd_name}:{layer.name}:{config.error_log.group_name_error.value}')
|
||
print(e)
|
||
continue
|
||
if group_num <= 0 or group_num > psd_point_num:
|
||
error_log.append(f'{psd_name}:{layer.name}:{config.error_log.group_name_error.value}')
|
||
continue
|
||
else:
|
||
all_group_num.append(group_num)
|
||
|
||
if not layer.is_group():
|
||
error_log.append(f'{psd_name}:{layer.name}:{config.error_log.group_name_error.value}')
|
||
elif layer.name in all_group:
|
||
error_log.append(f'{psd_name}:{layer.name}:{config.error_log.exit_repeat_group.value}')
|
||
else:
|
||
all_group.append(layer.name)
|
||
is_exit_full = False
|
||
for child_layer in layer:
|
||
if group_num_str not in child_layer.name:
|
||
error_log.append(f'{psd_name}:{child_layer.name}:{config.error_log.child_layer_not_in_group.value}')
|
||
|
||
if child_layer.is_group():
|
||
error_log.append(f'{psd_name}:{child_layer.name}:{config.error_log.child_layer_is_group.value}')
|
||
else:
|
||
if tfull_name in child_layer.name:
|
||
if child_layer.name in all_full.keys():
|
||
error_log.append(f'{psd_name}:{child_layer.name}:{config.error_log.exit_repeat_layer.value}')
|
||
else:
|
||
all_full[child_layer.name] = (child_layer.left, child_layer.top, child_layer.width, child_layer.height)
|
||
is_exit_full = True
|
||
elif tmask_name in child_layer.name:
|
||
if child_layer.name in all_mask:
|
||
error_log.append(f'{psd_name}:{child_layer.name}:{config.error_log.exit_repeat_layer.value}')
|
||
else:
|
||
all_mask.append(child_layer.name)
|
||
elif tshadow_name in child_layer.name:
|
||
if child_layer.name in all_tshadow:
|
||
error_log.append(f'{psd_name}:{child_layer.name}:{config.error_log.exit_repeat_layer.value}')
|
||
else:
|
||
all_tshadow.append(child_layer.name)
|
||
elif tlight_name in child_layer.name:
|
||
if child_layer.name in all_tlight:
|
||
error_log.append(f'{psd_name}:{child_layer.name}:{config.error_log.exit_repeat_layer.value}')
|
||
else:
|
||
all_tlight.append(child_layer.name)
|
||
else:
|
||
error_log.append(
|
||
f'{psd_name}:{child_layer.name}:{config.error_log.child_layer_unknown.value}')
|
||
if not is_exit_full:
|
||
error_log.append(f'{psd_name}:{layer.name}:{config.error_log.group_tfull_not_exit.value}')
|
||
elif layer.name == 'base':
|
||
if is_exist_base:
|
||
error_log.append(f'{psd_name}:{config.error_log.exit_more_base.value}')
|
||
else:
|
||
is_exist_base = True
|
||
else:
|
||
error_log.append(f'{psd_name}:{layer.name}:{config.error_log.layer_not_need.value}')
|
||
|
||
if not is_exist_base:
|
||
error_log.append(f'{psd_name}:{config.error_log.psd_not_exit_base.value}')
|
||
if not is_exist_titem_group:
|
||
error_log.append(f'{psd_name}:{config.error_log.item_group_not_exit.value}')
|
||
|
||
for i in range(1,psd_point_num + 1):
|
||
if i not in all_group_num:
|
||
error_log.append(f'{psd_name}:{i}:{config.error_log.group_lack.value}')
|
||
|
||
all_item_count = len(all_items)
|
||
if len(all_full) != all_item_count:
|
||
#or ( len(all_color_it_item) != 0 and len(all_color_it_item) != all_item_count) or (
|
||
#len(all_color_it_full) != 0 and len(all_color_it_full) != all_item_count):
|
||
error_log.append(f'{psd_name}:{config.error_log.item_or_full_num_error.value}')
|
||
|
||
for item in all_items:
|
||
full_name = item.replace('item', 'full')
|
||
if full_name not in all_full.keys():
|
||
error_log.append(f'{psd_name}:{item}:{config.error_log.psd_not_full.value}')
|
||
# if len(all_color_it_item) > 0 and len(all_color_it_full) > 0:
|
||
# for item in all_items:
|
||
# fullcolor_name = item.replace('item', 'fullcolor')
|
||
# itemcolor_name = item.replace('item', 'itemcolor')
|
||
# if fullcolor_name not in all_color_it_full:
|
||
# error_log.append(f'{psd_name}:{item}:{config.error_log.psd_not_fullcolor.value}')
|
||
# if itemcolor_name not in all_color_it_item:
|
||
# error_log.append(f'{psd_name}:{item}:{config.error_log.psd_not_itemcolor.value}')
|
||
|
||
# if not os.path.exists('./test'):
|
||
# os.mkdir('./test')
|
||
|
||
def find_layer_by_name(layers, target_name):
|
||
"""递归搜索图层中指定名称的图层"""
|
||
for layer in layers:
|
||
if layer.name == target_name:
|
||
return layer
|
||
if layer.is_group(): # 检查是否为组
|
||
found = find_layer_by_name(layer, target_name)
|
||
if found:
|
||
return found
|
||
return None
|
||
|
||
# 计算mask和full的遮罩百分比
|
||
for cur_mask in all_mask:
|
||
# img_mask = psd.composite(layer_filter=lambda mask_layer: mask_layer.name == cur_mask)
|
||
layer_mask = find_layer_by_name(psd,cur_mask)
|
||
full_name = cur_mask.replace('mask', 'full')
|
||
cur_full_rect = None
|
||
if full_name in all_full.keys():
|
||
cur_full_rect = all_full[full_name]
|
||
# img_full = psd.composite(layer_filter=lambda full_layer: full_layer.name == full_name)
|
||
layer_full = find_layer_by_name(psd, full_name)
|
||
if layer_mask is None or layer_full is None:
|
||
error_log.append(f'{psd_name}:{cur_mask}:{config.error_log.not_find_mask_full.value}')
|
||
break
|
||
img_mask = psd.composite(layer_mask)
|
||
img_full = psd.composite(layer_full)
|
||
|
||
# img_mask.save(f'./test/{cur_mask}.png')
|
||
# img_full.save(f'./test/{full_name}.png')
|
||
per = get_item_mask_contact_ratio(img_full, img_mask, cur_full_rect)
|
||
all_occlusion_degree.append(f'{cur_mask}遮挡{full_name}百分比：{per}\n')
|
||
# print(f'{cur_mask}遮挡{full_name}百分比：{per}')
|
||
else:
|
||
error_log.append(f'{psd_name}:{cur_mask}:{config.error_log.psd_mask_not_full.value}')
|
||
|
||
is_error = len(error_log) > 0
|
||
is_error_text = '否' if is_error else '是'
|
||
if is_error:
|
||
error_psd_dic[psd_name] = error_log
|
||
log(f'{psd_name}:检测完毕，是否检测通过：{is_error_text}')
|
||
for clog in error_log:
|
||
log(clog)
|
||
|
||
if not is_error:
|
||
occlusion_degree_file_path = f'./OcclusionDegree/{psd_name}.txt'
|
||
if not os.path.exists('./OcclusionDegree/'):
|
||
os.mkdir('./OcclusionDegree/')
|
||
if not os.path.exists(occlusion_degree_file_path):
|
||
with open(occlusion_degree_file_path, 'w') as occlusion_degree_file:
|
||
for text in all_occlusion_degree:
|
||
occlusion_degree_file.writelines(text)
|
||
pass
|
||
|
||
return len(error_log) == 0
|
||
|
||
|
||
def get_item_mask_contact_ratio(full_image, mask_image, full_rect):
|
||
# full_image = full_image.convert('RGBA')
|
||
# mask_image = mask_image.convert('RGBA')
|
||
x, y, w, h = full_rect
|
||
full_pix_count = 0
|
||
mask_pix_count = 0
|
||
for cur_x in range(x, x + w):
|
||
for cur_y in range(y, y + h):
|
||
full_pix = full_image.getpixel((cur_x, cur_y))
|
||
mask_pix = mask_image.getpixel((cur_x, cur_y))
|
||
if not full_pix == (255, 255, 255, 0):
|
||
full_pix_count += 1
|
||
if not mask_pix == (255, 255, 255, 0) and not mask_pix == full_pix:
|
||
mask_pix_count += 1
|
||
# print(f'mask_pix_count::: {mask_pix_count}')
|
||
# print(f'full_pix_count {full_pix_count}')
|
||
percentage = "{:.2%}".format(float(mask_pix_count) / float(full_pix_count))
|
||
return percentage
|
||
|
||
|
||
def log(content, new_line=True):
|
||
curr_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
||
if new_line:
|
||
print(curr_time + ' ' + content)
|
||
else:
|
||
print(curr_time + ' ' + content, end=' ')
|
||
|
||
|
||
###生成version文件
|
||
def make_version_file(remake):
|
||
version_file = 'version.json';
|
||
|
||
if remake:
|
||
if os.path.exists(version_file):
|
||
os.remove(version_file)
|
||
|
||
if not os.path.exists(version_file):
|
||
with open(version_file, 'w') as f:
|
||
json.dump({}, f, sort_keys=True, indent=4)
|
||
|
||
|
||
###加载版本文件
|
||
def load_version():
|
||
global versions
|
||
make_version_file(False);
|
||
with open('version.json', 'r') as f:
|
||
try:
|
||
versions = json.load(f)
|
||
except:
|
||
# 删除版本文件并重新生成
|
||
make_version_file(True);
|
||
|
||
|
||
def write_version():
|
||
with open('version.json', 'w') as f:
|
||
json.dump(versions, f, sort_keys=True, indent=4)
|
||
|
||
|
||
def convert_2_zip_file(path, is_name_end_zip=False):
|
||
file_info_arr = os.path.split(path)
|
||
file_path = file_info_arr[0]
|
||
psd_name = os.path.split(path)[1].replace('.psd', '')
|
||
psd_type = psd_name.split('_')[0]
|
||
if is_name_end_zip:
|
||
psd_path = os.path.join('./', file_info_arr[1])
|
||
os.chdir(file_path)
|
||
# 检测成功，压缩到zip
|
||
utils.zip_file(psd_path, f'./{psd_name}.zip')
|
||
os.remove(psd_path)
|
||
os.chdir('../')
|
||
os.chdir('../')
|
||
else:
|
||
zip_out_path = os.path.join('./Level', psd_type)
|
||
if not os.path.exists(zip_out_path):
|
||
os.mkdir(zip_out_path)
|
||
psd_path = os.path.join('./', file_info_arr[1])
|
||
os.chdir(file_path)
|
||
# 检测成功，压缩到zip
|
||
utils.zip_file(psd_path, f'../{zip_out_path}/{psd_name}.zip')
|
||
os.remove(psd_path)
|
||
os.chdir('../')
|
||
|
||
def get_psd_color_mode(psd_path):
|
||
with open(psd_path, "rb") as f:
|
||
# 读取文件头信息
|
||
header = f.read(24)
|
||
|
||
# 获取颜色模式字节
|
||
color_mode_byte = header[10]
|
||
|
||
if color_mode_byte == 0:
|
||
return "Bitmap"
|
||
elif color_mode_byte == 2:
|
||
return "RGB"
|
||
elif color_mode_byte == 4:
|
||
return "CMYK"
|
||
else:
|
||
return "Unknown"
|
||
|
||
|
||
if __name__ == '__main__':
|
||
global all_psd_count
|
||
all_psd_count = 0
|
||
global pass_psd_count
|
||
pass_psd_count = 0
|
||
global no_change_psd_count
|
||
no_change_psd_count = 0
|
||
global error_psd_dic
|
||
error_psd_dic = {}
|
||
global zip_fail_count
|
||
zip_fail_count = 0
|
||
global re_zip_count
|
||
re_zip_count = 0
|
||
|
||
load_version()
|
||
psd_name_list = []
|
||
is_force = sys.argv[1] if len(sys.argv) > 1 else False
|
||
# is_force = True
|
||
for root, dirs, files in os.walk('./'):
|
||
for name in files:
|
||
cur_file_path = os.path.join(root, name)
|
||
|
||
if name.endswith('.psd'):
|
||
all_psd_count += 1
|
||
|
||
if name.endswith('.psd') or name.endswith('.zip'):
|
||
file_info_arr = os.path.split(cur_file_path)
|
||
file_path = file_info_arr[0]
|
||
psd_name = file_info_arr[1]
|
||
if psd_name not in psd_name_list:
|
||
psd_name_list.append(psd_name)
|
||
else:
|
||
log("存在重复名称关卡：{}".format(psd_name))
|
||
else:
|
||
continue
|
||
|
||
if name.endswith('.psd') or (name.endswith('.zip') and is_force):
|
||
file_info_arr = os.path.split(cur_file_path)
|
||
file_path = file_info_arr[0]
|
||
psd_name = file_info_arr[1]
|
||
|
||
hash_old = ""
|
||
if psd_name in versions:
|
||
hash_old = versions[psd_name]
|
||
|
||
is_zip = name.endswith('.zip')
|
||
if os.path.exists(cur_file_path):
|
||
hash_now = ""
|
||
if name.endswith('.psd'):
|
||
hash_now = utils.calc_hash(cur_file_path)
|
||
if hash_now != hash_old or is_force:
|
||
psd_file_path = cur_file_path
|
||
if is_zip:
|
||
utils.unzip_file(cur_file_path)
|
||
psd_file_path = "{}.psd".format(os.path.splitext(psd_file_path)[0])
|
||
hash_now = utils.calc_hash(psd_file_path)
|
||
try:
|
||
is_passed = check_psd(psd_file_path)
|
||
if is_passed:
|
||
# 检测成功，压缩到zip
|
||
convert_2_zip_file(psd_file_path, is_zip)
|
||
if is_zip:
|
||
re_zip_count += 1
|
||
versions[psd_name] = hash_now
|
||
write_version()
|
||
if name.endswith('.psd'):
|
||
pass_psd_count += 1
|
||
except Exception as e:
|
||
zip_fail_count += 1
|
||
log(f'检测或者压缩失败::{psd_file_path},{e}')
|
||
error_psd_dic[psd_name] = [f'{psd_file_path},{e}']
|
||
elif hash_now == hash_old:
|
||
no_change_psd_count += 1
|
||
convert_2_zip_file(cur_file_path, is_zip)
|
||
log(f'psd压缩完毕，总共psd数量：{all_psd_count}')
|
||
log(f'成功压缩新增psd个数：{pass_psd_count}')
|
||
log(f'成功压缩无改动psd个数：{no_change_psd_count}')
|
||
log(f'未检测通过psd个数：{len(error_psd_dic)}')
|
||
log(f'重新压缩psd数量：{re_zip_count}')
|
||
log(f'压缩失败个数：{zip_fail_count}')
|
||
if len(error_psd_dic) > 0:
|
||
log(f'以下psd出现问题：')
|
||
for error_psd_name, error_psd_log in error_psd_dic.items():
|
||
log(f'psd名称：{error_psd_name}')
|
||
for error_log in error_psd_log:
|
||
log(error_log)
|