From 9cb9c4a083441176a56e2d567efb209ca059db44 Mon Sep 17 00:00:00 2001 From: "deepsource-autofix[bot]" <62050782+deepsource-autofix[bot]@users.noreply.github.com> Date: Sat, 31 May 2025 16:27:11 +0000 Subject: [PATCH] style: format code with Autopep8, Black and Ruff Formatter This commit fixes the style issues introduced in 5bf63ea according to the output from Autopep8, Black and Ruff Formatter. Details: None --- ai_services/api/endpoints/incidents.py | 1 + ai_services/api/endpoints/users.py | 28 ++++++++++++++---------- ai_services/core/auth.py | 4 +++- ai_services/models/incident.py | 4 +++- ai_services/services/incident_service.py | 11 +++++----- 5 files changed, 29 insertions(+), 19 deletions(-) diff --git a/ai_services/api/endpoints/incidents.py b/ai_services/api/endpoints/incidents.py index bbbabfc..2b5ea7f 100644 --- a/ai_services/api/endpoints/incidents.py +++ b/ai_services/api/endpoints/incidents.py @@ -6,6 +6,7 @@ router = APIRouter() + @router.post("/incidents", response_model=IncidentOut) def report_incident(payload: IncidentCreate, user=Depends(get_current_user)): try: diff --git a/ai_services/api/endpoints/users.py b/ai_services/api/endpoints/users.py index cc11827..5b79650 100644 --- a/ai_services/api/endpoints/users.py +++ b/ai_services/api/endpoints/users.py @@ -1,30 +1,32 @@ from fastapi import APIRouter, HTTPException, status, Depends, Request from pydantic import BaseModel, EmailStr from core.auth import get_current_user -from core.database import supabase # Assuming supabase client is initialized in core/database.py +from core.database import ( + supabase, +) # Assuming supabase client is initialized in core/database.py from supabase import Client from core.auth import get_current_user router = APIRouter() + # Request schemas class RegisterRequest(BaseModel): email: EmailStr password: str + class LoginRequest(BaseModel): email: EmailStr password: str + # Route: Register user @router.post("/register") def register_user(payload: RegisterRequest): try: result = supabase.auth.sign_up( - { - "email": payload.email, - "password": payload.password - } + {"email": payload.email, "password": payload.password} ) return {"message": "User registered successfully", "user": result.user} except Exception as e: @@ -36,20 +38,22 @@ def register_user(payload: RegisterRequest): def login_user(payload: LoginRequest): try: result = supabase.auth.sign_in_with_password( - { - "email": payload.email, - "password": payload.password - } + {"email": payload.email, "password": payload.password} ) if result.session is None: raise HTTPException(status_code=401, detail="Invalid credentials") # Insert user data into the users table if needed - user_id = result.user.id if result.user and hasattr(result.user, "id") else None + user_id = result.user.id if result.user and hasattr( + result.user, "id") else None user_data = { "id": user_id, "email": payload.email, - "created_at": result.user.created_at.isoformat() if result.user and hasattr(result.user, "created_at") else None + "created_at": ( + result.user.created_at.isoformat() + if result.user and hasattr(result.user, "created_at") + else None + ), } # print(user_data) Bugs fixing supabase.table("users").insert(user_data).execute() @@ -57,7 +61,7 @@ def login_user(payload: LoginRequest): return { "access_token": result.session.access_token, "refresh_token": result.session.refresh_token, - "user": result.user + "user": result.user, } except Exception as e: raise HTTPException(status_code=401, detail=str(e)) diff --git a/ai_services/core/auth.py b/ai_services/core/auth.py index 72734ac..4bd9b05 100644 --- a/ai_services/core/auth.py +++ b/ai_services/core/auth.py @@ -5,13 +5,15 @@ security = HTTPBearer() + def get_current_user(credentials: HTTPAuthorizationCredentials = Depends(security)): token = credentials.credentials try: # Verify the token with Supabase (optional) user = supabase.auth.get_user(token) if user.user is None: - raise HTTPException(status_code=401, detail="Invalid or expired token") + raise HTTPException( + status_code=401, detail="Invalid or expired token") return user.user except Exception as e: raise HTTPException(status_code=401, detail=str(e)) diff --git a/ai_services/models/incident.py b/ai_services/models/incident.py index 5d3d204..75306a3 100644 --- a/ai_services/models/incident.py +++ b/ai_services/models/incident.py @@ -3,14 +3,16 @@ from typing import Optional from datetime import datetime + class IncidentCreate(BaseModel): title: str description: str location: Optional[str] = None reporter_id: Optional[str] = None # Could be user_id or email + class IncidentOut(BaseModel): - id: str + id: str title: str description: str location: Optional[str] diff --git a/ai_services/services/incident_service.py b/ai_services/services/incident_service.py index cdfc387..f42db10 100644 --- a/ai_services/services/incident_service.py +++ b/ai_services/services/incident_service.py @@ -4,10 +4,11 @@ from uuid import uuid4 from datetime import datetime + def create_incident(data: dict) -> dict: priority = triage_incident(data["description"]) incident_id = str(uuid4()) - + new_incident = { "id": incident_id, "title": data["title"], @@ -15,9 +16,9 @@ def create_incident(data: dict) -> dict: "location": data.get("location"), "reporter_id": data.get("reporter_id"), "priority": priority, - "created_at": datetime.utcnow().isoformat() + "created_at": datetime.utcnow().isoformat(), } - + response = supabase.table("incidents").insert(new_incident).execute() return response.data[0] if response.data else new_incident @@ -27,7 +28,7 @@ def create_incident(data: dict) -> dict: "title": "Unauthorized Access Detected", "description": "An unknown individual was seen entering the restricted server room.", "location": "Server Room 3", - "reported_by": "John Doe" + "reported_by": "John Doe", } result = create_incident(sample_data) - print(result) \ No newline at end of file + print(result)