Update 'print_single_mice.py'

print_single_mice.py

@@ -4,22 +4,25 @@ import shimatta_label.label_image as li
 import shimatta_label.brother_ql_wrapper as ql_wrapper
 import sys
 
-printer_model = 'QL-800'
-printer_connection = 'usb://0x04f9:0x209b'
+def print_label():
+	printer_model = 'QL-800'
+	printer_connection = 'usb://0x04f9:0x209b'
 
-if len(sys.argv) < 4:
-	print(f'Usage: {sys.argv[0]} <Heading> <Line2> <Line3>')
-	heading = input('First (Bold Heading) Line:')
-	line2 = input('Second Line:')
-	line3 = input('Third line:')
-else:
-	heading = sys.argv[1]
-	line2 = sys.argv[2]
-	line3 = sys.argv[3]
+	if len(sys.argv) < 4:
+		print(f'Usage: {sys.argv[0]} <Heading> <Line2> <Line3>')
+		heading = input('First (Bold Heading) Line:')
+		line2 = input('Second Line:')
+		line3 = input('Third line:')
+	else:
+		heading = sys.argv[1]
+		line2 = sys.argv[2]
+		line3 = sys.argv[3]
 
-label = li.MiceToiletLabel()
-label.put_text(heading, line2, line3)
+	label = li.MiceToiletLabel()
+	label.put_text(heading, line2, line3)
 
-printer = ql_wrapper.BrotherQlPrinter(model=printer_model, printer_connection=printer_connection, label_format='12')
-printer.print_image(label.get_pillow_image(), cut=True, rotation=270)
+	printer = ql_wrapper.BrotherQlPrinter(model=printer_model, printer_connection=printer_connection, label_format='12')
+	printer.print_image(label.get_pillow_image(), cut=True, rotation=270)
 
+if __name__ == '__main__':
+	print_label()