current state helper function added

This commit is contained in:
Thomas Nahuel Rusiecki 2024-12-12 17:01:21 -03:00 committed by Cayo Puigdefabregas
parent 268666909f
commit e646cdfa56

View file

@ -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:
@ -137,6 +138,11 @@ class Device:
return self.uuid
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)]