spotdlweb/app.py
Gu1llaum-3 74e936c0f8
Some checks reported errors
continuous-integration/drone/push Build encountered an error
Changes to be committed:
modified:   .drone.yml
	modified:   app.py
	new file:   docker-compose.yaml
	modified:   static/js/script.js
	new file:   templates/finish_local.html
	renamed:    templates/finish.html -> templates/finish_server.html
	modified:   templates/index.html

Il est maintenant possible de choisir entre télécharger sur le server
ou en local.
2023-06-05 22:56:34 +02:00

110 lines
3.4 KiB
Python

from flask import Flask, request, redirect, url_for, send_file, render_template, send_from_directory
from subprocess import run
from datetime import datetime
import os
import logging
app = Flask(__name__)
def process_file(urls):
download_param_album = '{artist}/{album}/{artist} - {title}'
download_param_playlist = '{playlist}/{artists}/{album} - {title} {artist}'
download_param_track = '{artist}/{album}/{artist} - {title}'
os.chdir('downloads')
# os.system(f'rm -rf *')
for url in urls:
if url:
if "album" in url:
run(['python3', '-m', 'spotdl', url, '--output', download_param_album])
elif "playlist" in url:
run(['python3', '-m', 'spotdl', url, '--output', download_param_playlist])
elif "track" in url:
run(['python3', '-m', 'spotdl', url, '--output', download_param_track])
# os.system(f'zip -r musics.zip ./downloads')
# run(['zip', '-r', 'musics.zip', '.'])
os.chdir('../')
def process_file_local(urls):
download_param_album = '{artist}/{album}/{artist} - {title}'
download_param_playlist = '{playlist}/{artists}/{album} - {title} {artist}'
download_param_track = '{artist}/{album}/{artist} - {title}'
os.chdir('temp')
os.system(f'rm -rf *')
for url in urls:
if url:
if "album" in url:
run(['python3', '-m', 'spotdl', url, '--output', download_param_album])
elif "playlist" in url:
run(['python3', '-m', 'spotdl', url, '--output', download_param_playlist])
elif "track" in url:
run(['python3', '-m', 'spotdl', url, '--output', download_param_track])
run(['zip', '-r', 'musics.zip', '.'])
os.chdir('../')
@app.route('/', methods=['GET', 'POST'])
def upload_form():
return render_template('index.html')
# Fonctionne
# @app.route('/download/<filename>')
# def download_file(filename):
# PATH='file.txt'
# return send_file(PATH, as_attachment=True)
@app.route('/download', methods=['POST'])
def download_file():
if request.method == 'POST':
action = request.form.get('action')
url1 = request.form['url1']
url2 = request.form['url2']
url3 = request.form['url3']
urls = [url1, url2, url3]
# Vérifier si au moins un champ est vide
if not url1 and not url2 and not url3 :
return render_template('erreur.html')
if action == 'download':
# Créer le dossier 'downloads' s'il n'existe pas
if not os.path.exists('downloads'):
os.makedirs('downloads')
process_file(urls)
# path = "downloads/musics.zip"
# return send_file(path, as_attachment=True)
return render_template('finish_server.html')
if action == 'downloadlocal':
# Créer le dossier 'downloads' s'il n'existe pas
if not os.path.exists('temp'):
os.makedirs('temp')
process_file_local(urls)
return render_template('finish_local.html')
@app.route('/zip', methods=['GET', 'POST'])
def zip():
path = "temp/musics.zip"
return send_file(path, as_attachment=True)
@app.errorhandler(404)
def page_not_found(error): # error est necessaire
return render_template('404.html'), 404
if __name__ == '__main__':
app.run(host='0.0.0.0', debug=True, port=3000)