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):