From 97c5191c472e5cb086cacefa4b4a020be00a8573 Mon Sep 17 00:00:00 2001
From: Jeremy Fee <jmfee@usgs.gov>
Date: Fri, 29 Jan 2021 16:45:33 -0700
Subject: [PATCH] Update session get/save methods
---
geomagio/api/db/session_table.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/geomagio/api/db/session_table.py b/geomagio/api/db/session_table.py
index f5a3f89e..a62ebb78 100644
--- a/geomagio/api/db/session_table.py
+++ b/geomagio/api/db/session_table.py
@@ -26,7 +26,7 @@ async def delete_session(session_id: str) -> None:
async def get_session(session_id: str) -> str:
query = session.select().where(session.c.session_id == session_id)
row = await database.fetch_one(query)
- return row.data
+ return row.get("data", None)
async def remove_expired_sessions(max_age: timedelta) -> None:
@@ -44,7 +44,7 @@ async def save_session(session_id: str, data: str) -> None:
.values(data=data, updated=updated)
)
count = await database.execute(query)
- if count == 0:
+ if count is None or count == 0:
# no matching session, insert
query = session.insert().values(
session_id=session_id, data=data, updated=updated
--
GitLab