All user queries and services are async now.

Author: mikeckennedy

code/ch8-async-databases/db/pypi.sqlite

@@ -15,37 +15,32 @@ async def user_count() -> int:
         return result.scalar()
 
 
-def create_account(name: str, email: str, password: str) -> User:
-    session = db_session.create_session()
-
-    try:
-        user = User()
-        user.email = email
-        user.name = name
-        user.hash_password = crypto.hash(password, rounds=172_434)
+async def create_account(name: str, email: str, password: str) -> User:
+    user = User()
+    user.email = email
+    user.name = name
+    user.hash_password = crypto.hash(password, rounds=172_434)
 
+    async with db_session.create_async_session() as session:
         session.add(user)
-        session.commit()
+        await session.commit()
 
-        return user
-    finally:
-        session.close()
+    return user
 
 
-def login_user(email: str, password: str) -> Optional[User]:
-    session = db_session.create_session()
+async def login_user(email: str, password: str) -> Optional[User]:
+    async with db_session.create_async_session() as session:
+        query = select(User).filter(User.email == email)
+        results = await session.execute(query)
 
-    try:
-        user = session.query(User).filter(User.email == email).first()
+        user = results.scalar_one_or_none()
         if not user:
             return user
 
         if not crypto.verify(password, user.hash_password):
             return None
 
         return user
-    finally:
-        session.close()
 
 
 async def get_user_by_id(user_id: int) -> Optional[User]:

code/ch8-async-databases/services/user_service.py

@@ -1,3 +1,5 @@
+import asyncio
+
 import fastapi
 from fastapi_chameleon import template
 from starlette import status
@@ -37,7 +39,7 @@ async def register(request: Request):
         return vm.to_dict()
 
     # Create the account
-    account = user_service.create_account(vm.name, vm.email, vm.password)
+    account = await user_service.create_account(vm.name, vm.email, vm.password)
 
     # Login user
     response = fastapi.responses.RedirectResponse(url='/account', status_code=status.HTTP_302_FOUND)
@@ -65,8 +67,9 @@ async def login_post(request: Request):
     if vm.error:
         return vm.to_dict()
 
-    user = user_service.login_user(vm.email, vm.password)
+    user = await user_service.login_user(vm.email, vm.password)
     if not user:
+        await asyncio.sleep(5)
         vm.error = "The account does not exist or the password is wrong."
         return vm.to_dict()