Cleanup sprite font script

Author: TeamSpen210

sprite_font.py

@@ -1,43 +1,53 @@
 """Generate entity sprite text images."""
 from pathlib import Path
+import sys
 
 from PIL import Image, ImageChops
-from collections import namedtuple
+import attrs
 
-import sys
 
-try:
-    text = sys.argv[1]
-except IndexError:
-    text = input('Enter text to produce: ')
+@attrs.frozen
+class Char:
+    """A character."""
+    img: Image.Image
+    width: int
+
 
-golden = text.startswith("comp_")
-print('Gold' if golden else 'White', 'text selected')
+def main() -> None:
+    """Generate text."""
+    try:
+        text = sys.argv[1]
+    except IndexError:
+        text = input('Enter text to produce, prefix with GOLD to make it gold: ')
 
-Char = namedtuple('Char', 'img width')
+    golden = text.startswith("comp_")
+    if text.startswith('GOLD'):
+        golden = True
+        text = text.removeprefix('GOLD').lstrip()
+    print('Gold' if golden else 'White', 'text selected')
 
-LETTERS = {}
+    LETTERS: dict[str, Char] = {}
 
-for file in Path('text').glob('*.png'):
-    letter = file.name[0]
-    img = Image.open(file)
-    img.load()
-    if golden:
-        img = ImageChops.multiply(img, Image.new('RGBA', img.size, (224, 174, 0, 255)))
-    LETTERS[letter] = Char(img, img.width-1)
+    for file in Path('text').glob('*.png'):
+        letter = file.name[0]
+        img = Image.open(file)
+        img.load()
+        if golden:
+            img = ImageChops.multiply(img, Image.new('RGBA', img.size, (224, 174, 0, 255)))
+        LETTERS[letter] = Char(img, img.width-1)
 
-chars = list(map(LETTERS.__getitem__, text.lower()))
+    chars = [LETTERS[x] for x in text.lower()]
 
-width = sum(c.width for c in chars) + 1
-height = max(c.img.height for c in chars)
+    width = sum(c.width for c in chars) + 1
+    height = max(c.img.height for c in chars)
 
-img = Image.new('RGBA', (width, height), (0, 0, 0, 0))
+    img = Image.new('RGBA', (width, height), (0, 0, 0, 0))
 
-offset = 0
-for ch in chars:
-    img.alpha_composite(ch.img, (offset, 0))
-    offset += ch.width
+    offset = 0
+    for ch in chars:
+        img.alpha_composite(ch.img, (offset, 0))
+        offset += ch.width
 
-img.save(text + '.png')
+    img.save(f'{text}.png')
 
-print('Done!')
+    print('Done!')