xkcd.py

#!/usr/bin/env python2

"""sudo cp xkcd.py /usr/local/bin/xkcd"""

__author__ = '1337'

from os.path import basename, join
from subprocess import call
import re
import sys

try:
    from urllib.request import urlopen
except ImportError:
    from urllib import urlopen


IMAGE_PATH = '/tmp'


def get_image(url):
    filename = basename(url)
    filepath = join(IMAGE_PATH, filename)
    with file(filepath, 'wb') as f:
        f.write(urlopen(url).read())
        f.close()

    return filepath


def get_permalink(url):
    page_contents = urlopen(url).read()
    return re.findall(r'^\s*[a-zA-Z \(\)/]+:\s*([a-zA-Z0-9:/\.-_]+)\s*$',
                      page_contents, flags=re.U | re.M | re.I)[0]


if __name__ == '__main__':
    if len(sys.argv) > 1:
        url = "http://xkcd.com/{}".format(sys.argv[1])
    else:
        url = "http://xkcd.com"

    filepath = get_image(get_permalink(url))
    call(["display", filepath])