Files
shopdb-flask/shopdb/core/api/dashboard.py
cproudlock 1196de6e88 Initial commit: Shop Database Flask Application
Flask backend with Vue 3 frontend for shop floor machine management.
Includes database schema export for MySQL shopdb_flask database.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2026-01-13 16:07:34 -05:00

118 lines
3.6 KiB
Python

"""Dashboard API endpoints."""
from flask import Blueprint
from flask_jwt_extended import jwt_required
from shopdb.extensions import db
from shopdb.core.models import Machine, MachineType, MachineStatus
from shopdb.utils.responses import success_response
dashboard_bp = Blueprint('dashboard', __name__)
@dashboard_bp.route('/summary', methods=['GET'])
@dashboard_bp.route('', methods=['GET'])
@jwt_required(optional=True)
def get_dashboard():
"""Get dashboard summary data."""
# Count machines by category
equipment_count = db.session.query(Machine).join(MachineType).filter(
Machine.isactive == True,
MachineType.category == 'Equipment'
).count()
pc_count = db.session.query(Machine).join(MachineType).filter(
Machine.isactive == True,
MachineType.category == 'PC'
).count()
network_count = db.session.query(Machine).join(MachineType).filter(
Machine.isactive == True,
MachineType.category == 'Network'
).count()
# Count by status
status_counts = db.session.query(
MachineStatus.status,
db.func.count(Machine.machineid)
).outerjoin(
Machine,
db.and_(Machine.statusid == MachineStatus.statusid, Machine.isactive == True)
).group_by(MachineStatus.status).all()
# Recent machines
recent_machines = Machine.query.filter_by(isactive=True).order_by(
Machine.createddate.desc()
).limit(10).all()
# Build status dict
status_dict = {status: count for status, count in status_counts}
return success_response({
# Fields expected by frontend
'totalmachines': equipment_count + pc_count + network_count,
'totalequipment': equipment_count,
'totalpc': pc_count,
'totalnetwork': network_count,
'activemachines': status_dict.get('In Use', 0),
'inrepair': status_dict.get('In Repair', 0),
# Also include structured data
'counts': {
'equipment': equipment_count,
'pcs': pc_count,
'network_devices': network_count,
'total': equipment_count + pc_count + network_count
},
'by_status': status_dict,
'recent': [
{
'machineid': m.machineid,
'machinenumber': m.machinenumber,
'machinetype': m.machinetype.machinetype if m.machinetype else None,
'createddate': m.createddate.isoformat() + 'Z' if m.createddate else None
}
for m in recent_machines
]
})
@dashboard_bp.route('/stats', methods=['GET'])
@jwt_required(optional=True)
def get_stats():
"""Get detailed statistics."""
# Machine type breakdown
type_counts = db.session.query(
MachineType.machinetype,
MachineType.category,
db.func.count(Machine.machineid)
).outerjoin(
Machine,
db.and_(Machine.machinetypeid == MachineType.machinetypeid, Machine.isactive == True)
).filter(MachineType.isactive == True).group_by(
MachineType.machinetypeid
).all()
return success_response({
'by_type': [
{'type': t, 'category': c, 'count': count}
for t, c, count in type_counts
]
})
@dashboard_bp.route('/health', methods=['GET'])
def health_check():
"""Health check endpoint (no auth required)."""
try:
# Test database connection
db.session.execute(db.text('SELECT 1'))
db_status = 'healthy'
except Exception as e:
db_status = f'unhealthy: {str(e)}'
return success_response({
'status': 'ok' if db_status == 'healthy' else 'degraded',
'database': db_status,
'version': '1.0.0'
})