from app import db
from app.dao.dao_utils import transactional
from app.models import ServiceUser, User


def dao_get_service_user(user_id, service_id):
    return ServiceUser.query.filter_by(user_id=user_id, service_id=service_id).one()


def dao_get_active_service_users(service_id):
    query = ServiceUser.query.join(ServiceUser.user).filter(
        ServiceUser.service_id == service_id,
        User.state == 'active'
    )

    return query.all()


def dao_get_service_users_by_user_id(user_id):
    return ServiceUser.query.filter_by(user_id=user_id).all()


@transactional
def dao_update_service_user(service_user):
    db.session.add(service_user)