373 lines
10 KiB
Python
Executable File
373 lines
10 KiB
Python
Executable File
import os
|
||
import json
|
||
import socket
|
||
import hashlib
|
||
from datetime import datetime
|
||
|
||
from flask import (
|
||
Flask, render_template,
|
||
send_from_directory, redirect,
|
||
request, abort
|
||
)
|
||
from flask_login import (
|
||
LoginManager, login_user,
|
||
login_required, logout_user,
|
||
UserMixin
|
||
)
|
||
from werkzeug.utils import secure_filename
|
||
|
||
# -------------------------------------------------
|
||
# Grundkonfiguration
|
||
# -------------------------------------------------
|
||
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
||
MEDIA_DIR = os.path.join(BASE_DIR, "media")
|
||
CONFIG_FILE = os.path.join(BASE_DIR, "config.json")
|
||
|
||
APP_VERSION = "2.1.0"
|
||
UPLOAD_EXTENSIONS = {".jpg", ".jpeg", ".png", ".mp4"}
|
||
|
||
app = Flask(__name__)
|
||
app.secret_key = "CHANGE_THIS_SECRET!!!"
|
||
|
||
login_manager = LoginManager(app)
|
||
login_manager.login_view = "login"
|
||
|
||
# -------------------------------------------------
|
||
# Config Helpers
|
||
# -------------------------------------------------
|
||
def load_config():
|
||
if not os.path.exists(CONFIG_FILE):
|
||
return {"admin": {}, "screens": {}}
|
||
with open(CONFIG_FILE) as f:
|
||
return json.load(f)
|
||
|
||
def save_config(cfg):
|
||
with open(CONFIG_FILE, "w") as f:
|
||
json.dump(cfg, f, indent=2)
|
||
f.flush()
|
||
os.fsync(f.fileno())
|
||
print("✅ config.json gespeichert")
|
||
|
||
# -------------------------------------------------
|
||
# Auth / Benutzer
|
||
# -------------------------------------------------
|
||
class Admin(UserMixin):
|
||
id = 1
|
||
|
||
@login_manager.user_loader
|
||
def load_user(user_id):
|
||
return Admin()
|
||
|
||
# -------------------------------------------------
|
||
# Login / Logout
|
||
# -------------------------------------------------
|
||
@app.route("/login", methods=["GET", "POST"])
|
||
def login():
|
||
config = load_config()
|
||
|
||
if request.method == "POST":
|
||
if (
|
||
request.form.get("username") == config["admin"].get("username")
|
||
and request.form.get("password") == config["admin"].get("password")
|
||
):
|
||
login_user(Admin())
|
||
return redirect("/admin")
|
||
|
||
return render_template("login.html")
|
||
|
||
@app.route("/logout")
|
||
def logout():
|
||
logout_user()
|
||
return redirect("/login")
|
||
|
||
# -------------------------------------------------
|
||
# Medien ausliefern
|
||
# -------------------------------------------------
|
||
@app.route("/media/<screen>/<path:filename>")
|
||
def media(screen, filename):
|
||
path = os.path.join(MEDIA_DIR, screen)
|
||
return send_from_directory(path, filename)
|
||
|
||
# -------------------------------------------------
|
||
# Player
|
||
# -------------------------------------------------
|
||
|
||
@app.route("/player/<screen>")
|
||
def player(screen):
|
||
config = load_config()
|
||
screens_cfg = config.setdefault("screens", {})
|
||
screen_cfg = screens_cfg.get(screen)
|
||
|
||
if not screen_cfg:
|
||
abort(404)
|
||
|
||
folder = os.path.join(MEDIA_DIR, screen)
|
||
playlist = screen_cfg.get("playlist", [])
|
||
|
||
show_images = screen_cfg.get("show_images", True)
|
||
show_videos = screen_cfg.get("show_videos", True)
|
||
|
||
def allowed_file(name):
|
||
if name.startswith("._"):
|
||
return False
|
||
ext = name.lower()
|
||
if ext.endswith((".jpg", ".jpeg", ".png")):
|
||
return show_images
|
||
if ext.endswith(".mp4"):
|
||
return show_videos
|
||
return False
|
||
|
||
# ------------------------------
|
||
# Normale Screen-Playlist
|
||
# ------------------------------
|
||
normal_files = []
|
||
|
||
# 1. Playlist-Reihenfolge
|
||
for name in playlist:
|
||
if os.path.exists(os.path.join(folder, name)) and allowed_file(name):
|
||
normal_files.append(name)
|
||
|
||
# 2. Neue Dateien hintendran
|
||
for f in sorted(os.listdir(folder)):
|
||
if f not in normal_files and allowed_file(f):
|
||
normal_files.append(f)
|
||
|
||
# ------------------------------
|
||
# Priority-Playlist (global)
|
||
# WICHTIG: IMMER definieren!
|
||
# ------------------------------
|
||
priority_cfg = config.get("priority", {})
|
||
|
||
if priority_cfg.get("enabled", False):
|
||
prio_files = priority_cfg.get("playlist", [])
|
||
else:
|
||
prio_files = []
|
||
|
||
return render_template(
|
||
"player.html",
|
||
screen=screen,
|
||
normal_files=normal_files, # ✅ explizit
|
||
prio_files=prio_files, # ✅ IMMER Liste
|
||
interval=screen_cfg.get("interval", 10)
|
||
)
|
||
|
||
|
||
# -------------------------------------------------
|
||
# Hash für automatisches Player-Reload
|
||
# -------------------------------------------------
|
||
@app.route("/playlist/<screen>/hash")
|
||
def playlist_hash(screen):
|
||
config = load_config()
|
||
|
||
# ✅ Globaler Hash über ALLE relevanten Daten
|
||
relevant = {
|
||
"screens": {
|
||
screen: config["screens"].get(screen, {})
|
||
},
|
||
"priority": config.get("priority", {})
|
||
}
|
||
|
||
blob = json.dumps(relevant, sort_keys=True).encode()
|
||
return hashlib.md5(blob).hexdigest()
|
||
|
||
|
||
# -------------------------------------------------
|
||
# Admin Dashboard
|
||
# -------------------------------------------------
|
||
|
||
@app.route("/admin")
|
||
@login_required
|
||
def admin():
|
||
cfg = load_config()
|
||
|
||
screens = {}
|
||
media_files = {}
|
||
screen_status = {}
|
||
|
||
for screen in os.listdir(MEDIA_DIR):
|
||
if screen == "priority":
|
||
continue # ✅ kein echter Screen
|
||
|
||
path = os.path.join(MEDIA_DIR, screen)
|
||
if not os.path.isdir(path):
|
||
continue
|
||
|
||
files = []
|
||
playlist = cfg["screens"][screen].get("playlist", [])
|
||
|
||
# 1️⃣ Playlist-Reihenfolge
|
||
for name in playlist:
|
||
file_path = os.path.join(path, name)
|
||
if not os.path.exists(file_path):
|
||
continue
|
||
|
||
ext = os.path.splitext(name)[1].lower()
|
||
ftype = "video" if ext == ".mp4" else "image"
|
||
size = os.path.getsize(file_path) // 1024
|
||
|
||
files.append({
|
||
"name": name,
|
||
"type": ftype,
|
||
"size": size
|
||
})
|
||
|
||
# 2️⃣ Neue Dateien anhängen (nicht in Playlist)
|
||
for f in sorted(os.listdir(path)):
|
||
if f in playlist or f.startswith("._"):
|
||
continue
|
||
|
||
file_path = os.path.join(path, f)
|
||
ext = os.path.splitext(f)[1].lower()
|
||
ftype = "video" if ext == ".mp4" else "image"
|
||
size = os.path.getsize(file_path) // 1024
|
||
|
||
files.append({
|
||
"name": f,
|
||
"type": ftype,
|
||
"size": size
|
||
})
|
||
|
||
screens[screen] = cfg["screens"][screen]
|
||
media_files[screen] = files
|
||
screen_status[screen] = "active" if files else "empty"
|
||
|
||
|
||
# --- PRIORITY EXTENSION: Admin ---
|
||
prio_dir = os.path.join(MEDIA_DIR, "priority")
|
||
os.makedirs(prio_dir, exist_ok=True)
|
||
|
||
priority_files = []
|
||
prio_playlist = cfg["priority"].get("playlist", [])
|
||
|
||
# 1️⃣ Playlist-Reihenfolge
|
||
for name in prio_playlist:
|
||
file_path = os.path.join(MEDIA_DIR, "priority")
|
||
if not os.path.exists(file_path):
|
||
continue
|
||
|
||
ext = os.path.splitext(name)[1].lower()
|
||
ftype = "video" if ext == ".mp4" else "image"
|
||
size = os.path.getsize(file_path) // 1024
|
||
|
||
priority_files.append({
|
||
"name": name,
|
||
"type": ftype,
|
||
"size": size
|
||
})
|
||
|
||
# 2️⃣ Neue Dateien anhängen (nicht in Playlist)
|
||
for f in sorted(os.listdir(prio_dir)):
|
||
if f in prio_playlist or f.startswith("._"):
|
||
continue
|
||
|
||
file_path = os.path.join(prio_dir, f)
|
||
ext = os.path.splitext(f)[1].lower()
|
||
ftype = "video" if ext == ".mp4" else "image"
|
||
size = os.path.getsize(file_path) // 1024
|
||
|
||
priority_files.append({
|
||
"name": f,
|
||
"type": ftype,
|
||
"size": size
|
||
})
|
||
|
||
|
||
return render_template(
|
||
"admin.html",
|
||
screens=screens,
|
||
media_files=media_files,
|
||
screen_status=screen_status,
|
||
version=APP_VERSION,
|
||
year=datetime.now().year,
|
||
hostname=os.uname().nodename,
|
||
priority_files=priority_files
|
||
)
|
||
|
||
|
||
# -------------------------------------------------
|
||
# Admin: Screen-Einstellungen speichern
|
||
# -------------------------------------------------
|
||
@app.route("/admin/update/<screen>", methods=["POST"])
|
||
@login_required
|
||
def update_screen(screen):
|
||
config = load_config()
|
||
cfg = config["screens"][screen]
|
||
|
||
cfg["interval"] = int(request.form.get("interval", 10))
|
||
cfg["show_images"] = "show_images" in request.form
|
||
cfg["show_videos"] = "show_videos" in request.form
|
||
|
||
save_config(config)
|
||
return redirect("/admin")
|
||
|
||
# -------------------------------------------------
|
||
# Admin: Upload
|
||
# -------------------------------------------------
|
||
@app.route("/admin/upload/<screen>", methods=["POST"])
|
||
@login_required
|
||
def upload(screen):
|
||
file = request.files.get("file")
|
||
if not file:
|
||
return redirect("/admin")
|
||
|
||
ext = os.path.splitext(file.filename)[1].lower()
|
||
if ext not in UPLOAD_EXTENSIONS:
|
||
return "Ungültiger Dateityp", 400
|
||
|
||
filename = secure_filename(file.filename)
|
||
file.save(os.path.join(MEDIA_DIR, screen, filename))
|
||
|
||
return redirect("/admin")
|
||
|
||
# -------------------------------------------------
|
||
# Admin: Datei löschen
|
||
# -------------------------------------------------
|
||
|
||
@app.route("/admin/delete/<screen>/<filename>", methods=["POST"])
|
||
@login_required
|
||
def delete_file(screen, filename):
|
||
cfg = load_config()
|
||
|
||
# Datei löschen
|
||
path = os.path.join(MEDIA_DIR, screen, filename)
|
||
if os.path.exists(path):
|
||
os.remove(path)
|
||
|
||
# --- Playlist bereinigen ---
|
||
if screen == "priority":
|
||
playlist = cfg.get("priority", {}).get("playlist", [])
|
||
if filename in playlist:
|
||
playlist.remove(filename)
|
||
else:
|
||
playlist = cfg["screens"].get(screen, {}).get("playlist", [])
|
||
if filename in playlist:
|
||
playlist.remove(filename)
|
||
|
||
save_config(cfg)
|
||
return redirect("/admin")
|
||
|
||
|
||
# -------------------------------------------------
|
||
# Admin: Playlist speichern
|
||
# -------------------------------------------------
|
||
|
||
@app.route("/admin/playlist/<screen>", methods=["POST"])
|
||
@login_required
|
||
def save_playlist(screen):
|
||
cfg = load_config()
|
||
data = request.get_json()
|
||
|
||
if screen == "priority":
|
||
cfg["priority"]["playlist"] = data["playlist"]
|
||
else:
|
||
cfg["screens"][screen]["playlist"] = data["playlist"]
|
||
|
||
save_config(cfg)
|
||
return "", 204
|
||
|
||
|
||
# -------------------------------------------------
|
||
# Main
|
||
# -------------------------------------------------
|
||
if __name__ == "__main__":
|
||
app.run(debug=True, host="0.0.0.0", port=5005) |