76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
import json
|
|
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.db.models import Count
|
|
from dashboard.mixins import InventaryMixin, DetailsMixin
|
|
from device.models import Device
|
|
from snapshot.xapian import search
|
|
from snapshot.models import Annotation
|
|
from lot.models import Lot
|
|
|
|
|
|
class UnassignedDevicesView(InventaryMixin):
|
|
template_name = "unassigned_devices.html"
|
|
section = "Unassigned"
|
|
title = _("Unassigned Devices")
|
|
breadcrumb = "Devices / Unassigned Devices"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
annotations = Annotation.objects.filter(
|
|
owner=self.request.user).filter(
|
|
key="hidalgo1").order_by('created')
|
|
# 'created').distinct('value')
|
|
# ).annotate(num_lots=Count('lot')).filter(num_lots=0)
|
|
|
|
hids = {}
|
|
ids = []
|
|
for x in annotations:
|
|
if not hids.get(x.key):
|
|
hids[x.key] = x.uuid
|
|
ids.append(str(x.uuid))
|
|
|
|
devices = []
|
|
for xa in search(ids):
|
|
# import pdb; pdb.set_trace()
|
|
snap = json.loads(xa.document.get_data())
|
|
dev = snap.get("device", {})
|
|
dev["id"] = snap["uuid"]
|
|
devices.append(dev)
|
|
|
|
context.update({
|
|
'devices': devices
|
|
})
|
|
return context
|
|
|
|
|
|
class AllDevicesView(InventaryMixin):
|
|
template_name = "unassigned_devices.html"
|
|
section = "All"
|
|
title = _("All Devices")
|
|
breadcrumb = "Devices / All Devices"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
devices = Device.objects.filter(owner=self.request.user)
|
|
context.update({
|
|
'devices': devices,
|
|
})
|
|
return context
|
|
|
|
|
|
class LotDashboardView(InventaryMixin, DetailsMixin):
|
|
template_name = "unassigned_devices.html"
|
|
section = "Unassigned"
|
|
title = _("Lot Devices")
|
|
breadcrumb = "Devices / Lot Devices"
|
|
model = Lot
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
devices = self.object.devices.filter(owner=self.request.user)
|
|
context.update({
|
|
'devices': devices,
|
|
})
|
|
return context
|