feat(download screen):implement download capabilities

This commit is contained in:
Benex254
2024-08-05 09:46:58 +03:00
parent b1f3953f44
commit 02fc593fe6
11 changed files with 172 additions and 49 deletions

View File

@@ -1,4 +1,10 @@
from threading import Thread
from queue import Queue
import yt_dlp
from ... import downloads_dir
from ..utils import sanitize_filename
from ..show_notification import show_notification
class MyLogger:
@@ -12,27 +18,55 @@ class MyLogger:
print(msg)
def my_hook(d):
if d["status"] == "finished":
print("Done downloading, now converting ...")
def main_progress_hook(data):
match data["status"]:
case "error":
show_notification(
"Something went wrong while downloading the video", data["filename"]
)
case "finished":
show_notification("Downloaded", data["filename"])
# URL of the file you want to download
url = "http://example.com/path/to/file.mp4"
# Options for yt-dlp
ydl_opts = {
"outtmpl": "/path/to/downloaded/file.%(ext)s", # Specify the output path and template
"logger": MyLogger(), # Custom logger
"progress_hooks": [my_hook], # Progress hook
}
# Function to download the file
def download_file(url, options):
with yt_dlp.YoutubeDL(options) as ydl:
ydl.download([url])
class YtDLPDownloader:
downloads_queue = Queue()
def _worker(self):
while True:
task, args = self.downloads_queue.get()
try:
task(*args)
except Exception as e:
show_notification("Something went wrong", f"Reason: {e}")
self.downloads_queue.task_done()
def __init__(self):
self._thread = Thread(target=self._worker)
self._thread.daemon = True
self._thread.start()
# Function to download the file
def _download_file(self, url: str, title, custom_progress_hook, silent):
anime_title = sanitize_filename(title[0])
ydl_opts = {
"outtmpl": f"{downloads_dir}/{anime_title}/{anime_title}-episode {title[1]}.%(ext)s", # Specify the output path and template
"progress_hooks": [
main_progress_hook,
custom_progress_hook,
], # Progress hook
"silent": silent,
}
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
ydl.download([url])
def download_file(self, url: str, title, custom_progress_hook, silent=True):
self.downloads_queue.put(
(self._download_file, (url, title, custom_progress_hook, silent))
)
# Call the function
download_file(url, ydl_opts)
downloader = YtDLPDownloader()