错误处理和日志处理

This commit is contained in:
2025-12-14 00:05:04 +08:00
parent 01b0dde00a
commit f9e2eb6d76

33
main.py
View File

@@ -18,17 +18,26 @@ emoji = yield_emoji_path(emoji_folder)
gen = Generator(api_key) gen = Generator(api_key)
for i in range(get_emoji_count(emoji_folder)): def main():
emoji_file = next(emoji) for i in range(get_emoji_count(emoji_folder)):
emoji_uuid = '' emoji_file = next(emoji)
p = Path(emoji_file) emoji_uuid = ''
if p.suffix == ".jpg" or p.suffix == ".jpeg": p = Path(emoji_file)
emoji_uuid = p.stem if p.suffix == ".jpg" or p.suffix == ".jpeg":
image = compress_image(emoji_file) emoji_uuid = p.stem
image_path = Path(root_folder) / image image = compress_image(emoji_file)
image_path = Path(root_folder) / image
if check_emoji(emoji_uuid): if check_emoji(emoji_uuid):
description = gen.process_single_image(str(image_path)) description = gen.process_single_image(str(image_path))
update_description(emoji_uuid, description) update_description(emoji_uuid, description)
logger.info(f"图片 {emoji_file} 的描述词生成完毕,序号:{i} 其 description 为: {description}") if description:
logger.info(f"图片 {emoji_file} 的描述词生成完毕,序号:{i + 1} 其 description 为: {description}")
else:
logger.warning(f"图片 {emoji_file} 的描述词生成失败!序号:{i + 1}")
if __name__ == '__main__':
try:
main()
except KeyboardInterrupt:
exit()