47 lines
2.3 KiB
Python
47 lines
2.3 KiB
Python
|
|
import os
|
|
from lib.renderer import Renderer
|
|
from PIL import Image, ImageDraw, ImageFont, ImageFile
|
|
|
|
class PNG_Renderer(Renderer):
|
|
|
|
def __init__(self, config):
|
|
self._config = config
|
|
ImageFile.LOAD_TRUNCATED_IMAGES = True
|
|
|
|
def render(self, cover, artist, album, title):
|
|
title_font = ImageFont.truetype(font=self._config['TitleFont'], size=int(self._config['TitleFontSize']))
|
|
title_y = 0
|
|
title_height = title_font.getbbox('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz')[3]
|
|
text_width = title_font.getbbox(title)[2]
|
|
|
|
artist_font = ImageFont.truetype(font=self._config['ArtistFont'], size=int(self._config['ArtistFontSize']))
|
|
artist_y = title_y + title_height + 5
|
|
artist_height = artist_font.getbbox('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz')[3]
|
|
if artist_font.getbbox(artist)[2] > text_width: text_width = artist_font.getbbox(artist)[2]
|
|
|
|
album_font = ImageFont.truetype(font=self._config['AlbumFont'], size=int(self._config['AlbumFontSize']))
|
|
album_y = artist_y + artist_height + 2
|
|
album_height = album_font.getbbox('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz')[3]
|
|
if album_font.getbbox(album)[2] > text_width: text_width = album_font.getbbox(album)[2]
|
|
|
|
image_height = album_y + album_height + 10
|
|
image_width = text_width + image_height + 40
|
|
image = Image.new('RGBA', (image_width, image_height), color='#' + self._config['BackgroundColor'])
|
|
|
|
draw = ImageDraw.Draw(image)
|
|
text_x = 0
|
|
|
|
if self._config['Cover'] == '1':
|
|
cover = Image.open(cover)
|
|
cover.thumbnail((image_height, image_height), Image.ANTIALIAS)
|
|
image.paste(cover)
|
|
text_x = image_height + 20
|
|
|
|
draw.text((text_x,title_y), title, font=title_font, fill='#' + self._config['TitleFontColor'])
|
|
draw.text((text_x,artist_y), artist, font=artist_font, fill='#' + self._config['TitleFontColor'])
|
|
draw.text((text_x,album_y), album, font=album_font, fill='#' + self._config['TitleFontColor'])
|
|
|
|
# Write file and then move it in position
|
|
image.save(self._config['OutputPath'] + '.part', 'PNG')
|
|
os.replace(self._config['OutputPath'] + '.part', self._config['OutputPath']) |