From 797862870899a7c211d8fea91546385bc9d2edc8 Mon Sep 17 00:00:00 2001
From: jkmdev <jkm96.dev@gmail.com>
Date: Wed, 25 Sep 2024 20:13:14 +0300
Subject: [PATCH] check if file already exists before downloading

---
 .gitignore |  1 +
 mkbsd.py   | 15 ++++++++++++---
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index de2b2c6..9d7d89b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 .DS_Store
 downloads
+/mkbsdvenv/
diff --git a/mkbsd.py b/mkbsd.py
index 2b33310..5e8b5da 100644
--- a/mkbsd.py
+++ b/mkbsd.py
@@ -5,11 +5,14 @@ import time
 import aiohttp
 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):
     await asyncio.sleep(ms / 1000)
 
+
 async def download_image(session, image_url, file_path):
     try:
         async with session.get(image_url) as response:
@@ -21,6 +24,7 @@ async def download_image(session, image_url, file_path):
     except Exception as e:
         print(f"Error downloading image: {str(e)}")
 
+
 async def main():
     try:
         async with aiohttp.ClientSession() as session:
@@ -29,7 +33,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.')
 
@@ -48,8 +52,11 @@ async def main():
                         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}")
+                        if os.path.exists(file_path):
+                            print(f"⚠️ File {file_path} already exists. Skipping download.")
+                        else:
+                            await download_image(session, image_url, file_path)
+                            print(f"🖼️ Saved image to {file_path}")
 
                         file_index += 1
                         await delay(250)
@@ -57,6 +64,7 @@ async def main():
     except Exception as e:
         print(f"Error: {str(e)}")
 
+
 def ascii_art():
     print("""
  /$$      /$$ /$$   /$$ /$$$$$$$   /$$$$$$  /$$$$$$$
@@ -70,6 +78,7 @@ def ascii_art():
     print("")
     print("🤑 Starting downloads from your favorite sellout grifter's wallpaper app...")
 
+
 if __name__ == "__main__":
     ascii_art()
     time.sleep(5)