130 lines
4.8 KiB
Python
130 lines
4.8 KiB
Python
|
from flask import Flask, render_template, request, Response, redirect, url_for, send_file
|
||
|
from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user, current_user
|
||
|
import subprocess
|
||
|
import os
|
||
|
import shutil
|
||
|
import zipfile
|
||
|
from io import BytesIO
|
||
|
|
||
|
app = Flask(__name__)
|
||
|
app.secret_key = 'your_secret_key'
|
||
|
|
||
|
# Configuration Flask-Login
|
||
|
login_manager = LoginManager()
|
||
|
login_manager.init_app(app)
|
||
|
login_manager.login_view = 'login'
|
||
|
|
||
|
# Utilisateur fictif
|
||
|
class User(UserMixin):
|
||
|
def __init__(self, id):
|
||
|
self.id = id
|
||
|
|
||
|
# Stockage simple d'utilisateurs (en production, utiliser une base de données)
|
||
|
users = {'garfi': {'admin': 'password123'}}
|
||
|
|
||
|
@login_manager.user_loader
|
||
|
def load_user(user_id):
|
||
|
return User(user_id)
|
||
|
|
||
|
@app.route('/login', methods=['GET', 'POST'])
|
||
|
def login():
|
||
|
if request.method == 'POST':
|
||
|
username = request.form['username']
|
||
|
password = request.form['password']
|
||
|
if username in users and users[username]['password'] == password:
|
||
|
user = User(username)
|
||
|
login_user(user)
|
||
|
return redirect(url_for('index'))
|
||
|
else:
|
||
|
return "Nom d'utilisateur ou mot de passe incorrect", 401
|
||
|
return render_template('login.html')
|
||
|
|
||
|
@app.route('/logout')
|
||
|
@login_required
|
||
|
def logout():
|
||
|
logout_user()
|
||
|
return redirect(url_for('login'))
|
||
|
|
||
|
@app.route('/')
|
||
|
@login_required
|
||
|
def index():
|
||
|
return render_template('stream.html')
|
||
|
|
||
|
@app.route('/download', methods=['POST'])
|
||
|
@login_required
|
||
|
def download():
|
||
|
url = request.form['url']
|
||
|
format_choice = request.form['format']
|
||
|
delete_choice = request.form['delete_choice']
|
||
|
copy_choice = request.form['copy_choice'] # Nouvelle option pour copier ou non
|
||
|
|
||
|
def generate_output():
|
||
|
# Répertoires pour le téléchargement en fonction du format choisi
|
||
|
output_dir = "/home/garfi/Musique/flac" if format_choice == "flac" else "/home/garfi/Musique/mp3"
|
||
|
|
||
|
# Si l'utilisateur a choisi de supprimer les anciens répertoires
|
||
|
if delete_choice == "1":
|
||
|
yield "Suppression des anciens répertoires...\n"
|
||
|
try:
|
||
|
if os.path.exists("/home/garfi/Musique/flac"):
|
||
|
shutil.rmtree("/home/garfi/Musique/flac")
|
||
|
yield "Le répertoire /home/garfi/Musique/flac a été supprimé.\n"
|
||
|
|
||
|
if os.path.exists("/home/garfi/Musique/mp3"):
|
||
|
shutil.rmtree("/home/garfi/Musique/mp3")
|
||
|
yield "Le répertoire /home/garfi/Musique/mp3 a été supprimé.\n"
|
||
|
except FileNotFoundError as e:
|
||
|
yield f"Erreur : {str(e)} - Le répertoire n'existe pas.\n"
|
||
|
except Exception as e:
|
||
|
yield f"Erreur inattendue lors de la suppression : {str(e)}\n"
|
||
|
|
||
|
# Créer le répertoire pour stocker les nouveaux fichiers
|
||
|
os.makedirs(output_dir, exist_ok=True)
|
||
|
|
||
|
# Lancer le téléchargement avec spotdl
|
||
|
command = f'spotdl --output "{{artist}}/{{album}}/{{track-number}} - {{title}}.{{output-ext}}" "{url}" --format={format_choice}'
|
||
|
process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, cwd=output_dir)
|
||
|
|
||
|
for line in process.stdout:
|
||
|
yield line # Envoyer les lignes de la commande en temps réel
|
||
|
|
||
|
process.wait()
|
||
|
if process.returncode == 0:
|
||
|
yield "\nTéléchargement terminé avec succès.\n"
|
||
|
|
||
|
|
||
|
# Si l'utilisateur a choisi de copier vers /mnt/data/Musique/
|
||
|
if copy_choice == "1":
|
||
|
share_dir = "/mnt/data/Musique/"
|
||
|
try:
|
||
|
shutil.copytree(output_dir, share_dir, dirs_exist_ok=True)
|
||
|
yield f"\nLes fichiers ont été copiés vers {share_dir}.\n"
|
||
|
except Exception as e:
|
||
|
yield f"\nErreur lors de la copie des fichiers vers {share_dir} : {str(e)}\n"
|
||
|
else:
|
||
|
yield f"\nUne erreur est survenue avec le code {process.returncode}.\n"
|
||
|
|
||
|
return Response(generate_output(), mimetype='text/plain')
|
||
|
|
||
|
@app.route('/download_zip')
|
||
|
@login_required
|
||
|
def download_zip():
|
||
|
format_choice = request.args.get('format')
|
||
|
output_dir = "/home/garfi/Musique/flac" if format_choice == "flac" else "/home/garfi/Musique/mp3"
|
||
|
|
||
|
# Créer un fichier ZIP dans la mémoire
|
||
|
memory_file = BytesIO()
|
||
|
with zipfile.ZipFile(memory_file, 'w', zipfile.ZIP_DEFLATED) as zf:
|
||
|
for root, dirs, files in os.walk(output_dir):
|
||
|
for file in files:
|
||
|
file_path = os.path.join(root, file)
|
||
|
zf.write(file_path, arcname=os.path.relpath(file_path, output_dir))
|
||
|
|
||
|
memory_file.seek(0)
|
||
|
|
||
|
# Télécharger le fichier ZIP
|
||
|
return send_file(memory_file, mimetype='application/zip', as_attachment=True, download_name=f'musique_{format_choice}.zip')
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
app.run(debug=True)
|