cloud-api/src/service/router.py

"""
Router endpoints for <this module>

Endpoints:
 - List: Description
 - Endpoints: Description
"""
from fastapi import APIRouter, status
from psycopg.errors import UniqueViolation
from sqlalchemy.exc import IntegrityError

from src.exceptions import Conflict
from src.database import db_dependency
from src.auth.dependencies import super_admin_dependency, org_model_root_claim_query_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, \
	ServiceWithKeySchema, 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 = ServiceWithKeySchema(**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 = ServiceWithKeySchema(**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()