from PIL import Image, ImageDraw, ImageFont
-def drawbox(img, d, context, get_font_path):
- for version in d['versions']:
- if draw_version(img, version, context, get_font_path):
- break
-
-
def split_to_lines(text, draw, font, max_width):
words = text.split()
current = ''
yield current
-def draw_version(img, d, context, get_font_path):
- # todo: do this in a subimg
+def draw_box(img, d, context, get_font_path, scale):
newimg = Image.new(
'RGBA',
(
cursor = 0
for item in d['items']:
if item.get('vskip'):
- cursor += item['vskip']
+ cursor += int(round(item['vskip'] * scale))
text = item['text'].format(**context)
if not text:
continue
if item.get('uppercase'):
text = text.upper()
- font = ImageFont.truetype(get_font_path(item['font-family']), item['font-size'])
+ font = ImageFont.truetype(
+ get_font_path(item['font-family']),
+ int(round(item['font-size'] * scale)),
+ layout_engine=ImageFont.LAYOUT_BASIC
+ )
max_width = item.get('max-width', newimg.size[0])
for line in split_to_lines(text, draw, font, max_width):
if cursor + realheight > newimg.size[1]:
return False
draw.text((0, cursor), line, font=font, fill=item.get('color'))
- cursor += item['line-height']
+ cursor += int(round(item['line-height'] * scale))
img.paste(newimg, (d.get('x', 0), d.get('y', 0)), newimg)
return True
+def draw_box_with_scaling(img, d, context, get_font_path):
+ scale = 1.0
+ while scale > 0:
+ if draw_box(img, d, context, get_font_path, scale):
+ return True
+ scale -= 0.05
+
+
+
def create_thumbnail(background_path, defn, context, get_font_path):
img = Image.open(background_path)
d = yaml.load(defn)
for boxdef in d['boxes']:
- drawbox(img, boxdef, context, get_font_path)
+ if not draw_box_with_scaling(img, boxdef, context, get_font_path):
+ raise ValueError()
return img