FEAT: Add admin create community function and endpoint, file name changed to project standard and route added to router

Author: IkeSalmonson

app/main.py

@@ -4,7 +4,7 @@
 from fastapi import FastAPI
 from slowapi import _rate_limit_exceeded_handler
 
-from app.routers.admin.admin import create_community_admin
+from app.routers.admin.routes import create_admin
 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
@@ -17,7 +17,7 @@ async def lifespan(app: FastAPI):
     # add check db file and create if not found
     await init_db()
     app.db_session_factory = AsyncSessionLocal()
-    await create_community_admin(app.db_session_factory)
+    await create_admin(app.db_session_factory)
     try:
         yield
     finally:

app/routers/admin/admin.py app/routers/admin/routes.pyapp/routers/admin/admin.py renamed to app/routers/admin/routes.py

@@ -9,12 +9,13 @@
 from app.services import auth
 from app.services.database.models import Community as DBCommunity  # Precisa?
 from app.services.database.orm.community import create_community
+from app.services.limiter import limiter
 
 # ADMIN_USER = os.getenv("ADMIN_USER")
 # ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD")
 
 
-async def create_community_admin(session: AsyncSession):
+async def create_admin(session: AsyncSession):
     ADMIN_USER = os.getenv("ADMIN_USER")
     ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD")
     password = ADMIN_PASSWORD
@@ -25,7 +26,6 @@ async def create_community_admin(session: AsyncSession):
         password=hashed_password,
         role="admin",
     )
-    session: AsyncSession = session
     session.add(community)
     await session.commit()
     await session.refresh(community)
@@ -37,30 +37,32 @@ class CommunityPostResponse(BaseModel):
 
 
 def setup():
-    router = APIRouter(prefix="/admin", tags=["news"])
+    router = APIRouter(prefix="/admin", tags=["admin"])
 
     @router.post(
-        "create_community",
+        "/create_community",
         response_model=CommunityPostResponse,
-        status_code=status.HTTP_200_OK,
+        status_code=status.HTTP_201_CREATED,
         summary="Create Community endpoint",
         description="Create Community and returns a confirmation message",
     )
+    @limiter.limit("60/minute")
     async def post_create_community(
         request: Request,
         admin_community: Annotated[
             DBCommunity, Depends(get_current_active_community)
         ],
-        community: Annotated[DBCommunity],
+        community: DBCommunity,
     ):
         """
-        Server Admin endpoint that creates Community and returns a confirmation message.
+        Server Admin endpoint that creates Community and returns a confirmation
+          message.
         """
-        admin_role = admin_community.get("role")
+        admin_role = admin_community.role
         if admin_role != "admin":
             return {"status": "Unauthorized"}
-        await create_community(
-            session=request.app.db_session_factory, community=community
-        )
+        await create_community(request=request, community=community)
 
         return CommunityPostResponse()
+
+    return router

app/routers/authentication.py

@@ -70,25 +70,6 @@ async def authenticate_community(
             return None
         return found_community
 
-    # Teste
-
-    @router.post("/create_commumity")
-    async def create_community(request: Request):
-        password = "123Asd!@#"
-        hashed_password = auth.hash_password(password)
-        community = DBCommunity(
-            username="username",
-            email="username@test.com",
-            password=hashed_password,
-        )
-        session: AsyncSession = request.app.db_session_factory
-        session.add(community)
-        await session.commit()
-        await session.refresh(community)
-        return {"msg": "succes? "}
-
-    # Teste
-
     @router.post("/token", response_model=Token)
     @limiter.limit("60/minute")
     async def login_for_access_token(

app/routers/router.py

@@ -1,5 +1,6 @@
 from fastapi import APIRouter
 
+from app.routers.admin.routes import setup as admin_router_setup
 from app.routers.authentication import setup as authentication_router_setup
 from app.routers.healthcheck.routes import setup as healthcheck_router_setup
 from app.routers.libraries.routes import setup as libraries_router_setup
@@ -12,4 +13,5 @@ def setup_router() -> APIRouter:
     router.include_router(news_router_setup(), prefix="")
     router.include_router(authentication_router_setup(), prefix="")
     router.include_router(libraries_router_setup(), prefix="")
+    router.include_router(admin_router_setup(), prefix="")
     return router

app/services/encryption.py

@@ -11,14 +11,14 @@
         "ENCRYPTION_KEY não está definida nas variáveis de ambiente."
     )
 
-cipher = Fernet(ENCRYPTION_KEY.encode())
+cipher = Fernet(ENCRYPTION_KEY)
 
 
 def encrypt_email(email: str) -> str:
     """Criptografa uma string de e-mail."""
-    return cipher.encrypt(email.encode()).decode()
+    return cipher.encrypt(email.encode())
 
 
 def decrypt_email(encrypted_email: str) -> str:
     """Descriptografa uma string de e-mail."""
-    return cipher.decrypt(encrypted_email.encode()).decode()
+    return cipher.decrypt(encrypted_email).decode()