137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
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,
|
|
)
|
|
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)
|