205 lines
8.6 KiB
Python
205 lines
8.6 KiB
Python
"""
|
||
WMT management web routes – global settings, device registry, update requests.
|
||
"""
|
||
from flask import Blueprint, render_template, request, redirect, url_for, flash
|
||
from datetime import datetime
|
||
from app.models import WMTGlobalConfig, Device, WMTUpdateRequest
|
||
from config.database_config import get_db
|
||
import logging
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
wmt_web_bp = Blueprint('wmt_web', __name__, url_prefix='/wmt')
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Helpers
|
||
# ---------------------------------------------------------------------------
|
||
|
||
def _get_or_create_global_config(session):
|
||
cfg = session.query(WMTGlobalConfig).first()
|
||
if cfg is None:
|
||
cfg = WMTGlobalConfig()
|
||
session.add(cfg)
|
||
session.flush()
|
||
return cfg
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Dashboard
|
||
# ---------------------------------------------------------------------------
|
||
|
||
@wmt_web_bp.route('/')
|
||
def index():
|
||
"""WMT management dashboard."""
|
||
try:
|
||
with get_db().get_session() as session:
|
||
global_cfg = _get_or_create_global_config(session)
|
||
devices = session.query(Device).filter(Device.mac_address.isnot(None)).order_by(Device.nume_masa).all()
|
||
pending_count = session.query(WMTUpdateRequest).filter_by(status='pending').count()
|
||
recent_requests = (
|
||
session.query(WMTUpdateRequest)
|
||
.order_by(WMTUpdateRequest.submitted_at.desc())
|
||
.limit(5)
|
||
.all()
|
||
)
|
||
return render_template(
|
||
'wmt/index.html',
|
||
global_cfg=global_cfg,
|
||
devices=devices,
|
||
pending_count=pending_count,
|
||
recent_requests=recent_requests,
|
||
breadcrumbs=[{'url': url_for('wmt_web.index'), 'title': 'WMT Management'}],
|
||
)
|
||
except Exception as e:
|
||
logger.error(f'WMT dashboard error: {e}')
|
||
flash(f'Error loading dashboard: {e}', 'error')
|
||
return render_template('wmt/index.html', global_cfg=None, devices=[],
|
||
pending_count=0, recent_requests=[])
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Global settings
|
||
# ---------------------------------------------------------------------------
|
||
|
||
@wmt_web_bp.route('/settings', methods=['GET', 'POST'])
|
||
def settings():
|
||
"""View and edit global WMT configuration."""
|
||
try:
|
||
with get_db().get_session() as session:
|
||
cfg = _get_or_create_global_config(session)
|
||
|
||
if request.method == 'POST':
|
||
cfg.chrome_url = request.form.get('chrome_url', '').strip()
|
||
cfg.chrome_local_url = request.form.get('chrome_local_url', '').strip() or None
|
||
cfg.chrome_insecure_origin = request.form.get('chrome_insecure_origin', '').strip()
|
||
cfg.card_api_base_url = request.form.get('card_api_base_url', '').strip()
|
||
cfg.server_log_url = request.form.get('server_log_url', '').strip()
|
||
cfg.internet_check_host = request.form.get('internet_check_host', '').strip()
|
||
cfg.update_host = request.form.get('update_host', '').strip()
|
||
cfg.update_user = request.form.get('update_user', '').strip()
|
||
cfg.notes = request.form.get('notes', '').strip() or None
|
||
cfg.updated_at = datetime.utcnow()
|
||
cfg.updated_by = 'admin'
|
||
flash('Global settings saved.', 'success')
|
||
return redirect(url_for('wmt_web.settings'))
|
||
|
||
return render_template(
|
||
'wmt/settings.html',
|
||
cfg=cfg,
|
||
breadcrumbs=[
|
||
{'url': url_for('wmt_web.index'), 'title': 'WMT Management'},
|
||
{'url': url_for('wmt_web.settings'), 'title': 'Global Settings'},
|
||
],
|
||
)
|
||
except Exception as e:
|
||
logger.error(f'WMT settings error: {e}')
|
||
flash(f'Error: {e}', 'error')
|
||
return redirect(url_for('wmt_web.index'))
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Update requests
|
||
# ---------------------------------------------------------------------------
|
||
|
||
@wmt_web_bp.route('/requests')
|
||
def update_requests():
|
||
"""List all device update requests."""
|
||
status_filter = request.args.get('status', 'pending')
|
||
try:
|
||
with get_db().get_session() as session:
|
||
query = session.query(WMTUpdateRequest)
|
||
if status_filter != 'all':
|
||
query = query.filter_by(status=status_filter)
|
||
req_list = query.order_by(WMTUpdateRequest.submitted_at.desc()).all()
|
||
pending_count = session.query(WMTUpdateRequest).filter_by(status='pending').count()
|
||
|
||
return render_template(
|
||
'wmt/requests.html',
|
||
requests=req_list,
|
||
status_filter=status_filter,
|
||
pending_count=pending_count,
|
||
breadcrumbs=[
|
||
{'url': url_for('wmt_web.index'), 'title': 'WMT Management'},
|
||
{'url': url_for('wmt_web.update_requests'), 'title': 'Update Requests'},
|
||
],
|
||
)
|
||
except Exception as e:
|
||
logger.error(f'WMT requests list error: {e}')
|
||
flash(f'Error: {e}', 'error')
|
||
return redirect(url_for('wmt_web.index'))
|
||
|
||
|
||
@wmt_web_bp.route('/requests/<int:req_id>/accept', methods=['POST'])
|
||
def accept_request(req_id):
|
||
"""Accept an update request: apply proposed values to WMTDevice."""
|
||
try:
|
||
with get_db().get_session() as session:
|
||
req = session.query(WMTUpdateRequest).filter_by(id=req_id).first()
|
||
if not req:
|
||
flash('Request not found.', 'error')
|
||
return redirect(url_for('wmt_web.update_requests'))
|
||
|
||
# Find or create the Device
|
||
device = session.query(Device).filter_by(mac_address=req.mac_address).first()
|
||
if device is None:
|
||
device = Device(
|
||
mac_address=req.mac_address,
|
||
hostname=req.proposed_hostname or '',
|
||
device_ip=req.proposed_device_ip or '',
|
||
nume_masa=req.proposed_device_name or '',
|
||
)
|
||
session.add(device)
|
||
session.flush()
|
||
req.device_id = device.id
|
||
|
||
# Apply proposed values
|
||
if req.proposed_device_name is not None:
|
||
device.nume_masa = req.proposed_device_name
|
||
if req.proposed_hostname is not None:
|
||
device.hostname = req.proposed_hostname
|
||
if req.proposed_device_ip is not None:
|
||
device.device_ip = req.proposed_device_ip
|
||
device.config_updated_at = datetime.utcnow()
|
||
device.info_reviewed_at = datetime.utcnow() # admin reviewed → push timestamp to devices
|
||
|
||
# Mark request as accepted
|
||
req.status = 'accepted'
|
||
req.admin_reviewed_at = datetime.utcnow()
|
||
req.admin_notes = request.form.get('admin_notes', '').strip() or None
|
||
|
||
flash('Request accepted and device record updated.', 'success')
|
||
except Exception as e:
|
||
logger.error(f'WMT accept request error: {e}')
|
||
flash(f'Error accepting request: {e}', 'error')
|
||
return redirect(url_for('wmt_web.update_requests'))
|
||
|
||
|
||
@wmt_web_bp.route('/requests/<int:req_id>/reject', methods=['POST'])
|
||
def reject_request(req_id):
|
||
"""Reject an update request (updates reviewed_at so WMT client won't re-submit)."""
|
||
try:
|
||
with get_db().get_session() as session:
|
||
req = session.query(WMTUpdateRequest).filter_by(id=req_id).first()
|
||
if not req:
|
||
flash('Request not found.', 'error')
|
||
return redirect(url_for('wmt_web.update_requests'))
|
||
|
||
req.status = 'rejected'
|
||
req.admin_reviewed_at = datetime.utcnow()
|
||
req.admin_notes = request.form.get('admin_notes', '').strip() or None
|
||
|
||
# Update device info_reviewed_at even though data didn't change –
|
||
# this signals to the WMT client that the server has reviewed the state
|
||
# so it won't keep re-submitting the same request.
|
||
if req.device_id:
|
||
device = session.query(Device).filter_by(id=req.device_id).first()
|
||
if device:
|
||
device.info_reviewed_at = datetime.utcnow()
|
||
|
||
flash('Request rejected.', 'warning')
|
||
except Exception as e:
|
||
logger.error(f'WMT reject request error: {e}')
|
||
flash(f'Error rejecting request: {e}', 'error')
|
||
return redirect(url_for('wmt_web.update_requests'))
|