Add refresh call after create and update to make sure eager relationships are loaded

Author: frankie567

fastapi_users_db_sqlalchemy/__init__.py

@@ -142,13 +142,15 @@ async def create(self, create_dict: Dict[str, Any]) -> UP:
         user = self.user_table(**create_dict)
         self.session.add(user)
         await self.session.commit()
+        await self.session.refresh(user)
         return user
 
     async def update(self, user: UP, update_dict: Dict[str, Any]) -> UP:
         for key, value in update_dict.items():
             setattr(user, key, value)
         self.session.add(user)
         await self.session.commit()
+        await self.session.refresh(user)
         return user
 
     async def delete(self, user: UP) -> None:

fastapi_users_db_sqlalchemy/access_token.py

@@ -73,13 +73,15 @@ async def create(self, create_dict: Dict[str, Any]) -> AP:
         access_token = self.access_token_table(**create_dict)
         self.session.add(access_token)
         await self.session.commit()
+        await self.session.refresh(access_token)
         return access_token
 
     async def update(self, access_token: AP, update_dict: Dict[str, Any]) -> AP:
         for key, value in update_dict.items():
             setattr(access_token, key, value)
         self.session.add(access_token)
         await self.session.commit()
+        await self.session.refresh(access_token)
         return access_token
 
     async def delete(self, access_token: AP) -> None: