import requests
from bs4 import BeautifulSoup
import argparse

parser = argparse.ArgumentParser()
parser.add_argument("password")
parser.add_argument("url")
parser.add_argument("--user")

args = parser.parse_args()


headers = {
    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:52.0) Gecko/20100101 Firefox/52.0",
    "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8",
    "Accept-Language": "en-US,en;q=0.5",
    "Accept-Encoding": "gzip, deflate, br",
    "Upgrade-Insecure-Requests": "1",
    "Referer": args.url
}

if args.user:

    payload = {"email": args.user, "password": args.password, "host": "unknown", "url": "unknown",
               "params": "host=unknown&url=unknown"}

    url = args.url.replace("embed", "videos")
    url = url[:url.rfind("/")+1]
    url += "video_login?embed=true"

    r = requests.post(url, headers=headers, data=payload)

else:

    payload = {"password": args.password, "host": "unknown", "url": "unknown", "queryParams": ""}
    r = requests.post(args.url.replace("embed", "video_password"), headers=headers, data=payload)

soup = BeautifulSoup(r.text, "html.parser")
try:
    print(soup.find("a", attrs={"class": "hd-download"}).get("href"))
    # print(soup.find("a", attrs={"class": "sd-download"}).get("href"))
except AttributeError:
    print("Wrong password/username")