mirror of
https://github.com/plexguide/Huntarr.git
synced 2026-04-26 13:38:23 -05:00
Admin9705
82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
from flask import Flask, render_template, request, redirect, jsonify
|
|
import os
|
|
import json
|
|
|
|
# Import the necessary function
|
|
from src.primary.stateful_manager import reset_stateful_management, get_stateful_management_info
|
|
|
|
# Configure Flask to use templates and static files from the frontend folder
|
|
template_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'frontend', 'templates'))
|
|
static_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', 'frontend', 'static'))
|
|
|
|
app = Flask(__name__, template_folder=template_dir, static_folder=static_dir)
|
|
|
|
# API Routes
|
|
|
|
@app.route('/api/stateful/reset', methods=['POST'])
|
|
def api_reset_stateful():
|
|
"""API endpoint to reset the stateful management system."""
|
|
success = reset_stateful_management()
|
|
if success:
|
|
return jsonify({"success": True, "message": "Stateful management reset successfully."}), 200
|
|
else:
|
|
return jsonify({"success": False, "message": "Failed to reset stateful management."}), 500
|
|
|
|
@app.route('/api/stateful/info', methods=['GET'])
|
|
def api_get_stateful_info():
|
|
"""API endpoint to get stateful management info."""
|
|
try:
|
|
info = get_stateful_management_info()
|
|
return jsonify(info), 200
|
|
except Exception as e:
|
|
# Log the exception details if possible
|
|
app.logger.error(f"Error getting stateful info: {e}")
|
|
return jsonify({"error": "Failed to retrieve stateful information."}), 500
|
|
|
|
def get_ui_preference():
|
|
"""Determine which UI to use based on config and user preference"""
|
|
# Check if ui_settings.json exists
|
|
config_file = os.path.join(os.path.dirname(__file__), 'config/ui_settings.json')
|
|
|
|
use_new_ui = False
|
|
|
|
if os.path.exists(config_file):
|
|
try:
|
|
with open(config_file, 'r') as f:
|
|
settings = json.load(f)
|
|
use_new_ui = settings.get('use_new_ui', False)
|
|
except Exception as e:
|
|
print(f"Error loading UI settings: {e}")
|
|
|
|
# Allow URL parameter to override
|
|
ui_param = request.args.get('ui', None)
|
|
if ui_param == 'new':
|
|
use_new_ui = True
|
|
elif ui_param == 'classic':
|
|
use_new_ui = False
|
|
|
|
return use_new_ui
|
|
|
|
@app.route('/')
|
|
def index():
|
|
"""Root route with UI switching capability"""
|
|
if get_ui_preference():
|
|
return redirect('/new')
|
|
else:
|
|
return render_template('index.html')
|
|
|
|
@app.route('/user')
|
|
def user_page():
|
|
"""User settings page with UI switching capability"""
|
|
if get_ui_preference():
|
|
return redirect('/user/new')
|
|
else:
|
|
return render_template('user.html')
|
|
|
|
@app.route('/user/new')
|
|
def user_page_new():
|
|
"""Serve the new user settings page"""
|
|
return render_template('user-new.html')
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True) |