def update_delivery(self, _id, data): from src.auth.models.user_table import DeliveryUserModel from src.auth.schemas.schemas import DeliveryUserSchema delivery_data = self.sqlachemy_to_dict( DeliveryUserModel.get_delivery(_id)) delivery_data.update(data) return DeliveryUserModel.get_delivery(_id).update(delivery_data)
def get_delivery(self, _id): from src.auth.models.user_table import DeliveryUserModel response = DeliveryUserModel.get_delivery(_id) return self.sqlachemy_to_dict(response)
def delete_delivery(self, _id): from src.auth.models.user_table import DeliveryUserModel return DeliveryUserModel.get_delivery(_id).delete()
def get_today_delivery_orders(self, delivery_id): from src.auth.models.order_table import OrderModel from src.auth.models.user_table import DeliveryUserModel DeliveryUserModel.get_delivery(delivery_id) today = datetime.date.today() return OrderModel.query.filter(OrderModel.delivery_id == delivery_id).filter(OrderModel.created_at >= today).filter(OrderModel.state == 'delivered').count()