Unreal Common Input Keyboard Action Icon Generator

generate-keyboard-icons.py

import argparse
import csv
import pyperclip
from pathlib import Path
from PIL import Image, ImageDraw, ImageFont

supersample = 3
fontsize = 80
padding = 64

def main():
	parser = argparse.ArgumentParser(description='Generate keyboard button icons from an input csv file')
	parser.add_argument('FILEPATH', type=CSVPath, help='the input csv file')
	parser.add_argument('-s', '--size', type=int, default=128)
	args = vars(parser.parse_args())

	size = (args['size'] * supersample, args['size'] * supersample)
	font = ImageFont.truetype(font="Fonts/OpenSans-Bold.ttf", size=fontsize*supersample)

	pasters = []
	pasters.append('(Key=LeftMouseButton,KeyBrush=(ImageSize=(X=256.000000,Y=256.000000),ResourceObject="/Script/Engine.Texture2D\'/Game/UI/Platform/KeyboardMouse/Icon-Mouse-Button-L.Icon-Mouse-Button-L\'"))')
	pasters.append('(Key=RightMouseButton,KeyBrush=(ImageSize=(X=256.000000,Y=256.000000),ResourceObject="/Script/Engine.Texture2D\'/Game/UI/Platform/KeyboardMouse/Icon-Mouse-Button-R.Icon-Mouse-Button-R\'"))')

	with open(args['FILEPATH'], newline='', encoding='utf-8') as csvfile:
		csvreader = csv.DictReader(csvfile)
		for row in csvreader:
			label = row['Label']
			text = label if label else row['Key']
			xoff = int(row['XOffset']) if row['XOffset'] else 0
			yoff = int(row['YOffset']) if row['YOffset'] else 0
			textbox = font.getbbox(text)
			textsize = (textbox[2] - textbox[0], textbox[3] - textbox[1])
			size = (max(args['size']*supersample, textsize[0] + (padding*supersample)), size[1])
			textcenter = (((size[0] - textsize[0]) // 2) + (xoff*supersample), 20 + yoff*supersample)
			
			im = Image.new('RGBA', size)
			draw = ImageDraw.Draw(im)
			draw.rounded_rectangle([(0, 0), size], radius=16*supersample, width=9*supersample)
			draw.text(textcenter, text, fill="white", font=font)
			im = im.resize((size[0] // supersample, size[1] // supersample), resample=Image.Resampling.LANCZOS)

			filename = f'Icon-Keyboard-{row["Key"]}'
			im.save(f"output/{filename}.png")
			pasters.append(f'(Key={row["Key"]},KeyBrush=(ImageSize=(X={im.size[0]},Y={im.size[1]}),ResourceObject="/Script/Engine.Texture2D\'/Game/UI/Platform/KeyboardMouse/{filename}.{filename}\'"))')

	joinedpasters = ','.join(pasters)
	pyperclip.copy(f'({joinedpasters})')

def CSVPath(pathstr):
	path = Path(pathstr)
	if path.is_file():
		if path.exists():
			ext = path.suffix.lower()
			if ext == '.csv':
				return pathstr
			else: 
				raise argparse.ArgumentTypeError(f"\"{pathstr}\" must be a .csv file")
		else:
			raise argparse.ArgumentTypeError(f"\"{pathstr}\" does not exist")
	else:
		raise argparse.ArgumentTypeError(f"\"{pathstr}\" is not a valid file path")

if __name__ == "__main__":
	main();

KeyboardKeyIcons.csv

requirements.txt

pillow==10.2.0
pyperclip==1.8.2