rollerverbrauch/app/routes/service.py

from flask import url_for, redirect, render_template
from flask_security import login_required, current_user
from datetime import date

from ..entities import Vehicle, Service
from ..forms import (
    CreateServiceForm,
    DeleteServiceForm,
    EditServiceForm,
    SelectVehicleForm,
)
from ..tools import (
    db_log_update,
    db_log_delete,
    get_event_line_for_vehicle,
    get_latest_pitstop_for_vehicle,
    get_users_active_vehicle,
)
from .. import app, db


@app.route("/service/vehicle/<int:vid>/create", methods=["GET", "POST"])
@login_required
def create_service_for_vehicle(vid):
    vehicle = Vehicle.query.get(vid)
    if vehicle is None or vehicle not in current_user.vehicles:
        return redirect(url_for("get_account_page"))

    form = CreateServiceForm()

    data = get_event_line_for_vehicle(vehicle)
    if len(data) > 0:
        form.set_pitstops(data)
        form.same_odometer_allowed = type(data[-1]) != Service
    else:
        form.set_pitstops([])
        form.same_odometer_allowed = True

    form.preinit_with_data()

    if form.validate_on_submit():
        new_service = Service(
            form.date.data,
            form.odometer.data,
            vid,
            form.costs.data,
            form.description.data,
        )
        db.session.add(new_service)
        vehicle.services.append(new_service)
        db.session.commit()
        return redirect(url_for("get_pit_stops", _anchor="v" + str(vehicle.id)))

    form.process()
    return render_template(
        "createServiceForm.html", form=form, vehicle=vehicle, messages=[]
    )


@app.route("/service/delete/<int:sid>", methods=["GET", "POST"])
@login_required
def delete_service_form(sid):
    service = Service.query.filter(Service.id == sid).first()
    if service is None:
        return redirect(url_for("get_pit_stops"))

    vehicle = Vehicle.query.filter(Vehicle.id == service.vehicle_id).first()
    if vehicle not in current_user.vehicles:
        return redirect(url_for("get_pit_stops"))

    form = DeleteServiceForm()
    if form.validate_on_submit():
        db.session.delete(service)
        db.session.commit()
        db_log_delete(service)
        return redirect(url_for("get_pit_stops", _anchor="v" + str(vehicle.id)))

    return render_template("deleteServiceForm.html", form=form, service=service)


@app.route("/service/edit/<int:sid>", methods=["GET", "POST"])
@login_required
def edit_service_form(sid):
    edit_service = Service.query.get(sid)
    if edit_service is None:
        return redirect(url_for("get_pit_stops"))

    vehicle = Vehicle.query.filter(Vehicle.id == edit_service.vehicle_id).first()
    if vehicle not in current_user.vehicles:
        return redirect(url_for("get_pit_stops"))

    data = get_event_line_for_vehicle(vehicle)
    data = [x for x in data if x != edit_service]
    form = EditServiceForm()
    form.same_odometer_allowed = True
    form.set_pitstops(data)

    if not form.is_submitted():
        form.odometer.default = edit_service.odometer
        form.description.default = edit_service.description
        form.date.default = edit_service.date
        form.costs.default = edit_service.costs
    if form.validate_on_submit():
        edit_service.costs = form.costs.data
        edit_service.date = form.date.data
        edit_service.description = form.description.data
        edit_service.odometer = form.odometer.data
        db.session.commit()
        db_log_update(edit_service)
        return redirect(url_for("get_pit_stops", _anchor="v" + str(vehicle.id)))

    form.preinit_with_data()
    form.process()
    return render_template(
        "editServiceForm.html",
        form=form,
        vehicle=vehicle,
        messages=form.get_hint_messages(),
    )


@app.route("/service/vehicle/select", methods=["GET", "POST"])
@login_required
def select_vehicle_for_new_service():
    active_vehicles = get_users_active_vehicle(current_user)
    if len(active_vehicles) == 1:
        return redirect(
            url_for("create_service_for_vehicle", vid=active_vehicles[0].id)
        )

    form = SelectVehicleForm()
    form.vehicle.choices = [(g.id, g.name) for g in active_vehicles]

    if form.validate_on_submit():
        return redirect(url_for("create_service_for_vehicle", vid=form.vehicle.data))

    return render_template("selectVehicle.html", form=form)