123 lines
No EOL
4.8 KiB
Python
123 lines
No EOL
4.8 KiB
Python
from flask import render_template, request, jsonify
|
|
|
|
from . import main
|
|
from .helpers import ACTIVE_STATUSES, user_headers, inventory_headers, worklog_headers
|
|
from .. import db
|
|
from ..utils.load import eager_load_user_relationships, eager_load_room_relationships, eager_load_inventory_relationships, eager_load_worklog_relationships
|
|
from ..models import User, Room, Inventory, WorkLog
|
|
|
|
@main.route("/users")
|
|
def list_users():
|
|
query = eager_load_user_relationships(db.session.query(User)).order_by(User.last_name, User.first_name)
|
|
users = query.all()
|
|
return render_template(
|
|
'table.html',
|
|
header = user_headers,
|
|
rows = [{"id": user.id, "cells": [fn(user) for fn in user_headers.values()]} for user in users],
|
|
title = "Users",
|
|
entry_route = 'user'
|
|
)
|
|
|
|
@main.route("/user/<id>")
|
|
def user(id):
|
|
try:
|
|
id = int(id)
|
|
except ValueError:
|
|
return render_template('error.html', title='Bad ID', message='ID must be an integer.', endpoint='user', endpoint_args={'id': -1})
|
|
|
|
users_query = db.session.query(User).order_by(User.first_name, User.last_name)
|
|
users = eager_load_user_relationships(users_query).all()
|
|
user = next((u for u in users if u.id == id), None)
|
|
rooms_query = db.session.query(Room)
|
|
rooms = eager_load_room_relationships(rooms_query).all()
|
|
inventory_query = (
|
|
eager_load_inventory_relationships(db.session.query(Inventory))
|
|
.filter(Inventory.owner_id == id) # type: ignore
|
|
.filter(Inventory.condition.in_(ACTIVE_STATUSES))
|
|
)
|
|
|
|
inventory = inventory_query.all()
|
|
filtered_inventory_headers = {k: v for k, v in inventory_headers.items() if k not in ['Date Entered', 'Inventory #', 'Serial #',
|
|
'Bar Code #', 'Condition', 'Owner', 'Notes',
|
|
'Brand', 'Model', 'Shared?', 'Location']}
|
|
worklog_query = eager_load_worklog_relationships(db.session.query(WorkLog)).filter(WorkLog.contact_id == id)
|
|
worklog = worklog_query.order_by(WorkLog.start_time.desc()).all()
|
|
filtered_worklog_headers = {k: v for k, v in worklog_headers.items() if k not in ['Contact', 'Follow Up?', 'Quick Analysis?']}
|
|
|
|
if user:
|
|
title = f"User Record - {user.full_name}" if user.active else f"User Record - {user.full_name} (Inactive)"
|
|
else:
|
|
title = f"User Record - User Not Found"
|
|
return render_template(
|
|
'error.html',
|
|
title=title,
|
|
message=f"User with id {id} not found!"
|
|
)
|
|
|
|
return render_template(
|
|
"user.html",
|
|
title=title,
|
|
user=user, users=users, rooms=rooms, assets=inventory,
|
|
inventory_headers=filtered_inventory_headers,
|
|
inventory_rows=[{"id": item.id, "cells": [fn(item) for fn in filtered_inventory_headers.values()]} for item in inventory],
|
|
worklog=worklog,
|
|
worklog_headers=filtered_worklog_headers,
|
|
worklog_rows=[{"id": log.id, "cells": [fn(log) for fn in filtered_worklog_headers.values()]} for log in worklog]
|
|
)
|
|
|
|
@main.route("/user/new", methods=["GET"])
|
|
def new_user():
|
|
rooms = eager_load_room_relationships(db.session.query(Room)).all()
|
|
users = eager_load_user_relationships(db.session.query(User)).all()
|
|
|
|
user = User(
|
|
active=True
|
|
)
|
|
|
|
return render_template(
|
|
"user.html",
|
|
title="New User",
|
|
user=user,
|
|
users=users,
|
|
rooms=rooms
|
|
)
|
|
|
|
@main.route("/api/user", methods=["POST"])
|
|
def create_user():
|
|
try:
|
|
data = request.get_json(force=True)
|
|
|
|
new_user = User.from_dict(data)
|
|
|
|
db.session.add(new_user)
|
|
db.session.commit()
|
|
|
|
return jsonify({"success": True, "id": new_user.id}), 201
|
|
|
|
except Exception as e:
|
|
db.session.rollback()
|
|
return jsonify({"success": False, "error": str(e)}), 400
|
|
|
|
@main.route("/api/user/<int:id>", methods=["PUT"])
|
|
def update_user(id):
|
|
try:
|
|
data = request.get_json(force=True)
|
|
user = db.session.query(User).get(id)
|
|
|
|
if not user:
|
|
return jsonify({"success": False, "error": f"User with ID {id} not found."}), 404
|
|
|
|
user.staff = bool(data.get("staff", user.staff))
|
|
user.active = bool(data.get("active", user.active))
|
|
user.last_name = data.get("last_name", user.last_name)
|
|
user.first_name = data.get("first_name", user.first_name)
|
|
user.location_id = data.get("location_id", user.location_id)
|
|
user.supervisor_id = data.get("supervisor_id", user.supervisor_id)
|
|
|
|
db.session.commit()
|
|
|
|
return jsonify({"success": True, "id": user.id}), 200
|
|
|
|
except Exception as e:
|
|
db.session.rollback()
|
|
return jsonify({"success": False, "error": str(e)}), 400 |