Update mkbsd.py

mkbsd.py

@@ -1,61 +1,54 @@
-# Licensed under the WTFPL License
+import requests
-
 import os
-import time
+import json
-import aiohttp
+from concurrent.futures import ThreadPoolExecutor
-import asyncio
-from urllib.parse import urlparse
-url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s'
 
-async def delay(ms):
+# URL of the JSON data
-    await asyncio.sleep(ms / 1000)
+all_url = "https://storage.googleapis.com/panels-cdn/data/20240730/all.json"
 
-async def download_image(session, image_url, file_path):
+# Function to download the URL
-    try:
+def download_url(url):
-        async with session.get(image_url) as response:
+    file_name = os.path.basename(url)
-            if response.status != 200:
+    file_path = os.path.join("downloads", file_name)
-                raise Exception(f"Failed to download image: {response.status}")
-            content = await response.read()
-            with open(file_path, 'wb') as f:
-                f.write(content)
-    except Exception as e:
-        print(f"Error downloading image: {str(e)}")
 
-async def main():
+    if not os.path.exists(file_path):
-    try:
+        print(f"Downloading {url}")
-        async with aiohttp.ClientSession() as session:
+        response = requests.get(url)
-            async with session.get(url) as response:
+        with open(file_path, 'wb') as file:
-                if response.status != 200:
+            file.write(response.content)
-                    raise Exception(f"⛔ Failed to fetch JSON file: {response.status}")
+    else:
-                json_data = await response.json()
+        print(f"Skipping {url}")
-                data = json_data.get('data')
-                
-                if not data:
-                    raise Exception('⛔ JSON does not have a "data" property at its root.')
 
-                download_dir = os.path.join(os.getcwd(), 'downloads')
+# Recursive function to extract URLs from JSON structure
-                if not os.path.exists(download_dir):
+def extract_urls(element, urls):
-                    os.makedirs(download_dir)
+    if isinstance(element, dict):
-                    print(f"📁 Created directory: {download_dir}")
+        for key, value in element.items():
+            if key == "url":
+                urls.append(value)
+            else:
+                extract_urls(value, urls)
+    elif isinstance(element, list):
+        for item in element:
+            extract_urls(item, urls)
 
-                file_index = 1
+# Main function to process the JSON and download files
-                for key, subproperty in data.items():
+def main():
-                    if subproperty and subproperty.get('dhd'):
+    # Fetch the JSON data
-                        image_url = subproperty['dhd']
+    response = requests.get(all_url)
-                        print(f"🔍 Found image URL!")
+    json_data = response.json()
-                        parsed_url = urlparse(image_url)
-                        ext = os.path.splitext(parsed_url.path)[-1] or '.jpg'
-                        filename = f"{file_index}{ext}"
-                        file_path = os.path.join(download_dir, filename)
 
-                        await download_image(session, image_url, file_path)
+    # Extract URLs
-                        print(f"🖼️ Saved image to {file_path}")
+    urls = []
+    extract_urls(json_data, urls)
+    print(f"Found {len(urls)} URLs")
 
-                        file_index += 1
+    # Ensure 'downloads' directory exists
-                        await delay(250)
+    if not os.path.exists("downloads"):
+        os.makedirs("downloads")
 
-    except Exception as e:
+    # Download files with parallelism (max 10 threads)
-        print(f"Error: {str(e)}")
+    with ThreadPoolExecutor(max_workers=10) as executor:
+        executor.map(download_url, urls)
 
 def ascii_art():
     print("""
@@ -73,4 +66,4 @@ def ascii_art():
 if __name__ == "__main__":
     ascii_art()
     time.sleep(5)
-    asyncio.run(main())
+    main()