From 7575253000c4b8a4147f31b17e5349b595fa39a3 Mon Sep 17 00:00:00 2001 From: Josh Date: Sun, 29 Nov 2020 02:35:30 +0000 Subject: [PATCH] regenerate policies and tasks on site/client change on agent --- api/tacticalrmm/agents/views.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/api/tacticalrmm/agents/views.py b/api/tacticalrmm/agents/views.py index 79fc180b..3c52ac16 100644 --- a/api/tacticalrmm/agents/views.py +++ b/api/tacticalrmm/agents/views.py @@ -93,6 +93,8 @@ def uninstall(request): @api_view(["PATCH"]) def edit_agent(request): agent = get_object_or_404(Agent, pk=request.data["id"]) + + old_site = agent.site.pk a_serializer = AgentSerializer(instance=agent, data=request.data, partial=True) a_serializer.is_valid(raise_exception=True) a_serializer.save() @@ -104,6 +106,11 @@ def edit_agent(request): p_serializer.is_valid(raise_exception=True) p_serializer.save() + # check if site changed and initiate generating correct policies + if old_site != request.data["site"]: + agent.generate_checks_from_policies(clear=True) + agent.generate_tasks_from_policies(clear=True) + return Response("ok")