correct use of asyncio

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

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:
@ -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)}")