""" Router endpoints for Endpoints: - List: Description - Endpoints: Description """ from fastapi import APIRouter, status from psycopg.errors import UniqueViolation from sqlalchemy.exc import IntegrityError from auth.exceptions import UnauthorizedException from exceptions import Conflict from src.database import db_dependency from src.auth.service import claims_dependency from src.auth.dependencies import super_admin_dependency, org_model_root_claim_query_dependency, org_model_root_claim_body_dependency from src.service.models import Service from src.service.utils import generate_api_key from src.service.dependencies import service_model_body_dependency from src.service.schemas import ServiceGetServiceResponse, ServicePostServiceRequest, ServicePostServiceResponse, \ ServiceWithKeyResponse, ServicePatchKeyResponse, ServicePatchKeyRequest, ServiceDeleteServiceRequest router = APIRouter( tags=["Service"], prefix="/service", ) @router.get("/", response_model=ServiceGetServiceResponse) async def get_all_services(db: db_dependency, org_model: org_model_root_claim_query_dependency): permission_models = db.query(Service).all() return {"services": permission_models} @router.post("/", response_model=ServicePostServiceResponse) async def register_service(db: db_dependency, su: super_admin_dependency, service_request: ServicePostServiceRequest): key = generate_api_key() service_model = Service(name=service_request.name, api_key=key) db.add(service_model) try: db.flush() except IntegrityError as e: if isinstance(e.orig, UniqueViolation): raise Conflict(message="Service with this name already exists") db.commit() response = ServiceWithKeyResponse(**service_model.__dict__) db.commit() return {"service": response} @router.patch("/key", response_model=ServicePatchKeyResponse) async def regenerate_api_key(db: db_dependency, su: super_admin_dependency, service_model: service_model_body_dependency, request_model: ServicePatchKeyRequest): key = generate_api_key() service_model.api_key = key db.flush() response = ServiceWithKeyResponse(**service_model.__dict__) db.commit() return {"service": response} @router.delete("/", status_code=status.HTTP_204_NO_CONTENT) async def remove_service(db: db_dependency, service_model: service_model_body_dependency, su: super_admin_dependency, request_model: ServiceDeleteServiceRequest): db.delete(service_model) db.commit()