Skip to content

Commit 3388698

Browse files
committed
fix(types): add type annotations for MyPy compliance
Add explicit type annotations to resolve MyPy type checking errors. Changes: - Add type annotation for JWT encode return value (str) - Add type annotation for JWT decode return value (dict[str, Any]) - Add type annotation for get_remote_address return value (str) - Extract retry_after with explicit type annotation All changes ensure strict type safety without changing runtime behavior. EOF )
1 parent 651967e commit 3388698

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

app/security/auth.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def create_access_token(
5050

5151
to_encode.update({"exp": expire, "iat": datetime.utcnow()})
5252

53-
encoded_jwt = jwt.encode(
53+
encoded_jwt: str = jwt.encode(
5454
to_encode,
5555
settings.security.secret_key,
5656
algorithm=settings.security.jwt_algorithm,
@@ -83,7 +83,7 @@ async def verify_token(
8383
settings = get_settings()
8484

8585
try:
86-
payload = jwt.decode(
86+
payload: dict[str, Any] = jwt.decode(
8787
credentials.credentials,
8888
settings.security.secret_key,
8989
algorithms=[settings.security.jwt_algorithm],

app/security/rate_limiting.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,8 @@ def get_rate_limit_key(request: Request) -> str:
4747
return f"token:{token_hash}"
4848

4949
# Fallback to IP address
50-
return get_remote_address(request)
50+
ip_address: str = get_remote_address(request)
51+
return ip_address
5152

5253

5354
# Initialize limiter with configuration
@@ -105,12 +106,13 @@ async def rate_limit_exceeded_handler(
105106
client=get_rate_limit_key(request),
106107
)
107108

109+
retry_after: str | None = exc.retry_after if hasattr(exc, "retry_after") else None
108110
return {
109111
"error": {
110112
"code": "RATE_LIMIT_EXCEEDED",
111113
"message": "Too many requests. Please try again later.",
112114
"details": {
113-
"retry_after": exc.retry_after if hasattr(exc, "retry_after") else None,
115+
"retry_after": retry_after,
114116
},
115117
}
116118
}

0 commit comments

Comments
 (0)