#!/usr/bin/python

# Author: xanz
# Copyright 2019

import re
import random
import requests
from bs4 import BeautifulSoup as scrap

red     = '\033[31m'
green   = '\033[32m'
yellow  = '\033[33m'
stop    = '\033[37m'
user_agent_list = [
    'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.113 Safari/537.36',
    'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.90 Safari/537.36',
    'Mozilla/5.0 (Windows NT 5.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.90 Safari/537.36',
    'Mozilla/5.0 (Windows NT 6.2; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.90 Safari/537.36',
    'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/44.0.2403.157 Safari/537.36',
    'Mozilla/5.0 (Windows NT 6.3; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.113 Safari/537.36',
    'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/57.0.2987.133 Safari/537.36',
    'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/57.0.2987.133 Safari/537.36',
    'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/55.0.2883.87 Safari/537.36',
    'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/55.0.2883.87 Safari/537.36',
    'Mozilla/4.0 (compatible; MSIE 9.0; Windows NT 6.1)',
    'Mozilla/5.0 (Windows NT 6.1; WOW64; Trident/7.0; rv:11.0) like Gecko',
    'Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; WOW64; Trident/5.0)',
    'Mozilla/5.0 (Windows NT 6.1; Trident/7.0; rv:11.0) like Gecko',
    'Mozilla/5.0 (Windows NT 6.2; WOW64; Trident/7.0; rv:11.0) like Gecko',
    'Mozilla/5.0 (Windows NT 10.0; WOW64; Trident/7.0; rv:11.0) like Gecko',
    'Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.0; Trident/5.0)',
    'Mozilla/5.0 (Windows NT 6.3; WOW64; Trident/7.0; rv:11.0) like Gecko',
    'Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0)',
    'Mozilla/5.0 (Windows NT 6.1; Win64; x64; Trident/7.0; rv:11.0) like Gecko',
    'Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.1; WOW64; Trident/6.0)',
    'Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.1; Trident/6.0)',
    'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0; .NET CLR 2.0.50727; .NET CLR 3.0.4506.2152; .NET CLR 3.5.30729)'
]
ua      = random.choice(user_agent_list)
ses     = requests.Session()
h       = {'User-Agent':ua}
u       = "https://anitoki.com"
r       = ses.get(u,headers=h,timeout=6)
s       = scrap(r.content,"html.parser")
content = s.find_all("div",class_="content")
link    = []
teks    = []

def banner():
    print("""
    ##############################################
       DIH WIBU:V...          CANDA GOBLOG! :v
    ##############################################
    """)

def main():
    banner()
    print("RECENT UPDATE (site: anitoki.com):\n")
    for judul in content:
        link.append(judul.h2.a['href']) # MEMASUKKAN HASIL SCRAP KE LIST
        teks.append(judul.h2.a.text)
    for index in range(len(link)):
        print(f"[{red}{index}{stop}] {teks[index]}")
    num = int(input("\nchoose: "))
    goTo(num)

def goTo(num):
    r = ses.get(link[num],headers=h,timeout=6)
    s = scrap(r.content,"html.parser")

    for download in s.find_all("div",class_="smokeddl"):
        for formatvideo in download.find_all("div",class_="smokettl"):
            print(f"\n[ {green}{formatvideo.text}{stop} ]")
            print("")
            for url in download.find_all("div","smokeurl"):
                for resolusi in url.find_all("strong"):
                    print(f"\n{yellow}{resolusi.text}{stop}")
                for links in url.find_all("a",attrs={'href':re.compile("https://")}):
                    print(f" [{red}*{stop}] {links['href']}")
            print("")

if __name__ == "__main__":
    main()