from datetime import datetime

from app import db
from app.dao.dao_utils import transactional
from app.models import ServiceDataRetention


def get_service_data_retention(service_id):
    data_retention = ServiceDataRetention.query.filter_by(service_id=service_id).all()
    return data_retention


@transactional
def insert_service_data_retention(service_id, notification_type, days_of_retention):
    new_data_retention = ServiceDataRetention(service_id=service_id,
                                              notification_type=notification_type,
                                              days_of_retention=days_of_retention)

    db.session.add(new_data_retention)
    return new_data_retention


@transactional
def update_service_data_retention(service_data_retention_id, service_id, days_of_retention):
    updated_count = ServiceDataRetention.query.filter(
        ServiceDataRetention.id == service_data_retention_id,
        ServiceDataRetention.service_id == service_id
    ).update(
        {
            "days_of_retention": days_of_retention,
            "updated_at": datetime.utcnow()
        }
    )
    return updated_count