def put(self, order_id): # Get order by id order = Order.get_by_id(order_id) if order is None: abort(404) # Update order fields args = self.parser.parse_args() order.cleaner_id = assign(args['cleaner_id'], order.cleaner_id) order.address_id = assign(args['address_id'], order.address_id) order.date = assign(args['date'], order.date) order.start_time = assign(args['start_time'], order.start_time) order.end_time = assign(args['end_time'], order.end_time) order.rooms = assign(args['rooms'], order.rooms) order.special_rooms = assign(args['special_rooms'], order.special_rooms) order.extra_services = assign(args['extra_services'], order.extra_services) order.reference = assign(args['reference'], order.reference) order.transaction = assign(args['transaction'], order.transaction) order.price = assign(args['price'], order.price) # Persist changes and return order order.persist() return order
def get(self, order_id): # Get order by id order = Order.get_by_id(order_id) if order is None: abort(404) # Return order return order