agent_survey : ajout champ iteration

Etienne Pallier
1 parent a760bcc8
Showing 2 changed files with 3 additions and 1 deletions Show diff stats
src/core/pyros_django/agent/Agent.py
src/core/pyros_django/common/models.py
@@ -436,7 +436,7 @@ class Agent:
         if AgentSurvey.objects.filter(name=self.name).exists():
             self._agent_survey = AgentSurvey.objects.get(name=self.name)
         else:
-            self._agent_survey = AgentSurvey.objects.create(name=self.name, validity_duration=60, mode=self.mode, status=self.status)
+            self._agent_survey = AgentSurvey.objects.create(name=self.name, validity_duration=60, mode=self.mode, status=self.status, iteration=-1)
         self.printd("Agent survey is", self._agent_survey)
  
         ### self._agent_logs = AgentLogs.objects.create(name=self.name, message="Step __init__")
@@ -825,6 +825,7 @@ class Agent:
         #self._agent_survey = AgentSurvey.objects.get(name=self.name)
         self._agent_survey.mode = self.mode
         self._agent_survey.status = self.status
+        self._agent_survey.iteration = self._iter_num
         self._agent_survey.save()
         #self._agent_survey.save(update_fields=["mode", "status"])
  
@@ -286,6 +286,7 @@ class AgentSurvey(models.Model):
     #validity_duration = models.DurationField(default=90)
     mode = models.CharField('agent mode', max_length=15, blank=True, choices=MODE_CHOICES)
     status = models.CharField(max_length=15, blank=True, choices=STATUS_CHOICES)
+    iteration = models.IntegerField(blank=True, null=True)
  
     class Meta:
         managed = True