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)]