Add video thumbnail serving
This commit is contained in:
parent
4b95a8130e
commit
4af1885757
37
app.py
37
app.py
|
@ -1,7 +1,7 @@
|
|||
#!/usr/bin/env python
|
||||
# pilpil-client 0.1
|
||||
# abelliqueux <contact@arthus.net>
|
||||
from flask import Flask, flash, request, redirect, url_for
|
||||
from flask import Flask, flash, request, redirect, url_for, send_file
|
||||
from flask_httpauth import HTTPBasicAuth
|
||||
import gettext
|
||||
import os
|
||||
|
@ -32,6 +32,7 @@ HTTP_secret = str(app.config['DEFAULT']['auth'])
|
|||
debug = app.config['DEFAULT']['debug']
|
||||
useSSL = app.config['DEFAULT']['useSSL']
|
||||
rssi_signal = 0
|
||||
thumbnails_folder = "thumb"
|
||||
|
||||
# HTTP upload settings
|
||||
app.config['UPLOAD_FOLDER'] = upload_folder
|
||||
|
@ -51,6 +52,7 @@ if useSSL:
|
|||
if debug:
|
||||
print(HTTP_url_scheme)
|
||||
|
||||
|
||||
@auth.verify_password
|
||||
def verify_password(username, password):
|
||||
'''
|
||||
|
@ -59,6 +61,7 @@ def verify_password(username, password):
|
|||
if username in users and check_password_hash(users.get(username), password):
|
||||
return username
|
||||
|
||||
|
||||
def allowed_ext(filename):
|
||||
'''
|
||||
Check if filename has an allowed extension.
|
||||
|
@ -69,6 +72,7 @@ def allowed_ext(filename):
|
|||
if filename.rsplit(".", 1)[-1] in media_exts:
|
||||
return True
|
||||
|
||||
|
||||
def XMLify(string, child_node_name="child"):
|
||||
'''
|
||||
'''
|
||||
|
@ -77,6 +81,7 @@ def XMLify(string, child_node_name="child"):
|
|||
print("<{0}><{1}>{2}</{1}></{0}>".format(root_element, child_node_name, string))
|
||||
return "<{0}><{1}>{2}</{1}></{0}>".format(root_element, child_node_name, string)
|
||||
|
||||
|
||||
def get_RSSI():
|
||||
'''
|
||||
Get wireless signal level to AP
|
||||
|
@ -90,6 +95,7 @@ def get_RSSI():
|
|||
print(rssi_signal)
|
||||
return XMLify(rssi_signal, child_node_name="rssi")
|
||||
|
||||
|
||||
def running_on_rpi():
|
||||
with open("/proc/cpuinfo", "r") as wl:
|
||||
for line in wl:
|
||||
|
@ -97,6 +103,7 @@ def running_on_rpi():
|
|||
return True
|
||||
return False
|
||||
|
||||
|
||||
def led_init():
|
||||
'''
|
||||
Set ACT and PWR leds trigger and turn them off
|
||||
|
@ -106,6 +113,7 @@ def led_init():
|
|||
led_set(0, 0)
|
||||
led_set(1, 0)
|
||||
|
||||
|
||||
def led_set(led_id, state):
|
||||
'''
|
||||
Set led with id led_id to state.
|
||||
|
@ -114,6 +122,7 @@ def led_set(led_id, state):
|
|||
# state : 0 = off, 1 = on
|
||||
os.system("echo {} | sudo tee /sys/class/leds/led{}/brightness".format(str(state), str(led_id)))
|
||||
|
||||
|
||||
def blink_pi(n):
|
||||
'''
|
||||
Blink ACT and PWR leds altenatively n times to allow physical identification.
|
||||
|
@ -130,6 +139,7 @@ def blink_pi(n):
|
|||
led_set(1, 0)
|
||||
return "OK"
|
||||
|
||||
|
||||
def thread_blink():
|
||||
'''
|
||||
Blink leds as a thread to avoid blocking.
|
||||
|
@ -137,12 +147,13 @@ def thread_blink():
|
|||
th = threading.Thread(target=blink_pi, args=(16,))
|
||||
th.start()
|
||||
|
||||
|
||||
def list_media_files(folder):
|
||||
'''
|
||||
List files in folder which extension is allowed (exists in media_exts).
|
||||
'''
|
||||
if os.path.exists(folder):
|
||||
files = os.listdir(folder);
|
||||
files = os.listdir(folder)
|
||||
medias = []
|
||||
for fd in files:
|
||||
if len(fd.split('.')) > 1:
|
||||
|
@ -152,16 +163,25 @@ def list_media_files(folder):
|
|||
else:
|
||||
return []
|
||||
|
||||
|
||||
def generate_thumbnails():
|
||||
media_files = list_media_files(upload_folder)
|
||||
for media in media_files:
|
||||
subprocess.call(['ffmpeg', '-i', upload_folder + "/" + media, '-q:v', '30', '-s' , '160x120' , '-vf' , 'boxblur=2' , '-ss', '00:00:01.000', '-vframes', '1', thumbnails_folder + "/" + media + ".jpg", "-y"])
|
||||
|
||||
|
||||
@app.route("/")
|
||||
@auth.login_required
|
||||
def main():
|
||||
return _("Nothing to see here !")
|
||||
|
||||
|
||||
@app.route("/rssi")
|
||||
@auth.login_required
|
||||
def rssi():
|
||||
return get_RSSI()
|
||||
|
||||
|
||||
@app.route("/blink")
|
||||
@auth.login_required
|
||||
def blink():
|
||||
|
@ -169,6 +189,7 @@ def blink():
|
|||
thread_blink()
|
||||
return _("Blinkin")
|
||||
|
||||
|
||||
@app.route("/reboot")
|
||||
@auth.login_required
|
||||
def reboot():
|
||||
|
@ -176,6 +197,7 @@ def reboot():
|
|||
print(stdout)
|
||||
return _("Rebooting...")
|
||||
|
||||
|
||||
@app.route("/poweroff")
|
||||
@auth.login_required
|
||||
def shutdown():
|
||||
|
@ -185,6 +207,7 @@ def shutdown():
|
|||
|
||||
# File upload
|
||||
|
||||
|
||||
@app.route('/upload', methods=['GET', 'POST'])
|
||||
@auth.login_required
|
||||
def upload_file(over_write=1):
|
||||
|
@ -211,7 +234,17 @@ def upload_file(over_write=1):
|
|||
return "File exists, skipping..."
|
||||
return "OK"
|
||||
|
||||
|
||||
|
||||
@app.route("/thumb/<media>")
|
||||
@auth.login_required
|
||||
def get_thumbnail(media):
|
||||
if media:
|
||||
return send_file( thumbnails_folder + os.sep + media + ".jpg", mimetype='image/jpg')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
generate_thumbnails()
|
||||
# Turn ACT and POW leds off on start
|
||||
if running_on_rpi():
|
||||
led_init()
|
||||
|
|
Loading…
Reference in New Issue