"""
Written by: Shreyas Daniel - github.com/shreydan
Written on: 26 April 2017

Description: Download latest XKCD Comic with this program.

NOTE:
	if this script is launched from the cloned repo, a new folder is created.
	Please move the file to another directory to avoid messing with the folder structure.
"""

import os
import urllib.request

import requests
from lxml import html


def main():
    # opens xkcd.com
    try:
        page = requests.get("https://www.xkcd.com")
    except requests.exceptions.RequestException as e:
        print(e)
        exit()

    # parses xkcd.com page
    tree = html.fromstring(page.content)

    # finds image src url
    image_src = tree.xpath(".//*[@id='comic']/img/@src")[0]
    image_src = "https:" + str(image_src)

    # gets comic name from the image src url
    comic_name = image_src.split('/')[-1]

    # save location of comic
    comic_location = os.getcwd() + '/comics/'

    # checks if save location exists else creates
    if not os.path.exists(comic_location):
        os.makedirs(comic_location)

    # creates final comic location including name of the comic
    comic_location = comic_location + comic_name

    # downloads the comic
    urllib.request.urlretrieve(image_src, comic_location)


if __name__ == "__main__":
    main()