File tree Expand file tree Collapse file tree 6 files changed +20
-11
lines changed
Expand file tree Collapse file tree 6 files changed +20
-11
lines changed Original file line number Diff line number Diff line change @@ -20,7 +20,7 @@ async def get_logic() -> AsyncGenerator[_Logic, None]:
2020
2121
2222async def get_user (
23- token : Annotated [str , Depends (APIKeyHeader (name = " access-token" ))],
23+ token : Annotated [str , Depends (APIKeyHeader (name = ' access-token' ))],
2424 logic : Logic ,
2525) -> _User | None :
2626 return await logic .users .retrieve_by_token (token )
Original file line number Diff line number Diff line change 44from app .models .auth import AccessToken
55from app .models .users .user import UserCreate
66
7- router = APIRouter (prefix = " /token" )
7+ router = APIRouter (prefix = ' /token' )
88
99
10- @router .post ("/" , response_model = AccessToken )
10+ @router .post ('/' , response_model = AccessToken )
1111async def token (data : UserCreate , logic : deps .Logic ):
1212 """
1313 Retrieve new access token
1414 """
1515 return await logic .auth .generate_token (data )
1616
1717
18- __all__ = [" router" ]
18+ __all__ = [' router' ]
Original file line number Diff line number Diff line change 44
55from typing import Self
66
7- from sqlalchemy .ext .asyncio import AsyncEngine , async_sessionmaker , create_async_engine
7+ from sqlalchemy .ext .asyncio import (AsyncEngine , async_sessionmaker ,
8+ create_async_engine )
89from sqlmodel .ext .asyncio .session import AsyncSession
910
1011from app import repositories as repos
Original file line number Diff line number Diff line change 99
1010
1111class Auth :
12- def __init__ (self , logic : " Logic" ):
12+ def __init__ (self , logic : ' Logic' ):
1313 self .logic = logic
1414
1515 async def generate_token (self , data : UserCreate ) -> AccessToken | None :
16- if (user := await self .logic .db .user .retrieve_by_email (data .email )) is None :
16+ if (
17+ user := await self .logic .db .user .retrieve_by_email (data .email )
18+ ) is None :
1719 raise exps .UserNotFoundException ()
1820 if not self .logic .security .pwd .checkpwd (data .password , user .password ):
1921 raise exps .UserIsCorrectException ()
20- access_token = self .logic .security .jwt .encode_token ({"id" : user .id }, 1440 )
22+ access_token = self .logic .security .jwt .encode_token (
23+ {'id' : user .id }, 1440
24+ )
2125 return AccessToken (token = access_token )
Original file line number Diff line number Diff line change @@ -17,6 +17,6 @@ def __init__(self, db: Database):
1717
1818 @classmethod
1919 @asynccontextmanager
20- async def create (cls ) -> AsyncGenerator [" Logic" , None ]:
20+ async def create (cls ) -> AsyncGenerator [' Logic' , None ]:
2121 async with Database () as db :
2222 yield cls (db )
Original file line number Diff line number Diff line change 88
99
1010class Users :
11- def __init__ (self , logic : " Logic" ):
11+ def __init__ (self , logic : ' Logic' ):
1212 self .logic = logic
1313
1414 async def create (self , data : UserCreate ) -> User | None :
@@ -21,6 +21,10 @@ async def create(self, data: UserCreate) -> User | None:
2121
2222 async def retrieve_by_token (self , token : str ) -> User | None :
2323 payload = self .logic .security .jwt .decode_token (token )
24- if not (user := await self .logic .db .user .retrieve_one (ident = payload .get ("id" ))):
24+ if not (
25+ user := await self .logic .db .user .retrieve_one (
26+ ident = payload .get ('id' )
27+ )
28+ ):
2529 raise exps .UserNotFoundException ()
2630 return user
You can’t perform that action at this time.
0 commit comments