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