79 lines
1.8 KiB
Python
79 lines
1.8 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
#
|
||
|
|
import os, time, subprocess
|
||
|
|
from flask import Flask
|
||
|
|
# HTTP auth
|
||
|
|
from flask_httpauth import HTTPBasicAuth
|
||
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
||
|
|
# HTTP Serve
|
||
|
|
from waitress import serve
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
|
||
|
|
auth = HTTPBasicAuth()
|
||
|
|
users = {
|
||
|
|
"": generate_password_hash("secret"),
|
||
|
|
}
|
||
|
|
|
||
|
|
@auth.verify_password
|
||
|
|
def verify_password(username, password):
|
||
|
|
if username in users and check_password_hash(users.get(username), password):
|
||
|
|
return username
|
||
|
|
|
||
|
|
local_bin="/home/pi/.local/bin/"
|
||
|
|
signal = 0
|
||
|
|
|
||
|
|
DEBUG = 0
|
||
|
|
|
||
|
|
# Get Wifi signal level
|
||
|
|
def getRSSI():
|
||
|
|
signal = subprocess.run( local_bin + "get_rssi.sh", capture_output=True)
|
||
|
|
signal = str(signal.stdout, 'UTF-8')[:-1][1:]
|
||
|
|
#print(signal)
|
||
|
|
return signal
|
||
|
|
|
||
|
|
# Blink the Pi led to allow identification
|
||
|
|
def blinkPy():
|
||
|
|
for j in range(10):
|
||
|
|
os.system('echo 1 | sudo dd status=none of=/sys/class/leds/led0/brightness > /dev/null 2>&1') # led on
|
||
|
|
time.sleep(.2)
|
||
|
|
os.system('echo 0 | sudo dd status=none of=/sys/class/leds/led0/brightness > /dev/null 2>&1') # led off
|
||
|
|
time.sleep(.2)
|
||
|
|
return "OK"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/")
|
||
|
|
@auth.login_required
|
||
|
|
def main():
|
||
|
|
return "Nothing to see here !"
|
||
|
|
|
||
|
|
@app.route("/rssi")
|
||
|
|
@auth.login_required
|
||
|
|
def signal():
|
||
|
|
return getRSSI()
|
||
|
|
|
||
|
|
@app.route("/blink")
|
||
|
|
@auth.login_required
|
||
|
|
def blink():
|
||
|
|
return blinkPy()
|
||
|
|
|
||
|
|
@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 "Shuting down..."
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
app.run()
|
||
|
|
#serve(app, host='0.0.0.0', port=8181)
|