Update mkbsd.py to use local JSOM

This commit is contained in:
retardgerman 2024-10-09 12:19:31 +02:00 committed by GitHub
parent b0ce68033d
commit e96161c7c2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2,9 +2,11 @@ import os
import time import time
import aiohttp import aiohttp
import asyncio import asyncio
import json
from urllib.parse import urlparse from urllib.parse import urlparse
url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s' # Load the local JSON file
json_file_path = 'images.json'
async def delay(ms): async def delay(ms):
await asyncio.sleep(ms / 1000) await asyncio.sleep(ms / 1000)
@ -22,38 +24,37 @@ async def download_image(session, image_url, file_path):
async def main(): async def main():
try: try:
# Load JSON data from the local file
with open(json_file_path, 'r') as json_file:
json_data = json.load(json_file)
data = json_data.get('data')
if not data:
raise Exception('⛔ JSON does not have a "data" property at its root.')
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.get(url) as response: for key, subproperty in data.items():
if response.status != 200: if subproperty and subproperty.get('dhd'):
raise Exception(f"⛔ Failed to fetch JSON file: {response.status}") image_url = subproperty['dhd']
json_data = await response.json() print(f"🔍 Found image URL!")
data = json_data.get('data')
# Extract artist name from the URL
if not data: artist_name = image_url.split('a~')[1].split('_')[0]
raise Exception('⛔ JSON does not have a "data" property at its root.') artist_dir = os.path.join(os.getcwd(), 'downloads', artist_name)
for key, subproperty in data.items(): if not os.path.exists(artist_dir):
if subproperty and subproperty.get('dhd'): os.makedirs(artist_dir)
image_url = subproperty['dhd'] print(f"📁 Created directory: {artist_dir}")
print(f"🔍 Found image URL!")
# Extrahiere den Künstlernamen vor dem Unterstrich
parsed_url = urlparse(image_url)
artist_name = image_url.split('a~')[1].split('_')[0]
artist_dir = os.path.join(os.getcwd(), 'downloads', artist_name)
if not os.path.exists(artist_dir): # Extract filename from the URL
os.makedirs(artist_dir) filename = os.path.basename(urlparse(image_url).path) # Name including extension
print(f"📁 Created directory: {artist_dir}") file_path = os.path.join(artist_dir, filename)
# Extrahiere den Dateinamen und die Endung await download_image(session, image_url, file_path)
filename = os.path.basename(parsed_url.path) # Name inklusive Endung print(f"🖼️ Saved image to {file_path}")
file_path = os.path.join(artist_dir, filename)
await delay(250)
await download_image(session, image_url, file_path)
print(f"🖼️ Saved image to {file_path}")
await delay(250)
except Exception as e: except Exception as e:
print(f"Error: {str(e)}") print(f"Error: {str(e)}")
@ -74,4 +75,4 @@ def ascii_art():
if __name__ == "__main__": if __name__ == "__main__":
ascii_art() ascii_art()
time.sleep(5) time.sleep(5)
asyncio.run(main()) asyncio.run(main())