141 lines
5.3 KiB
Python
141 lines
5.3 KiB
Python
"""Inspection blueprint."""
|
|
|
|
from flask import Blueprint, render_template, redirect, url_for, flash, request
|
|
from flask_login import login_required, current_user
|
|
from app import db
|
|
from app.models import Inspection, InspectionInspector, Photo
|
|
from werkzeug.utils import secure_filename
|
|
import os
|
|
from uuid import uuid4
|
|
from app.forms import InspectionForm
|
|
|
|
from flask import current_app
|
|
|
|
inspections_bp = Blueprint("inspections", __name__)
|
|
|
|
|
|
@inspections_bp.route("", methods=["GET"])
|
|
@login_required
|
|
def index():
|
|
"""List all inspections for the logged-in user."""
|
|
inspections = (
|
|
Inspection.query.filter_by(created_by=current_user.id)
|
|
.order_by(Inspection.inspection_date.desc())
|
|
.all()
|
|
)
|
|
return render_template("dashboard.html", inspections=inspections)
|
|
|
|
|
|
@inspections_bp.route("/new", methods=["GET", "POST"])
|
|
@login_required
|
|
def create():
|
|
"""Create a new inspection."""
|
|
form = InspectionForm()
|
|
if form.validate_on_submit():
|
|
inspection = Inspection(
|
|
installation_name=form.installation_name.data,
|
|
location=form.location.data,
|
|
inspection_date=form.inspection_date.data,
|
|
reference_number=form.reference_number.data,
|
|
observations=form.observations.data,
|
|
conclusion_text=form.conclusion_text.data,
|
|
conclusion_status=form.conclusion_status.data,
|
|
created_by=current_user.id,
|
|
)
|
|
# Add inspectors
|
|
for inspector in form.inspectors.data:
|
|
if inspector.get("user_id"):
|
|
inspector_obj = InspectionInspector(
|
|
inspection_id=inspection.id,
|
|
user_id=inspector["user_id"],
|
|
free_text_name=inspector.get("free_text_name"),
|
|
)
|
|
db.session.add(inspector_obj)
|
|
else:
|
|
inspector_obj = InspectionInspector(
|
|
inspection_id=inspection.id,
|
|
free_text_name=inspector.get("free_text_name"),
|
|
)
|
|
db.session.add(inspector_obj)
|
|
db.session.commit()
|
|
# Handle photo uploads
|
|
for photo_file in form.photos.data:
|
|
filename = secure_filename(photo_file.filename)
|
|
unique_filename = f"{uuid4().hex}_{filename}"
|
|
upload_dir = os.path.join(current_app.instance_path, "uploads")
|
|
os.makedirs(upload_dir, exist_ok=True)
|
|
file_path = os.path.join(upload_dir, unique_filename)
|
|
photo_file.save(file_path)
|
|
photo = Photo(
|
|
inspection_id=inspection.id,
|
|
filename=unique_filename,
|
|
caption="",
|
|
action_required="none",
|
|
)
|
|
db.session.add(photo)
|
|
flash("Inspection created successfully.", "success")
|
|
return redirect(url_for("inspections.view", inspection_id=inspection.id))
|
|
return render_template("inspection_form.html", form=form)
|
|
|
|
|
|
@inspections_bp.route("/<int:inspection_id>", methods=["GET"])
|
|
@login_required
|
|
def view(inspection_id):
|
|
"""View an inspection."""
|
|
inspection = Inspection.query.get_or_404(inspection_id)
|
|
inspectors = inspection.inspectors
|
|
photos = inspection.photos
|
|
return render_template(
|
|
"inspection_view.html",
|
|
inspection=inspection,
|
|
inspectors=inspectors,
|
|
photos=photos,
|
|
)
|
|
|
|
|
|
@inspections_bp.route("/<int:inspection_id>/edit", methods=["GET", "POST"])
|
|
@login_required
|
|
def edit(inspection_id):
|
|
"""Edit an inspection."""
|
|
inspection = Inspection.query.get_or_404(inspection_id)
|
|
form = InspectionForm(obj=inspection)
|
|
if form.validate_on_submit():
|
|
inspection.installation_name = form.installation_name.data
|
|
inspection.location = form.location.data
|
|
inspection.inspection_date = form.inspection_date.data
|
|
inspection.reference_number = form.reference_number.data
|
|
inspection.observations = form.observations.data
|
|
inspection.conclusion_text = form.conclusion_text.data
|
|
inspection.conclusion_status = form.conclusion_status.data
|
|
# Update inspectors
|
|
# Simplified handling for brevity
|
|
db.session.commit()
|
|
# Handle additional photo uploads
|
|
for photo_file in form.photos.data:
|
|
filename = secure_filename(photo_file.filename)
|
|
unique_filename = f"{uuid4().hex}_{filename}"
|
|
upload_dir = os.path.join(current_app.instance_path, "uploads")
|
|
os.makedirs(upload_dir, exist_ok=True)
|
|
file_path = os.path.join(upload_dir, unique_filename)
|
|
photo_file.save(file_path)
|
|
photo = Photo(
|
|
inspection_id=inspection.id,
|
|
filename=unique_filename,
|
|
caption="",
|
|
action_required="none",
|
|
)
|
|
db.session.add(photo)
|
|
flash("Inspection updated successfully.", "success")
|
|
return redirect(url_for("inspections.view", inspection_id=inspection.id))
|
|
return render_template("inspection_form.html", form=form)
|
|
|
|
|
|
@inspections_bp.route("/<int:inspection_id>/delete", methods=["POST"])
|
|
@login_required
|
|
def delete(inspection_id):
|
|
"""Delete an inspection."""
|
|
inspection = Inspection.query.get_or_404(inspection_id)
|
|
db.session.delete(inspection)
|
|
db.session.commit()
|
|
flash("Inspection deleted successfully.", "info")
|
|
return redirect(url_for("inspections.index"))
|