refactor: reorganize limiter setup to dedicated services/limiter.py file and update imports across modules

Author: IkeSalmonson

app/main.py

@@ -2,11 +2,11 @@
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI
-from slowapi import Limiter, _rate_limit_exceeded_handler
-from slowapi.util import get_remote_address
+from slowapi import _rate_limit_exceeded_handler
 
 from app.routers.router import setup_router as setup_router_v2
 from app.services.database.database import AsyncSessionLocal, init_db
+from app.services.limiter import limiter
 
 logger = logging.getLogger(__name__)
 
@@ -22,7 +22,7 @@ async def lifespan(app: FastAPI):
         pass
 
 
-limiter = Limiter(key_func=get_remote_address)
+appLimiter = limiter
 
 
 app = FastAPI(
@@ -32,7 +32,7 @@ async def lifespan(app: FastAPI):
 )
 
 
-app.state.limiter = limiter
+app.state.limiter = appLimiter
 app.add_exception_handler(429, _rate_limit_exceeded_handler)
 
 app.include_router(setup_router_v2(), prefix="/api")

app/routers/authentication.py

@@ -4,17 +4,15 @@
 from fastapi import APIRouter, Depends, HTTPException, Request, status
 from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
 from jwt.exceptions import InvalidTokenError
-from slowapi import Limiter
-from slowapi.util import get_remote_address
 from sqlmodel.ext.asyncio.session import AsyncSession
 
 from app.schemas import Community, Token, TokenPayload
 from app.services import auth
 from app.services.database.models import Community as DBCommunity
 from app.services.database.orm.community import get_community_by_username
+from app.services.limiter import limiter
 
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/authentication/token")
-limiter = Limiter(key_func=get_remote_address)
 
 
 def setup():
@@ -115,6 +113,7 @@ async def login_for_access_token(
     @router.get("/me", response_model=Community)
     @limiter.limit("60/minute")
     async def read_community_me(
+        request: Request,
         current_community: Annotated[
             DBCommunity, Depends(get_current_active_community)
         ],

app/routers/news/routes.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter, Request, status
 from pydantic import BaseModel
-from services.database.orm.news import get_news_by_query_params
+
+from app.services.database.orm.news import get_news_by_query_params
 
 
 class NewsPostResponse(BaseModel):

app/services/database/models/subscriptions.py

@@ -1,9 +1,10 @@
 from typing import List, Optional
 
-from schemas import SubscriptionTagEnum
 from sqlalchemy import JSON, Column
 from sqlmodel import Field, SQLModel
 
+from app.schemas import SubscriptionTagEnum
+
 
 class Subscription(SQLModel, table=True):
     __tablename__ = "subscriptions"  # type: ignore

app/services/database/orm/news.py

@@ -1,9 +1,10 @@
 from typing import Optional
 
-from services.database.models import News
 from sqlmodel import select
 from sqlmodel.ext.asyncio.session import AsyncSession
 
+from app.services.database.models import News
+
 
 async def get_news_by_query_params(
     session: AsyncSession,

app/services/limiter.py

@@ -0,0 +1,4 @@
+from slowapi import Limiter
+from slowapi.util import get_remote_address
+
+limiter = Limiter(key_func=get_remote_address)