pilpil-client/app.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os, time, subprocess, threading, toml
from flask import Flask, flash, request, redirect, url_for
# HTTP auth
from flask_httpauth import HTTPBasicAuth
from werkzeug.security import generate_password_hash, check_password_hash
# FILE UPLOAD
from werkzeug.utils import secure_filename

app = Flask(__name__)

app.config.from_file("defaults.toml", load=toml.load)
config_locations = ["./", "~/.", "~/.config/"]
for location in config_locations:
    # Optional config files, ~ is expanded to $HOME on *nix, %USERPROFILE% on windows
    # ~ app.config.from_file("videopi.toml", load=toml.load, silent=True)
    if app.config.from_file(os.path.expanduser( location + "pilpil-client.toml"), load=toml.load, silent=True):
        print("Found configuration file in " + os.path.expanduser( location ))


#UPLOAD_FOLDER = os.path.expanduser('~/Videos')
#ALLOWED_EXTENSIONS = {'avi', 'mkv', 'mp4'}

UPLOAD_FOLDER = os.path.expanduser(app.config['DEFAULT']['media_folder_local'])
ALLOWED_EXTENSIONS = app.config['DEFAULT']['media_exts']
HTTP_SECRET = str(app.config['DEFAULT']['auth'])
DEBUG = app.config['DEFAULT']['debug']

#HTTPS
#ASSETS_DIR = os.path.dirname(os.path.abspath(__file__))
# HTTP Serve
from waitress import serve

#app.secret_key = b'flafoudi'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
# Max upload size 100M ( see nginx default also )
app.config['MAX_CONTENT_LENGTH'] = 100 * 1000 * 1000

auth = HTTPBasicAuth()
users = {
    "": generate_password_hash(HTTP_SECRET),
}

@auth.verify_password
def verify_password(username, password):
    if username in users and check_password_hash(users.get(username), password):
        return username

signal = 0

# Check file extension is allowed
def allowed_file(filename):
    # Check for dot in filename
    if "." in filename:
        # Split from right at first dot to find ext and allow files with "." in name
        if filename.rsplit(".",1)[-1] in ALLOWED_EXTENSIONS:
            return True

# Get Wifi signal level
def getRSSI():
    signal = subprocess.run( "./get_rssi.sh", capture_output=True)
    print(signal)
    signal = str(signal.stdout, 'UTF-8').strip("-").strip("\n")
    return signal

def ledSetup():
    os.system('echo none | sudo tee /sys/class/leds/led0/trigger')
    os.system('echo none | sudo tee /sys/class/leds/led1/trigger')
    led(0, 0)
    led(1, 0)

def led( led_id, state ):
    # 0 : off, 1: on
    os.system('echo ' + str(state) + ' | sudo tee /sys/class/leds/led' + str(led_id) +'/brightness')

# Blink the Pi led to allow identification
def blinkPy():
    # disable mmc access led
    # ~ os.system('echo none | sudo tee /sys/class/leds/led0/trigger')
    # ~ os.system('echo none | sudo tee /sys/class/leds/led1/trigger')
    # Blink 10 times
    for j in range(16):
        led(0, 1)
        led(1, 0)
        time.sleep(.2)
        led(0, 0)
        led(1, 1)
        time.sleep(.2)
    # restore default behavior
    # ~ os.system('echo mmc0 | sudo tee /sys/class/leds/led0/trigger')
    # ~ os.system('echo default-on | sudo tee /sys/class/leds/led1/trigger')
    led(0, 0)
    led(1, 0)
    return "OK"

def threadBlink():
    th=threading.Thread(target=blinkPy)
    th.start()
    
@app.route("/")
@auth.login_required
def main():
    return "Nothing to see here !"

@app.route("/rssi")
@auth.login_required
def rssi():
    return getRSSI()

@app.route("/blink")
@auth.login_required
def blink():
    threadBlink()
    return "Blinkin"

@app.route("/reboot")
@auth.login_required
def reboot():
    stdout = subprocess.run(["sudo", "/usr/sbin/reboot"],  capture_output=True)
    print(stdout)
    return "Rebooting..."

@app.route("/poweroff")
@auth.login_required
def shutdown():
    stdout = subprocess.run(["sudo", "/usr/sbin/poweroff"], capture_output=True)
    print(stdout)
    return "Shutting down..."

# File upload

@app.route('/upload', methods=['GET', 'POST'])
@auth.login_required
def upload_file():
    if request.method == 'POST':
        # check if the post request has the file part
        if 'file' not in request.files:
            return "No file part: " + str(request.files)
        file = request.files['file']
        # If the user does not select a file, the browser submits an
        # empty file without a filename.
        if file.filename == '':
            return 'No selected file'
        if file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            file.save(os.path.join(UPLOAD_FOLDER, filename))
            return "File saved in " + UPLOAD_FOLDER
    return "OK"

if __name__ == '__main__':
    # Turn ACT and POW leds off on start
    ledSetup()
#    app.run()
    serve(app, host='127.0.0.1', port=5000, url_scheme='https')