AIL/ail-framework
7
0
Fork 0
mirror of https://github.com/ail-project/ail-framework.git synced 2024-11-15 02:58:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: [qrcode extractor] fix image/screenshot tag

Browse source
This commit is contained in:
terrtia 2024-10-02 16:41:23 +02:00
parent 32761fb30e
commit 4dbda5975a
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
bin/modules/QrCodeReader.py
View file

@ -60,6 +60,7 @@ class QrCodeReader(AbstractModule):
try: try:
decodeds = decode(image) decodeds = decode(image)
for decoded in decodeds: for decoded in decodeds:
qr_codes = True
if decoded.data: if decoded.data:
contents.append(decoded.data.decode()) contents.append(decoded.data.decode())
except ValueError as e: except ValueError as e:
@ -76,6 +77,7 @@ class QrCodeReader(AbstractModule):
data_qr, box, qrcode_image = detector.detectAndDecode(image) data_qr, box, qrcode_image = detector.detectAndDecode(image)
if data_qr: if data_qr:
contents.append(data_qr) contents.append(data_qr)
qr_codes = True
if qr_codes and not contents: if qr_codes and not contents:
# # # # 0.5s per image # # # # 0.5s per image
@ -84,6 +86,7 @@ class QrCodeReader(AbstractModule):
decoded_text = qreader.detect_and_decode(image=image) decoded_text = qreader.detect_and_decode(image=image)
for d in decoded_text: for d in decoded_text:
contents.append(d) contents.append(d)
qr_codes = True
except ValueError as e: except ValueError as e:
self.logger.error(f'{e}: {self.obj.get_global_id()}') self.logger.error(f'{e}: {self.obj.get_global_id()}')
return qr_codes, contents return qr_codes, contents
@ -122,7 +125,7 @@ class QrCodeReader(AbstractModule):
# TODO only if new ??? # TODO only if new ???
self.add_message_to_queue(obj=qr_code, queue='Item') self.add_message_to_queue(obj=qr_code, queue='Item')
if is_qrcode: if is_qrcode or contents:
tag = 'infoleak:automatic-detection="qrcode"' tag = 'infoleak:automatic-detection="qrcode"'
self.add_message_to_queue(obj=self.obj, message=tag, queue='Tags') self.add_message_to_queue(obj=self.obj, message=tag, queue='Tags')