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")