90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
#!/usr/bin/python3
|
|
|
|
from flask import Flask, render_template, request
|
|
import variables as var
|
|
import os.path
|
|
from os import listdir
|
|
import random
|
|
|
|
|
|
class ReverseProxied(object):
|
|
'''Wrap the application in this middleware and configure the
|
|
front-end server to add these headers, to let you quietly bind
|
|
this to a URL other than / and to an HTTP scheme that is
|
|
different than what is used locally.
|
|
|
|
In nginx:
|
|
location /myprefix {
|
|
proxy_pass http://192.168.0.1:5001;
|
|
proxy_set_header Host $host;
|
|
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
|
proxy_set_header X-Scheme $scheme;
|
|
proxy_set_header X-Script-Name /myprefix;
|
|
}
|
|
|
|
:param app: the WSGI application
|
|
'''
|
|
|
|
def __init__(self, app):
|
|
self.app = app
|
|
|
|
def __call__(self, environ, start_response):
|
|
script_name = environ.get('HTTP_X_SCRIPT_NAME', '')
|
|
if script_name:
|
|
environ['SCRIPT_NAME'] = script_name
|
|
path_info = environ['PATH_INFO']
|
|
if path_info.startswith(script_name):
|
|
environ['PATH_INFO'] = path_info[len(script_name):]
|
|
|
|
scheme = environ.get('HTTP_X_SCHEME', '')
|
|
if scheme:
|
|
environ['wsgi.url_scheme'] = scheme
|
|
real_ip = environ.get('HTTP_X_REAL_IP', '')
|
|
if real_ip:
|
|
environ['REMOTE_ADDR'] = real_ip
|
|
return self.app(environ, start_response)
|
|
|
|
|
|
web = Flask(__name__)
|
|
web.wsgi_app = ReverseProxied(web.wsgi_app)
|
|
|
|
|
|
@web.route("/", methods=['GET', 'POST'])
|
|
def index():
|
|
folder_path = var.music_folder
|
|
files = {}
|
|
dirs = [f for f in listdir(folder_path) if os.path.isdir(os.path.join(folder_path, f))]
|
|
for director in dirs:
|
|
files[director] = [f for f in listdir(folder_path + director) if os.path.isfile(os.path.join(folder_path + director, f))]
|
|
|
|
if request.method == 'POST':
|
|
if 'add_music' in request.form and ".." not in request.form['add_music']:
|
|
var.playlist.append(request.form['add_music'])
|
|
if 'add_folder' in request.form and ".." not in request.form['add_folder']:
|
|
dir_files = [request.form['add_folder'] + '/' + i for i in files[request.form['add_folder']]]
|
|
var.playlist.extend(dir_files)
|
|
elif 'delete_music' in request.form:
|
|
try:
|
|
var.playlist.remove(request.form['delete_music'])
|
|
except ValueError:
|
|
pass
|
|
elif 'action' in request.form:
|
|
action = request.form['action']
|
|
if action == "randomize":
|
|
random.shuffle(var.playlist)
|
|
if var.current_music:
|
|
current_music = var.current_music[len(var.music_folder):]
|
|
else:
|
|
current_music = None
|
|
|
|
return render_template('index.html',
|
|
current_music=current_music,
|
|
user=var.user,
|
|
playlist=var.playlist,
|
|
all_files=files
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
web.run(port=8181, host="0.0.0.0")
|