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 utils.constants import ALGOS
from evidence.models import SystemProperty, UserProperty, Evidence from evidence.models import SystemProperty, UserProperty, Evidence
from lot.models import DeviceLot from lot.models import DeviceLot
from action.models import State
class Device: class Device:
@ -137,6 +138,11 @@ class Device:
return self.uuid return self.uuid
return self.uuids[0] 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): def get_lots(self):
self.lots = [ self.lots = [
x.lot for x in DeviceLot.objects.filter(device_id=self.id)] x.lot for x in DeviceLot.objects.filter(device_id=self.id)]