mirror of
https://github.com/PhoenixTwoFive/karaoqueue.git
synced 2025-05-19 02:51:48 +02:00
134 lines
4.7 KiB
Python
134 lines
4.7 KiB
Python
import requests
|
|
from bs4 import BeautifulSoup
|
|
import json
|
|
import os
|
|
import uuid
|
|
from flask import make_response, Flask
|
|
from functools import wraps, update_wrapper
|
|
from datetime import datetime
|
|
import database
|
|
|
|
data_directory = "data"
|
|
config_file = data_directory+"/config.json"
|
|
|
|
def create_data_directory():
|
|
if not os.path.exists(data_directory):
|
|
os.makedirs(data_directory)
|
|
|
|
|
|
def get_catalog_url():
|
|
r = requests.get('https://www.karafun.de/karaoke-song-list.html')
|
|
soup = BeautifulSoup(r.content, 'html.parser')
|
|
url = soup.findAll('a', href=True, text='Verfügbar in CSV-Format')[0]['href']
|
|
return url
|
|
|
|
def get_songs(url):
|
|
r = requests.get(url)
|
|
return r.text
|
|
|
|
def is_valid_uuid(val):
|
|
try:
|
|
uuid.UUID(str(val))
|
|
return True
|
|
except ValueError:
|
|
return False
|
|
|
|
def check_config_exists():
|
|
eng = database.get_db_engine()
|
|
with eng.connect() as conn:
|
|
if conn.dialect.has_table(conn, 'config'):
|
|
if (conn.execute("SELECT COUNT(*) FROM config").fetchone()[0] > 0): # type: ignore
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
return False
|
|
|
|
def load_version(app: Flask):
|
|
if os.environ.get("SOURCE_VERSION"):
|
|
app.config['VERSION'] = os.environ.get("SOURCE_VERSION")[0:7] # type: ignore
|
|
elif os.path.isfile(".version"):
|
|
with open('.version', 'r') as file:
|
|
data = file.read().replace('\n', '')
|
|
if data:
|
|
app.config['VERSION'] = data
|
|
else:
|
|
app.config['VERSION'] = ""
|
|
else:
|
|
app.config['VERSION'] = ""
|
|
|
|
def load_dbconfig(app: Flask):
|
|
if os.environ.get("FLASK_ENV") == "development":
|
|
app.config['DBCONNSTRING'] = os.environ.get("DBSTRING")
|
|
else:
|
|
if os.environ.get("DEPLOYMENT_PLATFORM") == "Heroku":
|
|
if os.environ.get("JAWSDB_MARIA_URL"):
|
|
app.config['DBCONNSTRING'] = os.environ.get("JAWSDB_MARIA_URL")
|
|
else:
|
|
app.config['DBCONNSTRING'] = ""
|
|
if os.environ.get("DEPLOYMENT_PLATFORM") == "Docker":
|
|
if os.environ.get("DBSTRING"):
|
|
app.config['DBCONNSTRING'] = os.environ.get("DBSTRING")
|
|
else:
|
|
app.config['DBCONNSTRING'] = ""
|
|
elif os.path.isfile(".dbconn"):
|
|
with open('.dbconn', 'r') as file:
|
|
data = file.read().replace('\n', '')
|
|
if data:
|
|
app.config['DBCONNSTRING'] = data
|
|
else:
|
|
app.config['DBCONNSTRING'] = ""
|
|
else:
|
|
exit("No database connection string found. Cannot continue. Please set the environment variable DBSTRING or create a file .dbconn in the root directory of the project.")
|
|
|
|
# Check if config exists in DB, if not, create it.
|
|
def setup_config(app: Flask):
|
|
if check_config_exists():
|
|
config = database.get_config_list()
|
|
print("Loaded existing config")
|
|
else:
|
|
config = {'username': 'admin', 'password': 'changeme', 'entryquota': 3, 'maxqueue': 20, 'entries_allowed': 1}
|
|
for key, value in config.items():
|
|
database.set_config(key, value)
|
|
print("Created new config")
|
|
app.config['BASIC_AUTH_USERNAME'] = config['username']
|
|
app.config['BASIC_AUTH_PASSWORD'] = config['password']
|
|
app.config['ENTRY_QUOTA'] = config['entryquota']
|
|
app.config['MAX_QUEUE'] = config['maxqueue']
|
|
app.config['ENTRIES_ALLOWED'] = bool(config['entries_allowed'])
|
|
|
|
# set queue admittance
|
|
def set_accept_entries(app: Flask, allowed: bool):
|
|
if allowed:
|
|
app.config['ENTRIES_ALLOWED'] = True
|
|
database.set_config('entries_allowed', '1')
|
|
else:
|
|
app.config['ENTRIES_ALLOWED'] = False
|
|
database.set_config('entries_allowed', '0')
|
|
|
|
# get queue admittance
|
|
def get_accept_entries(app: Flask) -> bool:
|
|
state = bool(int(database.get_config('entries_allowed')))
|
|
app.config['ENTRIES_ALLOWED'] = state
|
|
return state
|
|
|
|
|
|
# Write settings from current app.config to DB
|
|
def persist_config(app: Flask):
|
|
config = {'username': app.config['BASIC_AUTH_USERNAME'], 'password': app.config['BASIC_AUTH_PASSWORD'], 'entryquota': app.config['ENTRY_QUOTA'], 'maxqueue': app.config['MAX_QUEUE']}
|
|
for key, value in config.items():
|
|
database.set_config(key, value)
|
|
|
|
|
|
|
|
def nocache(view):
|
|
@wraps(view)
|
|
def no_cache(*args, **kwargs):
|
|
response = make_response(view(*args, **kwargs))
|
|
response.headers['Last-Modified'] = datetime.now() # type: ignore
|
|
response.headers['Cache-Control'] = 'no-store, no-cache, must-revalidate, post-check=0, pre-check=0, max-age=0'
|
|
response.headers['Pragma'] = 'no-cache'
|
|
response.headers['Expires'] = '-1'
|
|
return response
|
|
|
|
return update_wrapper(no_cache, view) |