#!/usr/bin/env python

"""This script fetches comic(s) from xkcd.com (using xkcd python module) and outputs them to an Epson thermal

printer using the python-escpos module"""

import escpos

import escpos.printer

import xkcd

import textwrap

import sys

import getopt

from random import randint

from PIL import Image

usageText = 'Usage: xkcdprint.py [-n <number of comics to print>] [-r] [-i <comicid>]\nDefaults to printing latest comic.'

nComicsToPrint = 1

nComicsPrinted = 0

comicRandom = False

maxPixelWidth = float(512)

nStartComic = xkcd.getLatestComic().number

try:

opts, args = getopt.getopt(sys.argv[1:], "hn:ri:", ["help", "to-print=", "random", "comicid="])

except getopt.GetoptError:

print(usageText)

sys.exit(2)

for opt, arg in opts:

if opt in ("-h", "--help"):

print(usageText)

sys.exit()

elif opt in ("-r", "--random"):

nStartComic = randint(1, xkcd.getLatestComicNum())

elif opt in ("-i", "--comicid"):

if 0 < int(arg) <= xkcd.getLatestComicNum():

nStartComic = int(arg)

else:

print("Error: Comic ID specified is out of valid range. Please specify a comic ID between 1 and " + str(xkcd.getLatestComicNum()))

sys.exit(1)

elif opt in ("-n", "--to-print"):

nComicsToPrint = int(arg)

print("Printing " + str(nComicsToPrint) + " comics.")

if nComicsToPrint > 1:

print("Comic # " + str(nStartComic) + " and " + str(nComicsToPrint - 1) + " previous comics will be printed.")

else:

print("Comic # " + str(nStartComic) + " selected for printing.")

# Initialize Epson thermal printer using usb-id (here using TM-T20II)

p = escpos.printer.Usb(0x04b8, 0x0e15)

while nComicsPrinted < nComicsToPrint:

comic = xkcd.getComic(nStartComic - nComicsPrinted)

comicImg = comic.download()

comicAltText = textwrap.wrap(comic.getAsciiAltText().decode('utf-8'), 40)

img = Image.open(comicImg.encode('utf-8'))

imgWidth, imgHeight = img.size

if imgWidth > imgHeight and (imgWidth / imgHeight) > 1.4: #handle square-ish images

img = img.rotate(270, expand=True)

imgWidth, imgHeight = img.size

scalingFactor = (maxPixelWidth / min(imgWidth, imgHeight))

elif imgWidth > imgHeight and (imgWidth / imgHeight) <= 1.4:

scalingFactor = (maxPixelWidth / max(imgWidth, imgHeight))

print(scalingFactor)

else:

scalingFactor = (maxPixelWidth / min(imgWidth, imgHeight))

resizedImg = img.resize((int(scalingFactor * imgWidth), int(scalingFactor * imgHeight)))

p.text('\n' + comic.getAsciiTitle().decode('utf-8'))

p.text('\n' + comic.link + '\n\n')

p.image(resizedImg)

p.text('\n')

for line in comicAltText:

p.text('\n' + line)

p.cut()

nComicsPrinted = nComicsPrinted + 1