upgrade org and template dao to sqlalchemy 2.0

This commit is contained in:
Kenneth Kehl
2024-10-11 13:30:59 -07:00
parent 70022c0596
commit 7e16eeb386
2 changed files with 8 additions and 6 deletions

View File

@@ -1,3 +1,4 @@
from sqlalchemy import select
from sqlalchemy.sql.expression import func
from app import db
@@ -6,14 +7,15 @@ from app.models import Domain, Organization, Service, User
def dao_get_organizations():
return Organization.query.order_by(
stmt = select(Organization).order_by(
Organization.active.desc(), Organization.name.asc()
).all()
)
return db.session.execute(stmt).scalars().all()
def dao_count_organizations_with_live_services():
return (
db.session.query(Organization.id)
stmt = (
select(func.count(Organization.id))
.join(Organization.services)
.filter(
Service.active.is_(True),
@@ -21,8 +23,8 @@ def dao_count_organizations_with_live_services():
Service.count_as_live.is_(True),
)
.distinct()
.count()
)
return db.session.execute(stmt).scalar() or 0
def dao_get_organization_services(organization_id):

View File

@@ -86,7 +86,7 @@ def _create_service_invite(invited_user, invite_link_host):
redis_store.set(
f"email-personalisation-{saved_notification.id}",
json.dumps(personalisation),
ex=2*24*60*60,
ex=2 * 24 * 60 * 60,
)
send_notification_to_queue(saved_notification, queue=QueueNames.NOTIFY)