jack 3 months ago
parent
commit
f2d2d1a921
1 changed files with 7 additions and 5 deletions
  1. 7 5
      app/services/token/token.py

+ 7 - 5
app/services/token/token.py

@@ -22,8 +22,8 @@ class TokenService:
         statement = select(Token).where(Token.assistant_token == token)
         result = await session.execute(statement)
         token = result.scalars().one_or_none()
-        if token is None:
-            raise ResourceNotFoundError(message="Token not found")
+        # if token is None:
+        #    raise ResourceNotFoundError(message="Token not found")
         return token
 
     @staticmethod
@@ -31,8 +31,8 @@ class TokenService:
         statement = select(Token).where(Token.id == token_id)
         result = session.execute(statement)
         token = result.scalars().one_or_none()
-        if token is None:
-            raise ResourceNotFoundError(message="Token not found")
+        # if token is None:
+        #    raise ResourceNotFoundError(message="Token not found")
         return token
 
     @staticmethod
@@ -45,7 +45,9 @@ class TokenService:
         return db_token
 
     @staticmethod
-    async def modify_token(session: AsyncSession, body: TokenUpdate, token: str) -> Token:
+    async def modify_token(
+        session: AsyncSession, body: TokenUpdate, token: str
+    ) -> Token:
         db_token = await TokenService.get_token(session=session, token=token)
         for key, value in body.model_dump(exclude_unset=True).items():
             setattr(db_token, key, value)