forked from server/soundboard
125 lines
2.9 KiB
Python
125 lines
2.9 KiB
Python
import os
|
|
import sys
|
|
import subprocess
|
|
import sqlite3
|
|
|
|
from flask import Flask, render_template, request, redirect, url_for, g
|
|
|
|
import config
|
|
|
|
app = Flask(__name__)
|
|
app.jinja_env.trim_blocks = True
|
|
app.jinja_env.lstrip_blocks = True
|
|
|
|
# Credits: http://flask.pocoo.org/docs/0.12/patterns/sqlite3/
|
|
def getDB():
|
|
db = getattr(g, "_database", None)
|
|
|
|
if db is None:
|
|
if os.path.isabs(config.db):
|
|
dbPath = config.db
|
|
else:
|
|
dbPath = os.path.join(sys.path[0], config.db)
|
|
|
|
db = g._database = sqlite3.connect(dbPath)
|
|
db.row_factory = sqlite3.Row
|
|
|
|
return db
|
|
|
|
@app.teardown_appcontext
|
|
def closeDBConnection(exception):
|
|
db = getattr(g, "_database", None)
|
|
|
|
if db is not None:
|
|
db.close()
|
|
|
|
def queryDB(query, args=(), one=False):
|
|
cur = getDB().execute(query, args)
|
|
result = cur.fetchall()
|
|
|
|
cur.close()
|
|
|
|
return (result[0] if result else None) if one else result
|
|
|
|
@app.route("/")
|
|
@app.route("/edit")
|
|
@app.route("/play/<sound>")
|
|
@app.route("/say/", methods=["POST"])
|
|
@app.route("/say/<text>")
|
|
def index(sound=None, text=None, video=None):
|
|
sounds = [os.fsencode(file).decode() for file in os.listdir(config.path)]
|
|
sounds = sorted(sounds)
|
|
|
|
tags = queryDB("SELECT name FROM tag ORDER BY name COLLATE NOCASE")
|
|
|
|
if sound is not None and sound in sounds:
|
|
subprocess.Popen(["omxplayer", os.path.join(config.path, sound).encode("utf-8")], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
|
|
if text is None:
|
|
text = request.form.get("text")
|
|
|
|
if text is not None:
|
|
voice = request.form.get("voice", default="")
|
|
voice = voice if voice.strip() != "" else "DE"
|
|
speed = request.form.get("speed", default="")
|
|
speed = speed if speed.strip() != "" else "160"
|
|
pitch = request.form.get("pitch", default="")
|
|
pitch = pitch if pitch.strip() != "" else "50"
|
|
|
|
subprocess.Popen(["espeak", "-v", voice, "-s", speed, "-p", pitch, text], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
return redirect("/")
|
|
|
|
video = request.args.get("video")
|
|
|
|
if video is not None:
|
|
url = subprocess.check_output(["youtube-dl", "-g", "-f", "mp4", video]).decode()
|
|
subprocess.Popen(["omxplayer", url.split("\n")[1]])
|
|
subprocess.Popen(["omxplayer", "-b", url.split("\n")[0]])
|
|
|
|
killvideo = request.args.get("killvideo")
|
|
|
|
if killvideo is not None and killvideo in ["1", "true", "yes"]:
|
|
subprocess.Popen(["pkill", "-f", "omxplayer"])
|
|
|
|
if request.path == "/edit":
|
|
edit = True
|
|
else:
|
|
edit = False
|
|
|
|
return render_template("index.html", sounds=sounds, tags=tags, edit=edit)
|
|
|
|
@app.route("/edit/<sound>", methods=["GET", "POST"])
|
|
def edit(sound):
|
|
if request.method == "POST":
|
|
# TODO: Store changes
|
|
return redirect("/edit")
|
|
|
|
tags = queryDB("""\
|
|
SELECT
|
|
tag.name,
|
|
checked.id IS NOT NULL AS checked
|
|
FROM
|
|
tag
|
|
LEFT OUTER JOIN (
|
|
SELECT
|
|
tag.id
|
|
FROM
|
|
tag
|
|
JOIN
|
|
button_tags
|
|
ON
|
|
fk_tag = tag.id
|
|
WHERE fk_button = (
|
|
SELECT
|
|
button.id
|
|
FROM
|
|
button
|
|
WHERE
|
|
button.file = ?
|
|
)
|
|
) AS checked
|
|
ON tag.id = checked.id""", (sound,))
|
|
print(tags)
|
|
|
|
return render_template("edit.html", sound=sound, tags=tags)
|