fix: [qrcode extractor] fix catch cv2 exception

ail-project · Oct 3, 2024 · 970a169 · 970a169
1 parent ef93254
commit 970a169
Showing 1 changed file with 16 additions and 10 deletions.
diff --git a/bin/modules/QrCodeReader.py b/bin/modules/QrCodeReader.py
@@ -69,16 +69,22 @@ def extract_qrcode(self, path):
 
         if not contents:
             detector = cv2.QRCodeDetector()
-            qr, decodeds, qarray, _ = detector.detectAndDecodeMulti(image)
-            if qr:
-                qr_codes = True
-                for d in decodeds:
-                    if d:
-                        contents.append(d)
-            data_qr, box, qrcode_image = detector.detectAndDecode(image)
-            if data_qr:
-                contents.append(data_qr)
-                qr_codes = True
+            try:
+                qr, decodeds, qarray, _ = detector.detectAndDecodeMulti(image)
+                if qr:
+                    qr_codes = True
+                    for d in decodeds:
+                        if d:
+                            contents.append(d)
+            except cv2.error as e:
+                self.logger.error(f'{e}: {self.obj.get_global_id()}')
+            try:
+                data_qr, box, qrcode_image = detector.detectAndDecode(image)
+                if data_qr:
+                    contents.append(data_qr)
+                    qr_codes = True
+            except cv2.error as e:
+                self.logger.error(f'{e}: {self.obj.get_global_id()}')
 
         if qr_codes and not contents:
             # # # # 0.5s per image