99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""
|
||
|
|
iCanHazDadJoke module for InkyCal Project
|
||
|
|
Special thanks to Erik Fredericks (@efredericks) for the template!
|
||
|
|
|
||
|
|
Copyright by aceisace
|
||
|
|
"""
|
||
|
|
from inkycal.modules.template import inkycal_module
|
||
|
|
from inkycal.custom import *
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
filename = os.path.basename(__file__).split('.py')[0]
|
||
|
|
logger = logging.getLogger(filename)
|
||
|
|
logger.setLevel(level=logging.ERROR)
|
||
|
|
|
||
|
|
class Jokes(inkycal_module):
|
||
|
|
"""Icanhazdad-api class
|
||
|
|
parses rss/atom feeds from given urls
|
||
|
|
"""
|
||
|
|
|
||
|
|
name = "iCanHazDad API - grab a random joke from icanhazdad api"
|
||
|
|
|
||
|
|
|
||
|
|
def __init__(self, config):
|
||
|
|
"""Initialize inkycal_feeds module"""
|
||
|
|
|
||
|
|
super().__init__(config)
|
||
|
|
|
||
|
|
config = config['config']
|
||
|
|
|
||
|
|
# give an OK message
|
||
|
|
print('{0} loaded'.format(filename))
|
||
|
|
|
||
|
|
def generate_image(self):
|
||
|
|
"""Generate image for this module"""
|
||
|
|
|
||
|
|
# Define new image size with respect to padding
|
||
|
|
im_width = int(self.width - (2 * self.padding_left))
|
||
|
|
im_height = int(self.height - (2 * self.padding_top))
|
||
|
|
im_size = im_width, im_height
|
||
|
|
logger.info('image size: {} x {} px'.format(im_width, im_height))
|
||
|
|
|
||
|
|
# Create an image for black pixels and one for coloured pixels
|
||
|
|
im_black = Image.new('RGB', size = im_size, color = 'white')
|
||
|
|
im_colour = Image.new('RGB', size = im_size, color = 'white')
|
||
|
|
|
||
|
|
# Check if internet is available
|
||
|
|
if internet_available() == True:
|
||
|
|
logger.info('Connection test passed')
|
||
|
|
else:
|
||
|
|
raise Exception('Network could not be reached :/')
|
||
|
|
|
||
|
|
# Set some parameters for formatting feeds
|
||
|
|
line_spacing = 1
|
||
|
|
line_height = self.font.getsize('hg')[1] + line_spacing
|
||
|
|
line_width = im_width
|
||
|
|
max_lines = (im_height // (self.font.getsize('hg')[1] + line_spacing))
|
||
|
|
|
||
|
|
logger.debug(f"max_lines: {max_lines}")
|
||
|
|
|
||
|
|
# Calculate padding from top so the lines look centralised
|
||
|
|
spacing_top = int( im_height % line_height / 2 )
|
||
|
|
|
||
|
|
# Calculate line_positions
|
||
|
|
line_positions = [
|
||
|
|
(0, spacing_top + _ * line_height ) for _ in range(max_lines)]
|
||
|
|
|
||
|
|
logger.debug(f'line positions: {line_positions}')
|
||
|
|
|
||
|
|
# Get the actual joke
|
||
|
|
url = "https://icanhazdadjoke.com"
|
||
|
|
header = {"accept": "text/plain"}
|
||
|
|
response = requests.get(url, headers=header)
|
||
|
|
response.encoding = 'utf-8' # Change encoding to UTF-8
|
||
|
|
joke = response.text
|
||
|
|
logger.debug(f"joke: {joke}")
|
||
|
|
|
||
|
|
# wrap text in case joke is too large
|
||
|
|
wrapped = text_wrap(joke, font = self.font, max_width = line_width)
|
||
|
|
logger.debug(f"wrapped: {wrapped}")
|
||
|
|
|
||
|
|
# Check if joke can actually fit on the provided space
|
||
|
|
if len(wrapped) > max_lines:
|
||
|
|
logger.error("Ohoh, Joke is too large for given space, please consider "
|
||
|
|
"increasing the size for this module")
|
||
|
|
logger.error("Removing lines in reverse order")
|
||
|
|
wrapped = wrapped[:max_lines]
|
||
|
|
|
||
|
|
# Write feeds on image
|
||
|
|
for _ in range(len(wrapped)):
|
||
|
|
write(im_black, line_positions[_], (line_width, line_height),
|
||
|
|
wrapped[_], font = self.font, alignment= 'left')
|
||
|
|
|
||
|
|
# Save image of black and colour channel in image-folder
|
||
|
|
return im_black, im_colour
|