kwikdown.py

#!/usr/bin/env python


import requests,os,tqdm,time
from selenium import webdriver
from selenium.webdriver.firefox.service import Service as chrome_service
from selenium.webdriver.chrome.service import Service as ff_service
from bs4 import BeautifulSoup

    


def kwik_download(url:str,browser: str = "firefox",dpath:str = os.getcwd(),chunk_size: int = 10 * 1024,ep=None,animename = None):
    # changing to specified path
    os.chdir(dpath)

    #Generating post url from url 
    posturl = url.replace("/f/","/d/")
    
    # browser handling
    chrome_guess = ["chrome","Chrome","google chrome","google"]
    ff_guess = ["ff","firefox","ffgui","ffox","fire"]
    
    if browser.lower() in chrome_guess:
        chserv = chrome_service("/snap/bin/geckodriver")
        
        options = webdriver.ChromeOptions()
        options.headless = True
        
        driver = webdriver.Chrome(service = chserv,options=options)
        
    elif browser.lower() in ff_guess:
        ffserv = ff_service("/snap/bin/geckodriver")
        
        options = webdriver.FirefoxOptions()
        options.add_argument("-headless")
        
        driver = webdriver.Firefox(service=ffserv,options=options)
    else:
        print(f"Sorry your browser is not supported :( ,\nfeel free to report the issue at https://github.com/haxsysgit/autopahe/issues")
        return 0
    
    driver.get(url)
    
    # Extract the page source
    page_source = driver.page_source
    
    # Create a BeautifulSoup object from the page source
    soup = BeautifulSoup(page_source, "lxml")
    
    # Find the form element
    form = soup.find("form")
    
    # Find the hidden input element within the form and Extract the value attribute of the hidden input
    token = form.find("input", attrs={"type": "hidden"})['value']
    # print(f"\n{token}")
    # Navigate to the desired page
    driver.get(posturl)

    # Get the cookies
    cookies = driver.get_cookies()
    # print(f"\n\n{cookies}")
    # Combine cookies into a single string
    cookie_string = ';'.join([cookie['name'] + '=' + cookie['value'] for cookie in cookies])
    # print(f"\n\n{cookie_string}")

    # Quit the driver
    driver.quit()
    
    # request handlin
    params = {"_token":token}
    preheaders = {
        'Host': 'kwik.si',
        'User-Agent': 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:129.0) Gecko/20100101 Firefox/129.0',
        'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/png,image/svg+xml,*/*;q=0.8',
        'Accept-Language': 'en-US,en;q=0.5',
        'Accept-Encoding': 'gzip, deflate, br, zstd',
        'Referer': url,
        'Content-Type': 'application/x-www-form-urlencoded',
        'Content-Length': '47',
        'Origin': 'https://kwik.si',
        'Connection': 'keep-alive',
        'Cookie': cookie_string,
        'Upgrade-Insecure-Requests': '1',
        'Sec-Fetch-Dest': 'document',
        'Sec-Fetch-Mode': 'navigate',
        'Sec-Fetch-Site': 'same-origin',
        'Sec-Fetch-User': '?1',
        'Priority': 'u=0, i',
        'TE': 'trailers'
    }

    

    response = requests.post(posturl,data=params,headers=preheaders,stream=True)

    total_size = int(response.headers.get('content-length', 0))
    filename = None

    # Extract the filename from the content disposition header
    content_disposition = response.headers.get("content-disposition")
    if content_disposition:
        filename_start = content_disposition.find("filename=")
        if filename_start != -1:
            filename = content_disposition[filename_start + 9:]


    # If the filename is not extracted, use a default name
    # changing to path is specified, if not the current path
    if not filename:
        filename = "video.mp4"  # Use a default filename if not extracted from headers

    # Check if the file already exists
    if os.path.exists(filename):
        file_size = os.path.getsize(filename)
        kwikhead = {**preheaders,"Range":f"bytes={file_size}-"}
    

    
    response = requests.post(posturl,data=params,headers=preheaders,stream=True)

    total_size = int(response.headers.get('content-length', 0))

    
    if response.status_code == 200:
        # Save the content to a file

        with open(filename, "wb") as file,tqdm.tqdm(
            desc=f'Downloading Episode {ep}',
            total=total_size,
            unit='B',
            unit_scale=True,
            unit_divisor=1024,
            ncols=80
        ) as progress_bar:
            for chunk in response.iter_content(chunk_size=chunk_size):
                file.write(chunk)
                progress_bar.update(len(chunk))

    # For Downloading partial content i.e 206 is partial content

    elif response.status_code == 206:
        # Save the content to a file
        with open(filename, "ab") as file,tqdm.tqdm(
            desc=f'Downloading ep {ep}',
            total=total_size,
            unit='B',
            unit_scale=True,
            unit_divisor=1024,
            ncols=80
        ) as progress_bar:
            for chunk in response.iter_content(chunk_size=chunk_size):
                file.write(chunk)
                progress_bar.update(len(chunk))

    else:

        print("Failed to download the MP4 file.\n")
        print(f"return code : {response.status_code}")