Update mkbsd.py

This commit is contained in:
Shivam M. Salunkhe 2024-09-27 00:59:03 +05:30 committed by GitHub
parent 82e50c64f0
commit 2b13ff3f38
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,12 +1,15 @@
# Licensed under the WTFPL License
import os import os
import time import time
import aiohttp import aiohttp
import asyncio import asyncio
from urllib.parse import urlparse from urllib.parse import urlparse
from threading import Event
url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s' url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s'
# Global stop event to signal when to stop downloading
stop_event = Event()
async def delay(ms): async def delay(ms):
await asyncio.sleep(ms / 1000) await asyncio.sleep(ms / 1000)
@ -41,6 +44,9 @@ async def main():
file_index = 1 file_index = 1
for key, subproperty in data.items(): for key, subproperty in data.items():
if subproperty and subproperty.get('dhd'): if subproperty and subproperty.get('dhd'):
if stop_event.is_set():
print("⛔ Download stopped!")
break
image_url = subproperty['dhd'] image_url = subproperty['dhd']
print(f"🔍 Found image URL!") print(f"🔍 Found image URL!")
parsed_url = urlparse(image_url) parsed_url = urlparse(image_url)
@ -57,6 +63,17 @@ async def main():
except Exception as e: except Exception as e:
print(f"Error: {str(e)}") print(f"Error: {str(e)}")
def start_download():
global stop_event
stop_event.clear() # Clear the stop event before starting
print("🚀 Starting download process!")
asyncio.run(main())
def stop_download():
global stop_event
stop_event.set() # Set the stop event to signal the process to stop
print("🛑 Stopping download process!")
def ascii_art(): def ascii_art():
print(""" print("""
/$$ /$$ /$$ /$$ /$$$$$$$ /$$$$$$ /$$$$$$$ /$$ /$$ /$$ /$$ /$$$$$$$ /$$$$$$ /$$$$$$$
@ -68,9 +85,19 @@ def ascii_art():
| $$ \\/ | $$| $$ \\ $$| $$$$$$$/| $$$$$$/| $$$$$$$/ | $$ \\/ | $$| $$ \\ $$| $$$$$$$/| $$$$$$/| $$$$$$$/
|__/ |__/|__/ \\__/|_______/ \\______/ |_______/""") |__/ |__/|__/ \\__/|_______/ \\______/ |_______/""")
print("") print("")
print("🤑 Starting downloads from your favorite sellout grifter's wallpaper app...") print("🤑 Ready to download from your favorite sellout grifter's wallpaper app...")
if __name__ == "__main__": if __name__ == "__main__":
ascii_art() ascii_art()
time.sleep(5) time.sleep(2)
asyncio.run(main())
user_input = input("Press 's' to start downloading, or 'q' to quit: ").lower()
while user_input != 'q':
if user_input == 's':
start_download()
user_input = input("Press 's' to start downloading again or 'q' to quit. Press 'x' to stop the download: ").lower()
if user_input == 'x':
stop_download()
print("👋 Exiting the program.")