convert other daos

This commit is contained in:
Kenneth Kehl
2024-10-10 07:49:26 -07:00
parent cbd2cd132f
commit 2db45c8b50
2 changed files with 34 additions and 13 deletions

View File

@@ -1,3 +1,5 @@
from sqlalchemy import select
from app import db
from app.dao.dao_utils import autocommit
from app.models import ServiceUser, User
@@ -7,19 +9,28 @@ def dao_get_service_user(user_id, service_id):
# TODO: This has been changed to account for the test case failure
# that used this method but have any service user to return. Somehow, this
# started to throw an error with one() method in sqlalchemy 2.0 unlike 1.4
return ServiceUser.query.filter_by(
user_id=user_id, service_id=service_id
).one_or_none()
# return ServiceUser.query.filter_by(
# user_id=user_id, service_id=service_id
# ).one_or_none()
stmt = select(ServiceUser).filter_by(user_id=user_id, service_id=service_id)
return db.session.execute(stmt).scalars().one_or_none()
def dao_get_active_service_users(service_id):
query = (
db.session.query(ServiceUser)
# query = (
# db.session.query(ServiceUser)
# .join(User, User.id == ServiceUser.user_id)
# .filter(User.state == "active", ServiceUser.service_id == service_id)
# )
# return query.all()
stmt = (
select(ServiceUser)
.join(User, User.id == ServiceUser.user_id)
.filter(User.state == "active", ServiceUser.service_id == service_id)
)
return query.all()
return db.session.execute(stmt).scalars().all()
def dao_get_service_users_by_user_id(user_id):