em/mkbsd
1
0
Fork 0
mirror of https://github.com/nadimkobeissi/mkbsd.git synced 2024-12-04 16:57:47 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

correct use of asyncio

Browse source
This commit is contained in:
Om Lanke 2024-09-28 01:10:36 +05:30 committed by GitHub
parent 82e50c64f0
commit 97ffaabcc1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
mkbsd.py
View file

@ -21,6 +21,8 @@ async def download_image(session, image_url, file_path):
except Exception as e:
print(f"Error downloading image: {str(e)}")
print(f"🖼️ Saved image to {file_path}")
async def main():
try:
async with aiohttp.ClientSession() as session:
@ -29,7 +31,7 @@ async def main():
raise Exception(f"⛔ Failed to fetch JSON file: {response.status}")
json_data = await response.json()
data = json_data.get('data')
if not data:
raise Exception('⛔ JSON does not have a "data" property at its root.')
@ -39,20 +41,19 @@ async def main():
print(f"📁 Created directory: {download_dir}")
file_index = 1
for key, subproperty in data.items():
if subproperty and subproperty.get('dhd'):
image_url = subproperty['dhd']
print(f"🔍 Found image URL!")
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)
async with asyncio.TaskGroup() as group:
for key, subproperty in data.items():
if subproperty and subproperty.get('dhd'):
image_url = subproperty['dhd']
print(f"🔍 Found image URL!")
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)
print(f"🖼️ Saved image to {file_path}")
group.create_task(download_image(session, image_url, file_path))
file_index += 1
await delay(250)
file_index += 1
except Exception as e:
print(f"Error: {str(e)}")