import urllib
import os
import mechanize

USERNAME = 'sorryicantgiveyouthiswill@weener.com'
PW = 'cryptocurrency'

def download_movie():
    """Download movie by passing URL"""
    os.chdir('/home/mojo/Videos')
    email = USERNAME
    password = PW
    browser = mechanize.Browser()
    browser.set_handle_robots(False)
    browser.addheaders = [("User-agent","Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.2.13) Gecko/20101206 Ubuntu/10.10 (maverick) Firefox/3.6.13")]
    sign_in = browser.open("http://superchillin.com/login.php")
    browser.select_form(nr = 0)
    browser["email"] = email
    browser["password"] = password
    browser.submit()
    movie_url = raw_input("Enter URL of movie you would like to download: ")
    r = browser.open(movie_url)
    html = r.read()
    download_link = browser.find_link(text="Download")
    download_url = "http://superchillin.com" + download_link.url
    title = raw_input("What do you want to name the file? ")
    print "Downloading..."
    browser.retrieve(download_url, title)
    print "Done!"

def download_series():
    """Download series by passing main URL"""
    os.chdir('/home/mojo/Videos/')
    email = USERNAME
    password = PW
    browser = mechanize.Browser()
    series = raw_input("Which series is it? ")
    if not os.path.exists(series):
        os.makedirs(series)
    browser.set_handle_robots(False)
    browser.addheaders = [("User-agent","Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.2.13) Gecko/20101206 Ubuntu/10.10 (maverick) Firefox/3.6.13")]
    sign_in = browser.open("http://superchillin.com/login.php")
    browser.select_form(nr = 0)
    browser["email"] = email
    browser["password"] = password
    browser.submit()
    series_url = raw_input("Enter the download links version of the series you would like to download: ")
    r = browser.open(series_url)
    html = r.read()
    urls = [link.absolute_url for link in
            browser.links(url_regex=(".mp4"))]
    print "Found", len(urls), "episodes to download"
    for link in urls:
        print "downloading..."
        browser.retrieve(link)
        print browser.retrieve(link)
    print "Done"