File name fix, notes, ask for unit

Made it so output filename is cleaner. Added Unit Type prompt.  Made notes about future work.
This commit is contained in:
Steve Dogiakos 2022-12-06 16:50:53 -07:00
parent 74233392d5
commit cdea4b652c
1 changed files with 4 additions and 1 deletions

View File

@ -3,12 +3,15 @@ import os
from PIL import Image
# Define the input value
unit_type = input("Enter the unit type:Troop, Pack, or Crew")
unit_number = input("Enter your unit number")
# Create an empty list to store the images
images = []
# Loop through each digit in the input value
# TO-DO: Move the pack images into folder Pack, and get similar images for Troop/Crew and put them in their respective folders.
# Then, have the program open folder (unit_type) and find (unit_number.jpg)
for digit in str(unit_number):
# Load the image from the file
image = Image.open(f"{digit}.jpg")
@ -25,4 +28,4 @@ for img in images:
x_offset += img.width
# Save the final result
result_image.save("str(unit_number).jpg")
result_image.save(f"{unit_type} {unit_number}.jpg")