mirror of
https://github.com/nadimkobeissi/mkbsd.git
synced 2024-12-22 02:35:34 +00:00
Update mkbsd.py to use local JSOM
This commit is contained in:
parent
b0ce68033d
commit
e96161c7c2
61
mkbsd.py
61
mkbsd.py
|
@ -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())
|
Loading…
Reference in a new issue