From 92ce6d89b29368beee63890e175856b3b73993e1 Mon Sep 17 00:00:00 2001 From: Thomas Rusiecki Date: Thu, 12 Dec 2024 17:01:21 -0300 Subject: [PATCH] current state helper function added --- device/models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/device/models.py b/device/models.py index 59ea09a..aa1a8fc 100644 --- a/device/models.py +++ b/device/models.py @@ -3,6 +3,7 @@ from django.db import models, connection from utils.constants import ALGOS from evidence.models import SystemProperty, UserProperty, Evidence from lot.models import DeviceLot +from action.models import State class Device: @@ -127,6 +128,11 @@ class Device: def last_uuid(self): return self.uuids[0] + def get_current_state(self): + uuid = self.last_uuid + + return State.objects.filter(snapshot_uuid=uuid).order_by('-date').first() + def get_lots(self): self.lots = [ x.lot for x in DeviceLot.objects.filter(device_id=self.id)]