示例#1
0
def requestOrder(postBody):
    response = {}
    try:
        order = Order(postBody["serviceType"], postBody["pickupAddress"],
                      postBody["dropoffAddress"], postBody["customerId"])
        orderObj = order.__dict__
        client = mongoConnect()
        db = client.team12_demand
        orders = db.orders
        publicId = orders.count_documents({}) + 1000
        orderObj["publicId"] = publicId
        orderId = orders.insert_one(orderObj).inserted_id
        address1 = (orderObj["pickupAddress"]).replace(" ", "+")
        address2 = (orderObj["dropoffAddress"]).replace(" ", "+")
        #### Uncomment this block when supply is up on server ####
        routeResponse = requests.get(
            f"http://localhost:8081/vehicles/req?service_type={orderObj['serviceType']}&order_id={orderId}&customer_id={orderObj['customerId']}&destination={address1}"
        )
        # routeResponse = requests.get(f"https://supply.team12.sweispring21.tk/api/vehicles/req?service_type={orderObj['serviceType']}&order_id={orderId}&customer_id={orderObj['customerId']}&destination={address1}")
        responseObj = routeResponse.json()
        routeObj = responseObj["data"]
        if responseObj['status'] == "OK":
            order.setStatus("vehicle on route")
            response = {
                'status': responseObj['status'],
                'data': {
                    "_id": orderId,
                    "publicId": publicId,
                    "status": order.getStatus(),
                    "routeObj": routeObj
                }
            }
            dbRouteObj = routeObj
        else:
            response = {
                'status': responseObj['status'],
                'data': responseObj['data']
            }
            order.setStatus("unfulfilled")
            dbRouteObj = routeObj

        orders.update_one(
            {"_id": orderId},
            {"$set": {
                "status": order.getStatus(),
                "routeObj": dbRouteObj
            }})
    except ValueError as err:
        response = {'status': 'CONFLICT', 'data': {'msg': err}}
    except Exception as err:
        logging.error(err)
        response = {
            'status': 'INTERNAL_SERVER_ERROR',
            'data': {
                'msg': 'Server stopped working, please try again later'
            }
        }

    return response
示例#2
0
 def test_valid_status_setter(self):
     order = Order()
     order.setStatus(validStatus1)
     correctValue = validStatus1
     self.assertEqual(
         order.getStatus(),
         correctValue,
         msg=
         f"Error in the email setter and getter values:\n\n{order.getStatus()}\n\n{correctValue}"
     )