""" iCanHazDadJoke module for InkyCal Project Special thanks to Erik Fredericks (@efredericks) for the template! Copyright by aceinnolab """ from inkycal.modules.template import inkycal_module from inkycal.custom import * import requests # Show less logging for request module logging.getLogger("urllib3").setLevel(logging.WARNING) logger = logging.getLogger(__name__) 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 logger.debug(f'{__name__} loaded') 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.debug(f'image size: {im_width} x {im_height} px') # 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(): logger.debug('Connection test passed') else: logger.error("Network not reachable. Please check your connection.") raise NetworkNotReachableError # Set some parameters for formatting feeds line_spacing = 5 text_bbox = self.font.getbbox("hg") line_height = text_bbox[3] + line_spacing line_width = im_width max_lines = (im_height // (line_height + 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.rstrip() # use to remove newlines 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") # Write the joke on the image for _ in range(len(wrapped)): if _ + 1 > max_lines: logger.error('Ran out of lines for this joke :/') break write(im_black, line_positions[_], (line_width, line_height), wrapped[_], font=self.font, alignment='left') # Return images for black and colour channels return im_black, im_colour