Add client ids and quota as well as queue limit

This commit is contained in:
Phillip Kühne 2021-10-06 00:45:57 +02:00
parent 14ce3e07f5
commit 78a39b8d17
5 changed files with 137 additions and 60 deletions

View File

@ -4,16 +4,18 @@ import sqlite3
import pandas import pandas
from io import StringIO from io import StringIO
song_table = "songs" song_table = "songs"
entry_table = "entries" entry_table = "entries"
index_label = "Id" index_label = "Id"
done_table = "done_songs" done_table = "done_songs"
def open_db(): def open_db():
conn = sqlite3.connect("/tmp/karaoqueue.db") conn = sqlite3.connect("/tmp/karaoqueue.db")
conn.execute('PRAGMA encoding = "UTF-8";') conn.execute('PRAGMA encoding = "UTF-8";')
return conn return conn
def import_songs(song_csv): def import_songs(song_csv):
print("Start importing Songs...") print("Start importing Songs...")
df = pandas.read_csv(StringIO(song_csv), sep=';') df = pandas.read_csv(StringIO(song_csv), sep=';')
@ -27,10 +29,11 @@ def import_songs(song_csv):
print("Imported songs ({} in Database)".format(num_songs)) print("Imported songs ({} in Database)".format(num_songs))
return("Imported songs ({} in Database)".format(num_songs)) return("Imported songs ({} in Database)".format(num_songs))
def create_entry_table(): def create_entry_table():
conn = open_db() conn = open_db()
conn.execute('CREATE TABLE IF NOT EXISTS '+entry_table + conn.execute('CREATE TABLE IF NOT EXISTS '+entry_table +
' (ID INTEGER PRIMARY KEY NOT NULL, Song_Id INTEGER NOT NULL, Name VARCHAR(255))') ' (ID INTEGER PRIMARY KEY NOT NULL, Song_Id INTEGER NOT NULL, Name VARCHAR(255), Client_Id VARCHAR(36))')
conn.close() conn.close()
@ -40,6 +43,7 @@ def create_done_song_table():
' (Song_Id INTEGER PRIMARY KEY NOT NULL, Plays INTEGER)') ' (Song_Id INTEGER PRIMARY KEY NOT NULL, Plays INTEGER)')
conn.close() conn.close()
def create_song_table(): def create_song_table():
conn = open_db() conn = open_db()
conn.execute("CREATE TABLE IF NOT EXISTS \""+song_table+"""\" ( conn.execute("CREATE TABLE IF NOT EXISTS \""+song_table+"""\" (
@ -55,6 +59,7 @@ def create_song_table():
)""") )""")
conn.close() conn.close()
def create_list_view(): def create_list_view():
conn = open_db() conn = open_db()
conn.execute("""CREATE VIEW IF NOT EXISTS [Liste] AS conn.execute("""CREATE VIEW IF NOT EXISTS [Liste] AS
@ -72,6 +77,7 @@ def create_done_song_view():
WHERE done_songs.Song_Id=songs.Id""") WHERE done_songs.Song_Id=songs.Id""")
conn.close() conn.close()
def get_list(): def get_list():
conn = open_db() conn = open_db()
conn.row_factory = sqlite3.Row conn.row_factory = sqlite3.Row
@ -86,35 +92,40 @@ def get_played_list():
cur.execute("SELECT * FROM Abspielliste") cur.execute("SELECT * FROM Abspielliste")
return cur.fetchall() return cur.fetchall()
def get_song_list(): def get_song_list():
conn =open_db() conn = open_db()
cur = conn.cursor() cur = conn.cursor()
cur.execute("SELECT Artist || \" - \" || Title AS Song, Id FROM songs;") cur.execute("SELECT Artist || \" - \" || Title AS Song, Id FROM songs;")
return cur.fetchall() return cur.fetchall()
def get_song_completions(input_string): def get_song_completions(input_string):
conn = open_db() conn = open_db()
cur = conn.cursor() cur = conn.cursor()
# Don't look, it burns... # Don't look, it burns...
prepared_string = "%{0}%".format(input_string).upper() # "Test" -> "%TEST%" prepared_string = "%{0}%".format(
input_string).upper() # "Test" -> "%TEST%"
print(prepared_string) print(prepared_string)
cur.execute( cur.execute(
"SELECT Title || \" - \" || Artist AS Song, Id FROM songs WHERE REPLACE(REPLACE(REPLACE(REPLACE(UPPER( SONG ),'ö','Ö'),'ü','Ü'),'ä','Ä'),'ß','') LIKE (?) LIMIT 20;", (prepared_string,)) "SELECT Title || \" - \" || Artist AS Song, Id FROM songs WHERE REPLACE(REPLACE(REPLACE(REPLACE(UPPER( SONG ),'ö','Ö'),'ü','Ü'),'ä','Ä'),'ß','') LIKE (?) LIMIT 20;", (prepared_string,))
return cur.fetchall() return cur.fetchall()
def add_entry(name,song_id):
def add_entry(name, song_id, client_id):
conn = open_db() conn = open_db()
cur = conn.cursor() cur = conn.cursor()
cur.execute( cur.execute(
"INSERT INTO entries (Song_Id,Name) VALUES(?,?);", (song_id,name)) "INSERT INTO entries (Song_Id,Name,Client_Id) VALUES(?,?,?);", (song_id, name, client_id))
conn.commit() conn.commit()
conn.close() conn.close()
return return
def add_sung_song(entry_id): def add_sung_song(entry_id):
conn = open_db() conn = open_db()
cur = conn.cursor() cur = conn.cursor()
cur.execute("""SELECT Song_Id FROM entries WHERE Id=?""",(entry_id,)) cur.execute("""SELECT Song_Id FROM entries WHERE Id=?""", (entry_id,))
song_id = cur.fetchone()[0] song_id = cur.fetchone()[0]
cur.execute("""INSERT OR REPLACE INTO done_songs (Song_Id, Plays) cur.execute("""INSERT OR REPLACE INTO done_songs (Song_Id, Plays)
VALUES("""+str(song_id)+""", VALUES("""+str(song_id)+""",
@ -127,6 +138,19 @@ def add_sung_song(entry_id):
conn.close() conn.close()
return True return True
def check_entry_quota(client_id):
conn = open_db()
cur = conn.cursor()
cur.execute("SELECT Count(*) FROM entries WHERE entries.Client_Id = ?", (client_id,))
return cur.fetchall()[0][0]
def check_queue_length():
conn = open_db()
cur = conn.cursor()
cur.execute("SELECT Count(*) FROM entries")
return cur.fetchall()[0][0]
def clear_played_songs(): def clear_played_songs():
conn = open_db() conn = open_db()
cur = conn.cursor() cur = conn.cursor()
@ -135,10 +159,11 @@ def clear_played_songs():
conn.close() conn.close()
return True return True
def delete_entry(id): def delete_entry(id):
conn = open_db() conn = open_db()
cur = conn.cursor() cur = conn.cursor()
cur.execute("DELETE FROM entries WHERE id=?",(id,)) cur.execute("DELETE FROM entries WHERE id=?", (id,))
conn.commit() conn.commit()
conn.close() conn.close()
return True return True
@ -147,7 +172,7 @@ def delete_entry(id):
def delete_entries(ids): def delete_entries(ids):
idlist = [] idlist = []
for x in ids: for x in ids:
idlist.append( (x,) ) idlist.append((x,))
try: try:
conn = open_db() conn = open_db()
cur = conn.cursor() cur = conn.cursor()
@ -158,6 +183,7 @@ def delete_entries(ids):
except sqlite3.Error as error: except sqlite3.Error as error:
return -1 return -1
def delete_all_entries(): def delete_all_entries():
conn = open_db() conn = open_db()
cur = conn.cursor() cur = conn.cursor()

View File

@ -2,6 +2,7 @@ import requests
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
import json import json
import os import os
import uuid
data_directory = "data" data_directory = "data"
config_file = data_directory+"/config.json" config_file = data_directory+"/config.json"
@ -21,6 +22,13 @@ def get_songs(url):
r = requests.get(url) r = requests.get(url)
return r.text return r.text
def is_valid_uuid(val):
try:
uuid.UUID(str(val))
return True
except ValueError:
return False
def check_config_exists(): def check_config_exists():
return os.path.isfile(config_file) return os.path.isfile(config_file)
@ -31,9 +39,11 @@ def setup_config(app):
config = json.load(handle) config = json.load(handle)
print("Loaded existing config") print("Loaded existing config")
else: else:
config = {'username': 'admin', 'password': 'changeme'} config = {'username': 'admin', 'password': 'changeme', 'entryquota': 3, 'maxqueue': 20}
with open(config_file, 'w') as handle: with open(config_file, 'w') as handle:
json.dump(config, handle, indent=4, sort_keys=True) json.dump(config, handle, indent=4, sort_keys=True)
print("Wrote new config") print("Wrote new config")
app.config['BASIC_AUTH_USERNAME'] = config['username'] app.config['BASIC_AUTH_USERNAME'] = config['username']
app.config['BASIC_AUTH_PASSWORD'] = config['password'] app.config['BASIC_AUTH_PASSWORD'] = config['password']
app.config['ENTRY_QUOTA'] = config['entryquota']
app.config['MAX_QUEUE'] = config['maxqueue']

View File

@ -19,16 +19,39 @@ def home():
@app.route('/api/enqueue', methods=['POST']) @app.route('/api/enqueue', methods=['POST'])
def enqueue(): def enqueue():
if accept_entries: if not request.json:
if not request.json: print(request.data)
print(request.data) abort(400)
abort(400) client_id = request.json['client_id']
name = request.json['name'] if not helpers.is_valid_uuid(client_id):
song_id = request.json['id'] print(request.data)
database.add_entry(name, song_id) abort(400)
return Response('{"status":"OK"}', mimetype='text/json') name = request.json['name']
song_id = request.json['id']
if request.authorization:
database.add_entry(name, song_id, client_id)
return Response('{"status":"OK"}', mimetype='text/json')
else: else:
return Response('{"status":"Currently not accepting entries"}', mimetype='text/json',status=423) if accept_entries:
if not request.json:
print(request.data)
abort(400)
client_id = request.json['client_id']
if not helpers.is_valid_uuid(client_id):
print(request.data)
abort(400)
name = request.json['name']
song_id = request.json['id']
if database.check_queue_length() < app.config['MAX_QUEUE']:
if database.check_entry_quota(client_id) < app.config['ENTRY_QUOTA']:
database.add_entry(name, song_id, client_id)
return Response('{"status":"OK"}', mimetype='text/json')
else:
return Response('{"status":"Too many queued entries from this client"}', mimetype='text/json',status=423)
else:
return Response('{"status":"Queue full"}', mimetype='text/json',status=423)
else:
return Response('{"status":"Currently not accepting entries"}', mimetype='text/json',status=423)
@app.route("/list") @app.route("/list")

View File

@ -68,8 +68,7 @@
<footer class="footer"> <footer class="footer">
<div class="container text-center py-3"> <div class="container text-center py-3">
{% if not auth %} {% if not auth %}
<a href="/login" class="ml-1 mr-1"><i <a href="/login" class="ml-1 mr-1"><i class="fas fa-sign-in-alt mr-1"></i><span>Login</span></a>
class="fas fa-sign-in-alt mr-1"></i><span>Login</span></a>
{% endif %} {% endif %}
<!--<a href="https://github.com/PhoenixTwoFive/karaoqueue" <!--<a href="https://github.com/PhoenixTwoFive/karaoqueue"
class="ml-1 mr-1"><i class="fab fa-github mr-1"></i><span>Github</span></a>--> class="ml-1 mr-1"><i class="fab fa-github mr-1"></i><span>Github</span></a>-->
@ -83,28 +82,46 @@
<!-- Placed at the end of the document so the pages load faster --> <!-- Placed at the end of the document so the pages load faster -->
<script src="https://code.jquery.com/jquery-3.2.1.min.js" <script src="https://code.jquery.com/jquery-3.2.1.min.js"
integrity="sha384-xBuQ/xzmlsLoJpyjoggmTEz8OWUFM0/RC5BsqQBDX2v5cMvDHcMakNTNrHIW2I5f" crossorigin="anonymous"> integrity="sha384-xBuQ/xzmlsLoJpyjoggmTEz8OWUFM0/RC5BsqQBDX2v5cMvDHcMakNTNrHIW2I5f" crossorigin="anonymous">
</script> </script>
<script src="https://cdnjs.cloudflare.com/ajax/libs/popper.js/1.14.3/umd/popper.min.js" <script src="https://cdnjs.cloudflare.com/ajax/libs/popper.js/1.14.3/umd/popper.min.js"
integrity="sha384-ZMP7rVo3mIykV+2+9J3UJ46jBk0WLaUAdn689aCwoqbBJiSnjAK/l8WvCWPIPm49" crossorigin="anonymous"> integrity="sha384-ZMP7rVo3mIykV+2+9J3UJ46jBk0WLaUAdn689aCwoqbBJiSnjAK/l8WvCWPIPm49" crossorigin="anonymous">
</script> </script>
<script src="https://stackpath.bootstrapcdn.com/bootstrap/4.1.3/js/bootstrap.min.js" <script src="https://stackpath.bootstrapcdn.com/bootstrap/4.1.3/js/bootstrap.min.js"
integrity="sha384-ChfqqxuZUCnJSK3+MXmPNIyE6ZbWh2IMqE241rYiqJxyMiZ6OW/JmZQ5stwEULTy" crossorigin="anonymous"> integrity="sha384-ChfqqxuZUCnJSK3+MXmPNIyE6ZbWh2IMqE241rYiqJxyMiZ6OW/JmZQ5stwEULTy" crossorigin="anonymous">
</script> </script>
<script src="https://cdnjs.cloudflare.com/ajax/libs/bootbox.js/4.4.0/bootbox.min.js" <script src="https://cdnjs.cloudflare.com/ajax/libs/bootbox.js/4.4.0/bootbox.min.js"
integrity="sha256-4F7e4JsAJyLUdpP7Q8Sah866jCOhv72zU5E8lIRER4w=" crossorigin="anonymous"> integrity="sha256-4F7e4JsAJyLUdpP7Q8Sah866jCOhv72zU5E8lIRER4w=" crossorigin="anonymous">
</script> </script>
<script src="https://unpkg.com/bootstrap-table@1.15.3/dist/bootstrap-table.min.js"></script> <script src="https://unpkg.com/bootstrap-table@1.15.3/dist/bootstrap-table.min.js"></script>
<script src="https://unpkg.com/bootstrap-table@1.15.3/dist/extensions/auto-refresh/bootstrap-table-auto-refresh.min.js"></script> <script
src="https://unpkg.com/bootstrap-table@1.15.3/dist/extensions/auto-refresh/bootstrap-table-auto-refresh.min.js"></script>
<script src="https://gitcdn.github.io/bootstrap-toggle/2.2.2/js/bootstrap-toggle.min.js"></script> <script src="https://gitcdn.github.io/bootstrap-toggle/2.2.2/js/bootstrap-toggle.min.js"></script>
{% block extrajs %}{% endblock %} {% block extrajs %}{% endblock %}
<script> <script>
$(document).ready(function () { $(document).ready(function () {
loadOrGenerateClientId()
// get current URL path and assign 'active' class // get current URL path and assign 'active' class
var pathname = window.location.pathname; var pathname = window.location.pathname;
$('.navbar-nav > li > a[href="' + pathname + '"]').parent().addClass('active'); $('.navbar-nav > li > a[href="' + pathname + '"]').parent().addClass('active');
$('[data-toggle="tooltip"]').tooltip() $('[data-toggle="tooltip"]').tooltip()
}) })
function create_UUID() {
var dt = new Date().getTime();
var uuid = 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, function (c) {
var r = (dt + Math.random() * 16) % 16 | 0;
dt = Math.floor(dt / 16);
return (c == 'x' ? r : (r & 0x3 | 0x8)).toString(16);
});
return uuid;
}
function loadOrGenerateClientId() {
if (!localStorage.getItem("clientId")) {
localStorage.setItem("clientId",create_UUID())
}
}
</script> </script>
</body> </body>

View File

@ -17,8 +17,8 @@
</button> </button>
</div> </div>
<div class="form-group"> <div class="form-group">
<form id="nameForm"> <form id="nameForm">
<div class="modal-body"> <div class="modal-body">
<label for="singerNameInput">Sängername</label> <label for="singerNameInput">Sängername</label>
<input type="text" class="form-control" id="singerNameInput" placeholder="Max Mustermann" <input type="text" class="form-control" id="singerNameInput" placeholder="Max Mustermann"
required> required>
@ -28,7 +28,7 @@
<button type="button" class="btn btn-secondary" data-dismiss="modal">Close</button> <button type="button" class="btn btn-secondary" data-dismiss="modal">Close</button>
<button type="submit" class="btn btn-primary" id="submitSongButton">Anmelden</button> <button type="submit" class="btn btn-primary" id="submitSongButton">Anmelden</button>
</div> </div>
</form> </form>
</div> </div>
</div> </div>
</div> </div>
@ -38,18 +38,18 @@
<script> <script>
$(document).ready(function () { $(document).ready(function () {
$("#filter").focus(); $("#filter").focus();
$("#filter").keyup( function () { $("#filter").keyup(function () {
var value = $(this).val().toLowerCase(); var value = $(this).val().toLowerCase();
//alert(value); //alert(value);
if(value.length >= 1) { if (value.length >= 1) {
$.getJSON("/api/songs/compl", { search: value }, function (data) { $.getJSON("/api/songs/compl", { search: value }, function (data) {
var items = []; var items = [];
$.each(data, function (key, val) { $.each(data, function (key, val) {
items.push("<tr><td>"+val[0]+`</td> items.push("<tr><td>" + val[0] + `</td>
<td><button type='button' <td><button type='button'
class='btn btn-primary justify-content-center align-content-between enqueueButton' class='btn btn-primary justify-content-center align-content-between enqueueButton'
data-toggle='modal' data-toggle='modal'
data-target='#enqueueModal' onclick='setSelectedId(`+val[1]+`)'><i data-target='#enqueueModal' onclick='setSelectedId(`+ val[1] + `)'><i
class="fas fa-plus"></i></button></td> class="fas fa-plus"></i></button></td>
</tr>`) </tr>`)
}); });
@ -63,17 +63,18 @@
} }
}); });
$("#nameForm").submit( function (e) { $("#nameForm").submit(function (e) {
e.preventDefault(); e.preventDefault();
submitModal(); submitModal();
}); });
}); });
function enqueue(id,name,success_callback, blocked_callback) { function enqueue(client_id, id, name, success_callback, blocked_callback) {
var data = { var data = {
"name": name, "name": name,
"id": id "id": id,
"client_id": client_id
} }
$.ajax({ $.ajax({
type: 'POST', type: 'POST',
@ -89,20 +90,20 @@
} }
function setSelectedId(id) { function setSelectedId(id) {
$("#selectedId").attr("value",id); $("#selectedId").attr("value", id);
} }
function submitModal() { function submitModal() {
var name = $("#singerNameInput").val(); var name = $("#singerNameInput").val();
var id = $("#selectedId").attr("value"); var id = $("#selectedId").attr("value");
enqueue(id,name,function () { enqueue(localStorage.getItem("clientId"),id, name, function () {
$("#enqueueModal").modal('hide'); $("#enqueueModal").modal('hide');
window.location.href = '/#end'; window.location.href = '/#end';
}, function () { }, function (response) {
bootbox.alert({ bootbox.alert({
message: "Es werden leider keine neuen Anmeldungen mehr angenommen. Tut mir leid :(", message: "Deine Eintragung konnte leider nicht vorgenommen werden.\nReason: "+response.responseJSON.status,
}); });
$(".enqueueButton").prop("disabled",true); entriesAccepted();
$("#enqueueModal").modal('hide'); $("#enqueueModal").modal('hide');
}); });
@ -111,15 +112,15 @@
} }
function entriesAccepted() { function entriesAccepted() {
$.getJSON("/api/entries/accept", (data,out) => { $.getJSON("/api/entries/accept", (data, out) => {
if(data["value"]==0) { if (data["value"] == 0) {
$(".enqueueButton").prop("disabled",true) $(".enqueueButton").prop("disabled", true)
$(".enqueueButton").prop("style","pointer-events: none;") $(".enqueueButton").prop("style", "pointer-events: none;")
$(".enqueueButton").wrap("<span class='tooltip-span' tabindex='0' data-toggle='tooltip' data-placement='top'></span>" ); $(".enqueueButton").wrap("<span class='tooltip-span' tabindex='0' data-toggle='tooltip' data-placement='top'></span>");
$(".tooltip-span").prop("title","Eintragungen sind leider nicht mehr möglich.") $(".tooltip-span").prop("title", "Eintragungen sind leider nicht mehr möglich.")
$('[data-toggle="tooltip"]').tooltip() $('[data-toggle="tooltip"]').tooltip()
} else { } else {
$(".enqueueButton").prop("disabled",false) $(".enqueueButton").prop("disabled", false)
} }
}) })