AFSD

API/app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, Query
-from fastapi.responses import StreamingResponse
+from fastapi.responses import FileResponse
 import aiohttp
 import requests
 import asyncio
@@ -219,37 +219,30 @@ async def pixif(
     image_exifs = {post_id: image_url.replace('https://i.pximg.net/img-original/', '', 1) for post_id, image_url in results if image_url}
     return image_exifs
 
-async def download_image(session, post_id, post_url, semaphore):
-    async with semaphore:
-        url = f"{img_base}{post_url}"
-        async with session.get(url) as response:
-            content = await response.read()
-            return (post_id, content)
+async def download_image(session, post_id, post_url):
+    url = f"{img_base}{post_url}"
+    async with session.get(url) as response:
+        content = await response.read()
+        return (post_id, content)
 
 @app.post("/download")
 async def download(
     items: PixifDownloadModel    
 ):
     posts = items.posts
-    semaphore = asyncio.Semaphore(100)  # Adjust the number as needed
-
+    
     async with aiohttp.ClientSession(cookies=cookies, headers=headers) as session:
-        tasks = [download_image(session, post_id, image_url, semaphore) for post_id, image_url in posts.items()]
+        tasks = [download_image(session, post_id, image_url) for post_id, image_url in posts.items()]
         images = await asyncio.gather(*tasks)
 
-    zip_buffer = io.BytesIO()
-    with ZipFile(zip_buffer, "w") as zipf:
+    timezip = f"{base26_time()}.zip"
+    with ZipFile(timezip, "w", compression=0) as zipf:
         for post_id, image_data in images:
             zipf.writestr(f"{post_id}.png", image_data)
-    zip_buffer.seek(0)
-    filename = f"{base26_time()}.zip"
-    headers = {
-        'Content-Disposition': f'attachment; filename="{filename}"'
-    }
-    return StreamingResponse(
-        zip_buffer, 
+
+    return FileResponse(
+        timezip, 
         media_type="application/zip", 
-        headers=headers
     )
 
 @app.get("/")

Client/Extract Pixiv/user.py

@@ -15,6 +15,11 @@ user_ids = re.findall(r"\d+", inp)
 
 user_ids = [int(uid) for uid in user_ids]
 
+if len(user_ids) == 0:
+    import sys
+    
+    sys.exit()
+    
 response = requests.get(f'{endpoint}/users', params={'user_ids': user_ids})
 response.raise_for_status()
 data = response.json()