From 7e16eeb386a1e34f20f47ce611cb04995305ebae Mon Sep 17 00:00:00 2001
From: Kenneth Kehl <@kkehl@flexion.us>
Date: Fri, 11 Oct 2024 13:30:59 -0700
Subject: [PATCH] upgrade org and template dao to sqlalchemy 2.0

---
 app/dao/organization_dao.py | 12 +++++++-----
 app/service_invite/rest.py  |  2 +-
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/app/dao/organization_dao.py b/app/dao/organization_dao.py
index 9e44bcdd5..3a8a5e602 100644
--- a/app/dao/organization_dao.py
+++ b/app/dao/organization_dao.py
@@ -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):
diff --git a/app/service_invite/rest.py b/app/service_invite/rest.py
index f6d9627da..5728b3ed5 100644
--- a/app/service_invite/rest.py
+++ b/app/service_invite/rest.py
@@ -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)