-
Notifications
You must be signed in to change notification settings - Fork 1
/
BoardGenerator.py
43 lines (32 loc) · 1.17 KB
/
BoardGenerator.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import StringIO
import random
from PIL import Image, ImageDraw, ImageFont
def generate_board(cards):
"""
Takes a list of card names like ["OneRedSolidSquiggle", ...]
Returns a filename of the generated image
"""
# numbers = ['One', 'Two', 'Three']
# colors = ['Red', 'Green', 'Purple']
# shadings = ['Solid', 'Shaded', 'Hollow']
# shapes = ['Squiggle', 'Oval', 'Diamond']
assert len(cards) % 3 == 0, 'Must pass a list of cards with length \
that\'s a multiple of 3'
num_rows = len(cards) / 3
canvas = Image.new('RGB', ((160+20)*3, (100+10)*num_rows-10), 'white')
fnt = ImageFont.truetype('./Arial.ttf', 24)
d = ImageDraw.Draw(canvas)
i = 0
for row in range(num_rows):
for col in range(3):
im = Image.open('cards/Card{}.png'.format(
cards[i]
))
canvas.paste(im, ((160+20)*col+20, (100+10)*row))
d.text(((160+20)*col+10,(100+10)*row+5), chr(i+65), font=fnt, fill=(0,0,0,255))
i += 1
output = StringIO.StringIO()
canvas.save(output, format='png')
output_string = output.getvalue()
output.close()
return output_string