Change the route prefixes

Author: ArtyomVancyan

.env

@@ -1,6 +1,6 @@
 OAUTH2_CLIENT_ID=eccd08d6736b7999a32a
 OAUTH2_CLIENT_SECRET=642999c1c5f2b3df8b877afdc78252ef5b594d31
-OAUTH2_CALLBACK_URL=http://127.0.0.1:8000/auth/callback
+OAUTH2_CALLBACK_URL=http://127.0.0.1:8000/oauth2/token
 OAUTH2_REDIRECT_URL=http://127.0.0.1:8000/
 
 JWT_SECRET=secret

fastapi_oauth2/base.py

@@ -98,7 +98,7 @@ async def get_token_data(
             params: Optional[Dict[str, Any]] = None,
             headers: Optional[Dict[str, Any]] = None,
             redirect_uri: Optional[str] = None,
-    ) -> Optional[dict]:
+    ) -> Optional[Dict[str, Any]]:
         params = params or {}
         additional_headers = headers or {}
         additional_headers.update(self.additional_headers or {})

fastapi_oauth2/router.py

@@ -10,7 +10,7 @@
     OAUTH2_REDIRECT_URL,
 )
 
-router = APIRouter()
+router = APIRouter(prefix="/oauth2")
 oauth2 = GitHubOAuth2(
     client_id=OAUTH2_CLIENT_ID,
     client_secret=OAUTH2_CLIENT_SECRET,
@@ -19,17 +19,17 @@
 )
 
 
-@router.get("/auth/login")
+@router.get("/login")
 async def login():
     return await oauth2.login_redirect()
 
 
-@router.get("/auth/callback")
-async def callback(request: Request):
+@router.get("/token")
+async def token(request: Request):
     return await oauth2.token_redirect(request)
 
 
-@router.get("/auth/logout")
+@router.get("/logout")
 async def logout():
     response = RedirectResponse(OAUTH2_REDIRECT_URL)
     response.delete_cookie("Authorization")