mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-11-01 10:22:08 +00:00
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
"""Read-only PBS status and administrative connection management."""
|
|
|
|
from flask import Blueprint, jsonify, request
|
|
import auth_manager
|
|
import pbs_manager as pbs
|
|
from jwt_middleware import require_auth, require_admin_scope
|
|
|
|
pbs_bp = Blueprint("pbs", __name__)
|
|
|
|
|
|
@pbs_bp.after_request
|
|
def no_cache(response):
|
|
response.headers["Cache-Control"] = "no-store"
|
|
return response
|
|
|
|
|
|
@pbs_bp.route("/api/platform")
|
|
@require_auth
|
|
def platform():
|
|
return jsonify(pbs.get_platform())
|
|
|
|
|
|
@pbs_bp.route("/api/pbs/servers")
|
|
@require_auth
|
|
def servers():
|
|
try:
|
|
return jsonify(servers=pbs.public_servers(), local_available=pbs.is_pbs_host())
|
|
except pbs.PBSError as exc:
|
|
return jsonify(error=str(exc)), 500
|
|
|
|
|
|
def _protected_configuration():
|
|
local = auth_manager.load_auth_config()
|
|
return local.get("enabled") and not local.get("declined") and local.get("password_hash")
|
|
|
|
|
|
@pbs_bp.route("/api/pbs/servers", methods=["POST"])
|
|
@require_admin_scope
|
|
def save_server():
|
|
if not _protected_configuration():
|
|
return jsonify(error="Enable local authentication before saving remote PBS credentials."), 409
|
|
try:
|
|
identifier = pbs.save_server(request.get_json(silent=True))
|
|
return jsonify(success=True, id=identifier, servers=pbs.public_servers())
|
|
except pbs.PBSError as exc:
|
|
return jsonify(error=str(exc)), 400
|
|
except OSError:
|
|
return jsonify(error="Could not save PBS connection."), 500
|
|
|
|
|
|
@pbs_bp.route("/api/pbs/servers/<identifier>", methods=["DELETE"])
|
|
@require_admin_scope
|
|
def delete_server(identifier):
|
|
if not _protected_configuration():
|
|
return jsonify(error="Enable local authentication before changing PBS connections."), 409
|
|
try:
|
|
pbs.delete_server(identifier)
|
|
return jsonify(success=True)
|
|
except pbs.PBSError as exc:
|
|
return jsonify(error=str(exc)), 400
|
|
except OSError:
|
|
return jsonify(error="Could not remove PBS connection."), 500
|
|
|
|
|
|
@pbs_bp.route("/api/pbs/status")
|
|
@require_auth
|
|
def status():
|
|
try:
|
|
return jsonify(pbs.get_status(request.args.get("server", "local")))
|
|
except pbs.PBSError as exc:
|
|
return jsonify(error=str(exc)), 502
|