99 lines
3.8 KiB
Python
99 lines
3.8 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, Pitstop
|
||
|
|
from ..forms import CreateServiceForm, DeleteServiceForm, EditServiceForm
|
||
|
|
from ..tools import db_log_update, db_log_delete, db_log_add, 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()
|
||
|
|
last_pitstop = get_latest_pitstop_for_vehicle(vid)
|
||
|
|
form.set_pitstop(last_pitstop)
|
||
|
|
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()
|
||
|
|
print(new_service)
|
||
|
|
return redirect(url_for('get_account_page'))
|
||
|
|
|
||
|
|
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'))
|
||
|
|
|
||
|
|
# last_pitstop_pos = vehicle.pitstops.index(edit_service) - 1
|
||
|
|
# if last_pitstop_pos > 0:
|
||
|
|
# last_pitstop = vehicle.pitstops[last_pitstop_pos]
|
||
|
|
# else:
|
||
|
|
# last_pitstop = Pitstop(0, 0, date(1970, 1, 1), 0, 0)
|
||
|
|
last_pitstop = Pitstop(0, 0, date(1970, 1, 1), 0, 0)
|
||
|
|
|
||
|
|
form = EditServiceForm()
|
||
|
|
form.set_pitstop(last_pitstop)
|
||
|
|
|
||
|
|
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.odometer.default = edit_service.odometer
|
||
|
|
form.description.default = edit_service.description
|
||
|
|
form.date.default = edit_service.date
|
||
|
|
form.costs.default = edit_service.costs
|
||
|
|
form.process()
|
||
|
|
messages = {
|
||
|
|
# 'date': 'Date must be between %s and %s (including).' % (str(last_pitstop.date), str(date.today())),
|
||
|
|
# 'odometer': 'Odometer must be greater than %s km.' % (str(last_pitstop.odometer))
|
||
|
|
}
|
||
|
|
if edit_service.costs is not None and edit_service.costs > 0:
|
||
|
|
messages['costs'] = 'Costs must be higher than 0.01 €.'
|
||
|
|
return render_template('editServiceForm.html', form=form, vehicle=vehicle, messages=messages)
|