错误处理和日志处理
This commit is contained in:
11
main.py
11
main.py
@@ -18,6 +18,7 @@ emoji = yield_emoji_path(emoji_folder)
|
|||||||
|
|
||||||
gen = Generator(api_key)
|
gen = Generator(api_key)
|
||||||
|
|
||||||
|
def main():
|
||||||
for i in range(get_emoji_count(emoji_folder)):
|
for i in range(get_emoji_count(emoji_folder)):
|
||||||
emoji_file = next(emoji)
|
emoji_file = next(emoji)
|
||||||
emoji_uuid = ''
|
emoji_uuid = ''
|
||||||
@@ -30,5 +31,13 @@ for i in range(get_emoji_count(emoji_folder)):
|
|||||||
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()
|
||||||
Reference in New Issue
Block a user