def create():
company_id = session["user"]["company_id"]
rq = request.get_json(force=True)
vehicle = Vehicle.parse(rq, "create")
if "errors" in vehicle:
return jsonify(errors=vehicle["errors"]), 400
vehicle = vehicle["vehicle"]
if db.is_existing(table="vehicles", conditions={"registration": vehicle["registration"], "company_id": company_id}):
return jsonify(info="Vehicle with the same registration already exist"), 400
vehicle_data = {
"registration": vehicle["registration"],
"type": vehicle["type"],
"max_weight": vehicle["max_weight"],
"max_area": vehicle["max_area"],
"company_id": company_id,
}
vehicle_id = db.insert(table="vehicles", params=vehicle_data)
return jsonify(info="Vehicle created successfully", vehicleId=vehicle_id)
def update(id: int):
company_id = session["user"]["company_id"]
if not db.is_existing(table="vehicles", conditions={"id": id, "company_id": company_id}):
return jsonify(info="Vehicle not found"), 404
rq = request.get_json(force=True)
vehicle = Vehicle.parse(rq, "update")
if "errors" in vehicle:
return jsonify(errors=vehicle["errors"]), 400
vehicle = vehicle["vehicle"]
vehicle_data = {}
if "registration" in vehicle:
if db.is_existing(
table="vehicles", conditions={"registration": vehicle["registration"], "company_id": company_id}
):
return jsonify(info="Vehicle with the same registration already exist"), 400
vehicle_data["registration"] = vehicle["registration"]
if "type" in vehicle:
vehicle_data["type"] = vehicle["type"]
if "max_weight" in vehicle:
max_weight = vehicle["max_weight"]
if not (isinstance(max_weight, float) or isinstance(max_weight, int)) or max_weight > 100000 or max_weight < 0:
abort(400)
vehicle_data["max_weight"] = vehicle["max_weight"]
if "max_area" in vehicle:
max_area = vehicle["max_area"]
if not (isinstance(max_area, float) or isinstance(max_area, int)) or max_area > 150 or max_area < 0:
abort(400)
vehicle_data["max_area"] = vehicle["max_area"]
db.update(table="vehicles", params=vehicle_data, conditions={"id": id})
return jsonify(info="Vehicle updated successfully"), 200
