zendriverpy.py

import asyncio
import curl
import zendriver as zd
from fake_useragent import UserAgent
import os

async def main():
    ua = UserAgent()
    browser = await zd.start()
    with open("links.txt") as f:
      for x in f:
        page = await browser.get(x.strip())
        await asyncio.sleep(2)
        elements = await page.select_all("a.album__main")
        print(f"Found {len(elements)} categories")
        for element in elements:
            Title = await element.get("title")
            folder_name = "".join(c for c in Title if c.isalnum() or c in " _-").strip()
            folder_name = folder_name.replace(" ", "_")
            os.makedirs(folder_name, exist_ok=True)
            await element.click()
            await asyncio.sleep(2)
            Referer = page.url
            images = await page.select_all("img")
            print(f"Found {len(images)} images in category")
            for idx, img in enumerate(images):
                if img.get("data-path") is None:
                    print(f"Image {idx} has no data-path attribute, skipping.")
                    continue
                src = img.get("data-path")
                if src and src.startswith("/yolo66/"):
                    
                    # download hell starts here
                    print(f"Image {idx}: src={src!r}")

                    url = "https://photo.yupoo.com" + src
                    headers = {
                        "User-Agent": ua.random,
                        "Referer": Referer,
                        "Accept": "image/avif,image/webp,image/apng,image/*,*/*;q=0.8"
                    }

                    response = requests.get(url, headers=headers, stream=True)
                    filename = os.path.join(folder_name, f"image{idx}.jpg")
                    if response.status_code == 200:
                        with open(filename, "wb") as f:
                            for chunk in response.iter_content(1024):
                                f.write(chunk)
                        print("Download completed: {filename}")
                    else:
                        print(f"Failed to download, status code: {response.status_code}")

                    
                else:
                    print(f"Image {idx} has no valid data-path attribute, skipping.")
            



        
    
    



if __name__ == "__main__":
    asyncio.run(main())