Ajout d'une barre de progréssion lors du chargement des méthadonnées
This commit is contained in:
parent
be93154128
commit
84bd38eb30
76
main.py
76
main.py
@ -4,16 +4,59 @@ import requests
|
||||
from io import BytesIO
|
||||
from tkinter import *
|
||||
from PIL import Image, ImageTk
|
||||
from tkinter.ttk import Progressbar
|
||||
import threading
|
||||
import queue
|
||||
|
||||
def add_url():
|
||||
|
||||
#Ajoute un classe pour intercepter les logs
|
||||
class YTDLLogger:
|
||||
"""
|
||||
Logger personnalisé pour capturer et afficher les messages de yt-dlp.
|
||||
"""
|
||||
def debug(self, msg):
|
||||
print(msg) # Affiche les messages de debug
|
||||
ProgresseLabel.config(text=msg)
|
||||
|
||||
def warning(self, msg):
|
||||
print(f"WARNING: {msg}") # Affiche les avertissements
|
||||
|
||||
def error(self, msg):
|
||||
print(f"ERROR: {msg}") # Affiche les erreurs
|
||||
|
||||
#Récuperer l'url
|
||||
url = EntryURL.get()
|
||||
# Effacer l'entrée URL
|
||||
EntryURL.delete(0, END)
|
||||
|
||||
ydl_opts = {}
|
||||
ydl_opts = {
|
||||
'quiet': True, # Supprime la sortie standard (redirigée vers le logger)
|
||||
'logger': YTDLLogger(), # Utilise le logger personnalisé
|
||||
}
|
||||
|
||||
# Queue pour recuperer les donnes
|
||||
q = queue.Queue()
|
||||
|
||||
# Fonction qui exécute le téléchargement dans un thread séparé
|
||||
def process_download():
|
||||
try:
|
||||
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
||||
info = ydl.extract_info(url, download=False)
|
||||
info = ydl.extract_info(url, download=False) # Extraire les métadonnées sans télécharger
|
||||
q.put(info) # Mettre l'info dans la queue pour être récupérée par le thread principal
|
||||
except Exception as e:
|
||||
q.put(f"Erreur : {str(e)}") # Mettre l'erreur dans la queue
|
||||
finally:
|
||||
progress_windows.after(2000, progress_windows.destroy) # Fermer la fenêtre de progression après 2s
|
||||
|
||||
# Fonction pour vérifier la queue et mettre à jour l'interface
|
||||
def check_queue():
|
||||
try:
|
||||
# Essayer de récupérer l'info sans bloquer
|
||||
info = q.get_nowait()
|
||||
|
||||
if isinstance(info, dict): # Si l'info est un dictionnaire (métadonnées)
|
||||
ProgresseLabel.config(text=f"Dowloading complet")
|
||||
title = info.get('title', 'Titre indisponible')
|
||||
subtitlte = (
|
||||
info.get('artist', info.get('uploader', 'Artiste indisponible'))
|
||||
@ -63,9 +106,34 @@ def add_url():
|
||||
# Ajouter un bouton "Supprimer"
|
||||
menu_button = Button(cadre, text="Supprimer", relief=FLAT, command=cadre.destroy)
|
||||
menu_button.pack(side=RIGHT)
|
||||
else: # Si c'est une erreur
|
||||
ProgresseLabel.config(text=info)
|
||||
|
||||
# Effacer l'entrée URL
|
||||
EntryURL.delete(0, END)
|
||||
progress_windows.after(2000, progress_windows.destroy) # Fermer après 2s
|
||||
|
||||
except queue.Empty: # Si la queue est vide, vérifier à nouveau
|
||||
progress_windows.after(100, check_queue)
|
||||
|
||||
# Création de la fenêtre de progression
|
||||
progress_windows = Toplevel(root)
|
||||
|
||||
# Barre de progression et label
|
||||
progressBarMetahdonne = Progressbar(progress_windows, orient=HORIZONTAL, length=400, mode='indeterminate')
|
||||
progressBarMetahdonne.pack(side=TOP)
|
||||
progressBarMetahdonne.start()
|
||||
|
||||
ProgresseLabel = Label(progress_windows, text="Téléchargement")
|
||||
ProgresseLabel.pack(side=BOTTOM)
|
||||
|
||||
# Démarrer le téléchargement dans un thread séparé
|
||||
thread = threading.Thread(target=process_download, daemon=True)
|
||||
thread.start()
|
||||
|
||||
# Lancer la vérification périodique de la queue
|
||||
check_queue()
|
||||
|
||||
# Mainloop Tkinter
|
||||
root.mainloop()
|
||||
|
||||
# Fenêtre principale
|
||||
root = Tk()
|
||||
|
Loading…
Reference in New Issue
Block a user