mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-12 21:51:11 -05:00
convert other daos
This commit is contained in:
@@ -1,4 +1,4 @@
|
|||||||
from sqlalchemy import desc
|
from sqlalchemy import desc, select
|
||||||
|
|
||||||
from app import db
|
from app import db
|
||||||
from app.dao.dao_utils import autocommit
|
from app.dao.dao_utils import autocommit
|
||||||
@@ -17,17 +17,27 @@ def insert_service_sms_sender(service, sms_sender):
|
|||||||
|
|
||||||
|
|
||||||
def dao_get_service_sms_senders_by_id(service_id, service_sms_sender_id):
|
def dao_get_service_sms_senders_by_id(service_id, service_sms_sender_id):
|
||||||
return ServiceSmsSender.query.filter_by(
|
# return ServiceSmsSender.query.filter_by(
|
||||||
|
# id=service_sms_sender_id, service_id=service_id, archived=False
|
||||||
|
# ).one()
|
||||||
|
stmt = select(ServiceSmsSender).filter_by(
|
||||||
id=service_sms_sender_id, service_id=service_id, archived=False
|
id=service_sms_sender_id, service_id=service_id, archived=False
|
||||||
).one()
|
)
|
||||||
|
return db.session.execute(stmt).scalars().one()
|
||||||
|
|
||||||
|
|
||||||
def dao_get_sms_senders_by_service_id(service_id):
|
def dao_get_sms_senders_by_service_id(service_id):
|
||||||
return (
|
# return (
|
||||||
ServiceSmsSender.query.filter_by(service_id=service_id, archived=False)
|
# ServiceSmsSender.query.filter_by(service_id=service_id, archived=False)
|
||||||
|
# .order_by(desc(ServiceSmsSender.is_default))
|
||||||
|
# .all()
|
||||||
|
# )
|
||||||
|
stmt = (
|
||||||
|
select(ServiceSmsSender)
|
||||||
|
.filter_by(ervice_id=service_id, archived=False)
|
||||||
.order_by(desc(ServiceSmsSender.is_default))
|
.order_by(desc(ServiceSmsSender.is_default))
|
||||||
.all()
|
|
||||||
)
|
)
|
||||||
|
return db.session.execute(stmt).scalars().all()
|
||||||
|
|
||||||
|
|
||||||
@autocommit
|
@autocommit
|
||||||
|
|||||||
@@ -1,3 +1,5 @@
|
|||||||
|
from sqlalchemy import select
|
||||||
|
|
||||||
from app import db
|
from app import db
|
||||||
from app.dao.dao_utils import autocommit
|
from app.dao.dao_utils import autocommit
|
||||||
from app.models import ServiceUser, User
|
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
|
# 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
|
# 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
|
# started to throw an error with one() method in sqlalchemy 2.0 unlike 1.4
|
||||||
return ServiceUser.query.filter_by(
|
# return ServiceUser.query.filter_by(
|
||||||
user_id=user_id, service_id=service_id
|
# user_id=user_id, service_id=service_id
|
||||||
).one_or_none()
|
# ).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):
|
def dao_get_active_service_users(service_id):
|
||||||
query = (
|
# query = (
|
||||||
db.session.query(ServiceUser)
|
# 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)
|
.join(User, User.id == ServiceUser.user_id)
|
||||||
.filter(User.state == "active", ServiceUser.service_id == service_id)
|
.filter(User.state == "active", ServiceUser.service_id == service_id)
|
||||||
)
|
)
|
||||||
|
return db.session.execute(stmt).scalars().all()
|
||||||
return query.all()
|
|
||||||
|
|
||||||
|
|
||||||
def dao_get_service_users_by_user_id(user_id):
|
def dao_get_service_users_by_user_id(user_id):
|
||||||
|
|||||||
Reference in New Issue
Block a user