Filter agent state api (not display state if agent is not updated since two days)

Alexis Koralewski
1 parent 3dd318d0
Showing 2 changed files with 10 additions and 2 deletions Show diff stats
CHANGELOG
src/core/pyros_django/api/views.py
 12-10-2022 (AKo): v0.6.1.0
+	- Filter agent state api (not display state if agent is not updated since two days)
 	- Update variable name in agent_detail & use method to check cmd status
 	- Adding do_restart_agent soft/hard mode
 	- Fixing display of get_specific_cmd error and sort
@@ -11,7 +11,7 @@ from common.models import PyrosUser, SP_Period, ScientificProgram, Sequence, Alb
 from routine_manager.functions import check_sequence_file_validity
 from rest_framework.request import Request
 from django.db.models import Q
-from datetime import datetime
+from datetime import datetime, timezone, timedelta
 from src.pyros_logger import log
  
 # Create your views here.
@@ -249,4 +249,11 @@ class AgentSurveyViewSet(viewsets.ModelViewSet):
     serializer_class = AgentSurveySerializer
     permission_classes = [IsAuthenticated]
     http_method_names = ["get"]
-    lookup_field = "name"
 \ No newline at end of file
+    lookup_field = "name"
+    def get_queryset(self):
+        agents = AgentSurvey.objects.all()
+        datetime_now = datetime.utcnow()
+        date_minus_two_days =  datetime_now - timedelta(days=2)
+        date_minus_two_days = date_minus_two_days.replace(tzinfo=timezone.utc)
+        agents = agents.exclude(updated__lt=date_minus_two_days)
+        return agents
 \ No newline at end of file