tcsenpai/qrare
1
0
Fork 0
mirror of https://github.com/tcsenpai/qrare.git synced 2025-06-06 03:05:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

added fallback method

Browse Source
This commit is contained in:
tcsenpai 2024-07-22 23:20:00 +02:00
parent d4b2713702
commit 5a59d4b60d
2 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example_with_files.py
View File

@ -3,7 +3,7 @@ import qraro
with open("pika.webp", "rb") as f: with open("pika.webp", "rb") as f:
data = f.read() data = f.read()
qraro.bin_to_qr(data, chunk_size=100, box_size=10, border=4) qraro.bin_to_qr(data, chunk_size=1000, box_size=10, border=4)
test_file = qraro.qr_to_bin() test_file = qraro.qr_to_bin()

26
qraro.py
View File

@ -1,6 +1,7 @@
import qrcode import qrcode
from PIL import Image from PIL import Image
import zxing import zxing
import qrtools
def bin_to_qr(data, chunk_size=100, filename_prefix="qr_code", box_size=10, border=4): def bin_to_qr(data, chunk_size=100, filename_prefix="qr_code", box_size=10, border=4):
hex_data = data.hex() hex_data = data.hex()
@ -24,6 +25,7 @@ def qr_to_bin(filename_prefix="qr_code"):
chunks = {} chunks = {}
i = 1 i = 1
reader = zxing.BarCodeReader() reader = zxing.BarCodeReader()
fallback_reader = qrtools.QR()
while True: while True:
try: try:
@ -33,15 +35,24 @@ def qr_to_bin(filename_prefix="qr_code"):
print(f"decoded...", end="") print(f"decoded...", end="")
if barcode and barcode.parsed: if barcode and barcode.parsed:
decoded = barcode.parsed decoded = barcode.parsed
chunk_info, chunk_data = decoded.split(':', 1)
chunk_num, total_chunks = map(int, chunk_info.split('/'))
chunks[chunk_num] = chunk_data
print(f"binary data extracted [OK]")
if chunk_num == total_chunks:
break
else: else:
print(f"Could not decode QR code {i}: {barcode.raw} with zxing, trying fallback reader...")
fallback_reader.decode(filename)
decoded = fallback_reader.data
# split the decoded string into chunk_info and chunk_data
chunk_info, chunk_data = decoded.split(':', 1)
chunk_num, total_chunks = map(int, chunk_info.split('/'))
chunks[chunk_num] = chunk_data
# print(chunk_data)
print(f"binary data extracted [OK]")
if chunk_num == total_chunks:
break
# print(fallback_reader.data)
if not chunks[i]:
print(f"Error decoding QR code {i}: {barcode.raw}") print(f"Error decoding QR code {i}: {barcode.raw}")
exit(-1)
print(f"binary data extracted [OK]")
i += 1 i += 1
except FileNotFoundError: except FileNotFoundError:
break break
@ -55,3 +66,4 @@ def qr_to_bin(filename_prefix="qr_code"):
hex_data = ''.join(chunks[i] for i in range(1, len(chunks) + 1)) hex_data = ''.join(chunks[i] for i in range(1, len(chunks) + 1))
return bytes.fromhex(hex_data) return bytes.fromhex(hex_data)