update

Author: arandomdev

generator.py

@@ -3,8 +3,8 @@
 This Script is used to generate a custom code that can be passed to
 the printer script.
 
-Each code consist of 4 segments, 3 four characters and 1 six character.
-for example "1100 1101 0100 001101"
+Each code can be summarized in 4 segments, 3 four character segments and 1 six
+character segment. For example "1100 1101 0100 001101"
 """
 
 import argparse
@@ -24,14 +24,52 @@
 """
 
 
-# 4 segments in order, seperated by a space
-SEGMENTS = "1100 1101 0100 001101"
+def _getCode() -> tuple[str, str, str, str]:
+	"""Get the code from the user
+	"""
 
+	parser = argparse.ArgumentParser(description="Generate a code to pass to the printer.")  # noqa
+	parser.add_argument(
+		"code",
+		nargs="+",
+		help="The code to generate with. Formated like this, '1100 1101 0100 001101'"
+	)
 
-def main():
-	code = ""
+	segments = parser.parse_args().code
+
+	# verify the segments
+	if len(segments) != 4:
+		print("Did not get 4 segments!")
+		parser.exit()
+		pass
+
+	for i, segment in enumerate(segments):
+		# check if each segment only contains 1s and 0s
+		if any(c not in "10" for c in segment):
+			print(f"Segment {i+1} has invalid characters! It can only contain 1s and 0s.")  # noqa
+			parser.exit()
+			pass
+
+		# check if the length is correct
+		if i == 3:
+			if len(segment) != 6:
+				print("Segment 4 does not have 6 characters!")
+				parser.exit()
+				pass
+			pass
+		else:
+			if len(segment) != 4:
+				print(f"Segment {i+1} does not have 4 characters!")
+				parser.exit()
+				pass
+			pass
+		pass
+
+	return segments
 
-	seg1, seg2, seg3, seg4 = SEGMENTS.split(" ")
+
+def main():
+	seg1, seg2, seg3, seg4 = _getCode()
 
 	formatter = "01{seg1}{sep1}{seg2}{sep2}{seg3}{sep3}{seg4}{sep4}"
 	formatter = partial(
@@ -42,38 +80,39 @@ def main():
 		seg4=seg4
 	)
 
-	code += formatter(
+	formattedCode = ""
+	formattedCode += formatter(
 		sep1="10",
 		sep2="10",
 		sep3="10",
 		sep4="01",
 	)
-	code += formatter(
+	formattedCode += formatter(
 		sep1="10",
 		sep2="01",
 		sep3="01",
 		sep4="10",
 	)
-	code += formatter(
+	formattedCode += formatter(
 		sep1="01",
 		sep2="10",
 		sep3="01",
 		sep4="01",
 	)
-	code += formatter(
+	formattedCode += formatter(
 		sep1="01",
 		sep2="01",
 		sep3="01",
 		sep4="01",
 	)
-	code += formatter(
+	formattedCode += formatter(
 		sep1="01",
 		sep2="01",
 		sep3="10",
 		sep4="10",
 	)
 
-	print(code)
+	print(formattedCode)
 	pass
 
 

printer.py

@@ -1,4 +1,8 @@
+import argparse
+import os
+import pathlib
 from dataclasses import dataclass
+
 from PIL import Image, ImageDraw
 
 TEST_CODE = "01110010110110010010001101010111001011010101000100110110011100011101100100010011010101110001110101010001001101010111000111010101001000110110"  # noqa
@@ -12,16 +16,64 @@
 CODON_WIDTH_TOLERANCE = 0.5
 
 
+class _PrinterArgs(argparse.Namespace):
+	code: str = ""
+	output: pathlib.Path = None
+	pass
+
+
 @dataclass
 class _Codon(object):
 	color: str
 	startAngle: int
 	endAngle: int
+	pass
+
+
+def _getArgs() -> _PrinterArgs:
+	parser = argparse.ArgumentParser(description="Create a printable barcode.")
+	parser.add_argument(
+		"code",
+		help="A 140 character code to make a barcode with."
+	)
+
+	parser.add_argument(
+		"-o", "--output",
+		type=pathlib.Path,
+		help="The path of the output image. By default it outputs to 'output/customBarcode.png'"  # noqa
+	)
+
+	args = parser.parse_args(namespace=_PrinterArgs())
+
+	# verify the input code
+	if len(args.code) != 140:
+		print("The input code is not 140 character!")
+		parser.exit()
+		pass
+
+	if any(c not in "10" for c in args.code):
+		print("The input code has invalid characters! It can only have 1s and 0s.")
+		parser.exit()
+		pass
+
+	return args
 
 
 def main():
-	inputData = TEST_CODE
+	args = _getArgs()
+
+	# Create the output directory if necessary
+	outputPath = args.output
+	if outputPath is None:
+		outputPath = pathlib.Path("output/customBarcode.png")
+		os.makedirs(outputPath.parent, exist_ok=True)
+		pass
+	else:
+		os.makedirs(outputPath.parent, exist_ok=True)
+		pass
 
+	# generate a list of codons
+	inputData = args.code
 	angleSize = 360 / len(inputData)
 
 	code: list[_Codon] = []
@@ -54,6 +106,7 @@ def main():
 		i += codonWidth
 		pass
 
+	# create the image
 	with Image.new("1", (IMAGE_SIZE, IMAGE_SIZE), color=1) as im:
 		draw = ImageDraw.Draw(im)
 
@@ -70,7 +123,7 @@ def main():
 			)
 			pass
 
-		with open("output\\testImage.png", mode="wb") as outFile:
+		with open(outputPath, mode="wb") as outFile:
 			im.save(outFile, dpi=(DPI, DPI))
 			pass
 		pass