QR-Code-generator/python/qrcodegen-worker.py

# 
# QR Code generator test worker (Python)
# 
# This program reads data and encoding parameters from standard input and writes
# QR Code bitmaps to standard output. The I/O format is one integer per line.
# Run with no command line arguments. The program is intended for automated
# batch testing of end-to-end functionality of this QR Code generator library.
# 
# Copyright (c) Project Nayuki. (MIT License)
# https://www.nayuki.io/page/qr-code-generator-library
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy of
# this software and associated documentation files (the "Software"), to deal in
# the Software without restriction, including without limitation the rights to
# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
# the Software, and to permit persons to whom the Software is furnished to do so,
# subject to the following conditions:
# - The above copyright notice and this permission notice shall be included in
#   all copies or substantial portions of the Software.
# - The Software is provided "as is", without warranty of any kind, express or
#   implied, including but not limited to the warranties of merchantability,
#   fitness for a particular purpose and noninfringement. In no event shall the
#   authors or copyright holders be liable for any claim, damages or other
#   liability, whether in an action of contract, tort or otherwise, arising from,
#   out of or in connection with the Software or the use or other dealings in the
#   Software.
# 

import sys
from typing import List, Sequence
import qrcodegen


def read_int() -> int:
	return int(input())


def main() -> None:
	while True:
		
		# Read data or exit
		length: int = read_int()
		if length == -1:
			break
		data = bytearray(read_int() for _ in range(length))
		
		# Read encoding parameters
		errcorlvl : int = read_int()
		minversion: int = read_int()
		maxversion: int = read_int()
		mask      : int = read_int()
		boostecl  : int = read_int()
		
		# Make segments for encoding
		if all((b < 128) for b in data):  # Is ASCII
			segs: List[qrcodegen.QrSegment] = qrcodegen.QrSegment.make_segments(data.decode("ASCII"))
		else:
			segs = [qrcodegen.QrSegment.make_bytes(data)]
		
		try:  # Try to make QR Code symbol
			qr: qrcodegen.QrCode = qrcodegen.QrCode.encode_segments(segs, ECC_LEVELS[errcorlvl], minversion, maxversion, mask, boostecl != 0)
			# Print grid of modules
			print(qr.get_version())
			for y in range(qr.get_size()):
				for x in range(qr.get_size()):
					print(1 if qr.get_module(x, y) else 0)
			
		except qrcodegen.DataTooLongError:
			print(-1)
		sys.stdout.flush()


ECC_LEVELS: Sequence[qrcodegen.QrCode.Ecc] = (
	qrcodegen.QrCode.Ecc.LOW,
	qrcodegen.QrCode.Ecc.MEDIUM,
	qrcodegen.QrCode.Ecc.QUARTILE,
	qrcodegen.QrCode.Ecc.HIGH,
)


if __name__ == "__main__":
	main()