Add try-catch block for error handling

pull/571/head
zanllp 2024-03-30 10:45:59 +08:00
parent 428664a2a4
commit 7e37356d2f
1 changed files with 7 additions and 1 deletions

View File

@ -2,6 +2,7 @@ from scripts.iib.parsers.comfyui import ComfyUIParser
from scripts.iib.parsers.sd_webui import SdWebUIParser
from scripts.iib.parsers.fooocus import FooocusParser
from scripts.iib.parsers.model import ImageGenerationInfo
from scripts.iib.logger import logger
from PIL import Image
@ -10,5 +11,10 @@ def parse_image_info(image_path: str) -> ImageGenerationInfo:
with Image.open(image_path) as img:
for parser in parsers:
if parser.test(img, image_path):
return parser.parse(img, image_path)
try:
return parser.parse(img, image_path)
except Exception as e:
logger.error(e, stack_info=True)
print(e)
return ImageGenerationInfo()
raise Exception("matched parser is not found")