def delete_vehicle(self, vehicle):
self.logger.debug("delete_vehicle")
self.logger.info("deleting vehicle from database")
Vehicle.delete().where(Vehicle.number == vehicle.number).execute()
self.logger.info("removing vehicle from model")
tree_iter = self.get_iter_from_selected_row(vehicle.number)
self.vehicles_model.remove(tree_iter)
def vehicles(id=None):
if request.method == 'GET':
if id is not None:
vehicle = Vehicle.query.get(id)
if not vehicle:
return jsonify({"fail": "Vehicle not found"}), 404
return jsonify({
"success": "Vehicle found",
"vehicle": vehicle.serialize()
}), 200
else:
vehicles = Vehicle.query.all()
vehicles = list(map(lambda vehicle: vehicle.serialize(), vehicles))
return jsonify({
"total": len(vehicles),
"results": vehicles
}), 200
if request.method == 'POST':
name = request.json.get('name')
model = request.json.get('model')
vehicle_class = request.json.get('vehicle_class')
crew = request.json.get('crew')
manufacturer = request.json.get('manufacturer')
cargo_capacity = request.json.get('cargo_capacity')
cost_in_credits = request.json.get('cost_in_credits')
consumables = request.json.get('consumables')
vehicle = Vehicle()
vehicle.name = name
vehicle.model = model
vehicle.vehicle_class = vehicle_class
vehicle.crew = crew
vehicle.manufacturer = manufacturer
vehicle.cargo_capacity = cargo_capacity
vehicle.cost_in_credits = cost_in_credits
vehicle.consumables = consumables
vehicle.save()
return jsonify({
"success": "vehicle created!",
"vehicle": vehicle.serialize()
}), 201
if request.method == 'PUT':
pass
if request.method == 'DELETE':
vehicle = Vehicle.query.get(id)
if not vehicle:
return jsonify({"fail": "vehicle not found"}), 404
vehicle.delete()
return jsonify({"success": "vehicle deleted"}), 200
def create_neo_db():
# Create temporary station and vehicle
tmp_station = Station(short='STATI',
name='Station',
latitude=10.0,
longitude=10.0).save()
tmp_vehicle = Vehicle(code='100').save()
spatial_query = "CALL spatial.addPointLayer('spati')"
neo_db.cypher_query(query=spatial_query)
tmp_station.delete()
tmp_vehicle.delete()
def register_customer():
if request.method == 'POST':
token = request.headers.get('token')
if not token:
return Response(json.dumps(
{"Message": "Please supply proper credentials"}),
status=400,
content_type="application/json")
if not check_auth(token):
return Response(json.dumps({"Message": "Unauthorized access"}),
status=401,
content_type="application/json")
json_request = json.loads(request.data)
c = Customer()
try:
c.first_name = json_request['first_name']
c.last_name = json_request['last_name']
c.address = json_request['address']
c.contact_no = json_request['contact_no']
c.driving_licence_link = json_request['driving_licence_link']
c.email = json_request['email']
c.password = json_request['password']
vehicles = json_request['vehicles']
except:
return Response(json.dumps({"Message": "In-Complete form data"}),
status=400,
content_type="application/json")
try:
if not (int(c.contact_no) > 1000000000
and int(c.contact_no) <= 9999999999):
return Response(json.dumps(
{"Message": "Contact no. not valid"}),
status=400,
content_type="application/json")
except:
return Response(json.dumps({"Message": "Contact no. not valid"}),
status=400,
content_type="application/json")
for vehicle in vehicles:
v = Vehicle()
try:
v.vehicle_type = vehicle['vehicle_type']
v.vehicle_number = vehicle['vehicle_number']
v.vehicle_rc_link = vehicle['vehicle_rc_link']
except:
return Response(json.dumps(
{"Message": "In-Complete form data"}),
status=400,
content_type="application/json")
v.save()
c.vehicles.append(v)
try:
c.save()
except NotUniqueError:
for vehicle in c.vehicles:
v = Vehicle.objects.get(vid=vehicle.vid)
v.delete()
return Response(json.dumps(
{"Message": "Customer with same e-mail already exists"}),
status=400,
content_type="application/json")
ewallet_reg_url = "http://0.0.0.0:8000/login?x=1&name=" + str(
c.cid) + "&password="******"QR_CODE_DATA": c.QR_CODE_DATA}),
status=200,
content_type="application/json")
