From 591159323ce7f5f4b928cea24481f25690a6cbf5 Mon Sep 17 00:00:00 2001 From: Fam Zheng Date: Fri, 14 Mar 2025 21:02:39 +0000 Subject: [PATCH] emcli: Allow roi 404 --- scripts/emcli | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/scripts/emcli b/scripts/emcli index f5ebe8c..f916fb5 100755 --- a/scripts/emcli +++ b/scripts/emcli @@ -89,8 +89,10 @@ class GetScanDataCommand(SubCommand): f.write(sd['message'] + "\n") with open(f"{args.output}/{i}-frame.jpg", "wb") as f: f.write(self.get_image(sd['image'])) - with open(f"{args.output}/{i}-roi.jpg", "wb") as f: - f.write(self.get_roi(sd['code'])) + roi = self.get_roi(sd['code']) + if roi: + with open(f"{args.output}/{i}-roi.jpg", "wb") as f: + f.write(roi) def get_scan_data(self, i): server = self.get_server() @@ -112,7 +114,8 @@ class GetScanDataCommand(SubCommand): token = self.get_env()['token'] url = f'{server}/api/v1/code-feature-roi/?token={token}&code={code}' r = requests.get(url) - r.raise_for_status() + if r.status_code == 404: + return None return r.content class UploadRoiCommand(SubCommand):