139 lines
4.8 KiB
Python
139 lines
4.8 KiB
Python
from ninja import Router
|
|
from ninja.errors import HttpError
|
|
|
|
from apps.accounts.api import require_permission
|
|
from apps.knowledge.governance import ensure_governed_asset
|
|
from apps.knowledge.models import GovernedAssetType
|
|
from apps.solutions.models import Solution, TicketSolutionLink
|
|
from apps.solutions.schemas import (
|
|
AdoptSolutionIn,
|
|
SolutionIn,
|
|
SolutionOut,
|
|
TicketSolutionLinkIn,
|
|
TicketSolutionLinkOut,
|
|
TicketSolutionOut,
|
|
)
|
|
from apps.tickets.models import Ticket
|
|
from apps.tickets.services import (
|
|
TicketDataError,
|
|
TicketStatusError,
|
|
)
|
|
from apps.tickets.services import (
|
|
adopt_solution_for_ticket as adopt_solution_for_ticket_service,
|
|
)
|
|
|
|
router = Router(tags=["solutions"])
|
|
|
|
|
|
@router.post("", response=SolutionOut)
|
|
def create_solution(request, payload: SolutionIn):
|
|
require_permission(request, "solution.manage")
|
|
solution = Solution.objects.create(
|
|
title=payload.title,
|
|
problem_signature=payload.problem_signature,
|
|
related_system=payload.related_system,
|
|
environment_scope=payload.environment_scope,
|
|
root_cause=payload.root_cause,
|
|
solution=payload.solution,
|
|
risk_note=payload.risk_note,
|
|
status="draft",
|
|
allow_auto_resolve=payload.allow_auto_resolve,
|
|
requires_human_action=payload.requires_human_action,
|
|
requires_external_confirmation=payload.requires_external_confirmation,
|
|
requires_production_change=payload.requires_production_change,
|
|
created_by=request.user if request.user.is_authenticated else None,
|
|
)
|
|
ensure_governed_asset(
|
|
GovernedAssetType.SOLUTION,
|
|
solution,
|
|
actor=request.user if request.user.is_authenticated else None,
|
|
)
|
|
return solution
|
|
|
|
|
|
@router.get("", response=list[SolutionOut])
|
|
def list_solutions(request, status: str | None = None):
|
|
require_permission(request, "solution.view")
|
|
queryset = Solution.objects.all()
|
|
if status:
|
|
queryset = queryset.filter(status=status)
|
|
return queryset[:100]
|
|
|
|
|
|
@router.get("/tickets/{ticket_id}", response=list[TicketSolutionOut])
|
|
def list_ticket_solutions(request, ticket_id: int):
|
|
require_permission(request, "solution.view")
|
|
queryset = TicketSolutionLink.objects.select_related("solution", "ticket").filter(
|
|
ticket_id=ticket_id
|
|
)
|
|
return [
|
|
{
|
|
"id": link.id,
|
|
"ticket_id": link.ticket_id,
|
|
"solution_id": link.solution_id,
|
|
"link_type": link.link_type,
|
|
"solution": link.solution,
|
|
"created_at": link.created_at,
|
|
}
|
|
for link in queryset
|
|
]
|
|
|
|
|
|
@router.get("/{solution_id}", response=SolutionOut)
|
|
def get_solution(request, solution_id: int):
|
|
require_permission(request, "solution.view")
|
|
return Solution.objects.get(id=solution_id)
|
|
|
|
|
|
@router.post("/{solution_id}/link-ticket", response=TicketSolutionLinkOut)
|
|
def link_solution_to_ticket(request, solution_id: int, payload: TicketSolutionLinkIn):
|
|
require_permission(request, "solution.manage")
|
|
if payload.link_type == "adopted":
|
|
raise HttpError(400, "Use adopt-ticket to adopt a solution")
|
|
solution = Solution.objects.get(id=solution_id)
|
|
ticket = Ticket.objects.get(id=payload.ticket_id)
|
|
link, _ = TicketSolutionLink.objects.get_or_create(
|
|
ticket=ticket,
|
|
solution=solution,
|
|
link_type=payload.link_type,
|
|
defaults={
|
|
"agent_run_id": payload.agent_run_id,
|
|
"created_by": request.user if request.user.is_authenticated else None,
|
|
},
|
|
)
|
|
return {
|
|
"id": link.id,
|
|
"ticket_id": ticket.id,
|
|
"solution_id": solution.id,
|
|
"link_type": link.link_type,
|
|
"solution_usage_count": solution.usage_count,
|
|
"created_at": link.created_at,
|
|
}
|
|
|
|
|
|
@router.post("/{solution_id}/adopt-ticket", response=TicketSolutionLinkOut)
|
|
def adopt_solution_for_ticket(request, solution_id: int, payload: AdoptSolutionIn):
|
|
require_permission(request, "solution.manage")
|
|
require_permission(request, "ticket.resolve")
|
|
operator = request.user if request.user.is_authenticated else None
|
|
try:
|
|
solution = Solution.objects.get(id=solution_id)
|
|
ticket, link = adopt_solution_for_ticket_service(
|
|
Ticket.objects.get(id=payload.ticket_id),
|
|
solution,
|
|
operator,
|
|
remark=payload.remark,
|
|
replied_to_customer=payload.replied_to_customer,
|
|
)
|
|
solution.refresh_from_db(fields=["usage_count"])
|
|
except (TicketDataError, TicketStatusError) as error:
|
|
raise HttpError(400, str(error)) from error
|
|
return {
|
|
"id": link.id,
|
|
"ticket_id": ticket.id,
|
|
"solution_id": solution.id,
|
|
"link_type": link.link_type,
|
|
"solution_usage_count": solution.usage_count,
|
|
"created_at": link.created_at,
|
|
}
|