Fix timestamp issue with PGSQL

Author: frankie567

fastapi_users_db_sqlalchemy/__init__.py

@@ -154,7 +154,6 @@ async def add_oauth_account(self, user: UP, create_dict: Dict[str, Any]) -> UP:
         self.session.add(user)
 
         await self.session.commit()
-        await self.session.refresh(user)
 
         return user
 
@@ -168,7 +167,7 @@ async def update_oauth_account(
             setattr(oauth_account, key, value)
         self.session.add(oauth_account)
         await self.session.commit()
-        await self.session.refresh(user)
+
         return user
 
     async def _get_user(self, statement: Select) -> Optional[UP]:

fastapi_users_db_sqlalchemy/access_token.py

@@ -24,7 +24,7 @@ class SQLAlchemyBaseAccessTokenTable(Generic[ID]):
     else:
         token: str = Column(String(length=43), primary_key=True)
         created_at: datetime = Column(
-            TIMESTAMPAware, index=True, nullable=False, default=now_utc
+            TIMESTAMPAware(timezone=True), index=True, nullable=False, default=now_utc
         )
 
 

tests/test_access_token.py

@@ -76,7 +76,9 @@ async def test_queries(
     updated_access_token = await sqlalchemy_access_token_db.update(
         access_token, update_dict
     )
-    assert updated_access_token.created_at == update_dict["created_at"]
+    assert updated_access_token.created_at.replace(microsecond=0) == update_dict[
+        "created_at"
+    ].replace(microsecond=0)
 
     # Get by token
     access_token_by_token = await sqlalchemy_access_token_db.get_by_token(