From 0cffa5a9513c2f8a6fc09f80219b99ad496f27ca Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 19 Jul 2024 17:40:01 +0200 Subject: [PATCH] new structure --- dashboard/mixins.py | 8 +- dashboard/templates/base.html | 5 -- dashboard/urls.py | 1 - dashboard/views.py | 29 ++----- device/forms.py | 14 ++++ device/migrations/0004_delete_device.py | 18 ++++ device/models.py | 83 ++++++++++++++----- device/templates/details.html | 43 ++-------- device/templates/new_device.html | 25 +++++- device/urls.py | 6 +- device/views.py | 49 +++++++++-- lot/forms.py | 5 +- .../0002_remove_lot_devices_devicelot.py | 39 +++++++++ lot/models.py | 18 +++- lot/views.py | 2 +- .../0002_remove_annotation_device.py | 17 ++++ snapshot/models.py | 1 - snapshot/parse.py | 12 --- snapshot/views.py | 2 +- 19 files changed, 262 insertions(+), 115 deletions(-) create mode 100644 device/migrations/0004_delete_device.py create mode 100644 lot/migrations/0002_remove_lot_devices_devicelot.py create mode 100644 snapshot/migrations/0002_remove_annotation_device.py diff --git a/dashboard/mixins.py b/dashboard/mixins.py index bd93513..470bf0d 100644 --- a/dashboard/mixins.py +++ b/dashboard/mixins.py @@ -5,6 +5,7 @@ from django.core.exceptions import PermissionDenied from django.contrib.auth.mixins import LoginRequiredMixin from django.views.generic.base import TemplateView from device.models import Device +from snapshot.models import Annotation from lot.models import LotTag @@ -45,7 +46,10 @@ class DashboardView(LoginRequiredMixin): def get_session_devices(self): # import pdb; pdb.set_trace() dev_ids = self.request.session.pop("devices", []) - self._devices = Device.objects.filter(id__in=dev_ids).filter(owner=self.request.user) + + self._devices = [] + for x in Annotation.objects.filter(value__in=dev_ids).filter(owner=self.request.user).distinct(): + self._devices.append(Device(id=x.value)) return self._devices @@ -53,7 +57,7 @@ class DetailsMixin(DashboardView, TemplateView): def get(self, request, *args, **kwargs): self.pk = kwargs['pk'] - self.object = get_object_or_404(self.model, pk=self.pk) + self.object = get_object_or_404(self.model, pk=self.pk, owner=self.request.user) return super().get(request, *args, **kwargs) def get_context_data(self, **kwargs): diff --git a/dashboard/templates/base.html b/dashboard/templates/base.html index 296dd35..be20573 100644 --- a/dashboard/templates/base.html +++ b/dashboard/templates/base.html @@ -90,11 +90,6 @@ {% trans 'Unassigned devices' %} - - @@ -47,7 +44,6 @@ (Edit Device)
- {% if object.hid %}Snapshot{% else %}Placeholder{% endif %}
@@ -56,29 +52,24 @@
{{ object.id }}
-
-
Id device internal
-
-
-
Type
-
{{ snapshot.doc.device.type }}
+
{{ object.type }}
Manufacturer
-
{{ snapshot.doc.device.manufacturer|default:"" }}
+
{{ object.manufacturer|default:"" }}
Model
-
{{ snapshot.doc.device.model|default:"" }}
+
{{ object.model|default:"" }}
Serial Number
-
{{ snapshot.doc.device.serialNumber|default:"" }}
+
{{ object.last_snapshot.doc.device.serialNumber|default:"" }}
Identifiers
@@ -132,7 +123,7 @@ {% for tag in lot_tags %}
{{ tag }}
- {% for lot in object.lot_set.filter %} + {% for lot in tag.lot_set.filter %} {% if lot.type == tag %}
@@ -171,32 +162,10 @@
-
-
Traceability log Details
-
- -
- Snapshot ✓ - 14:07 23-06-2024 -
- -
- EraseCrypto ✓ - 14:07 23-06-2024 -
- -
- EraseCrypto ✓ - 14:07 23-06-2024 -
- -
-
-
Components last snapshot
- {% for c in snapshot.components %} + {% for c in object.last_snapshot.doc.components %}
{{ c.type }}
diff --git a/device/templates/new_device.html b/device/templates/new_device.html index 8f2df37..8907ef0 100644 --- a/device/templates/new_device.html +++ b/device/templates/new_device.html @@ -22,11 +22,34 @@
{% endif %} -{% bootstrap_form form %} +{# bootstrap_form form #} + + + +
+ {% csrf_token %} +
+ {% for f in form %} +
+ {{ f.as_p }} +
+ {% endfor %} +
+ +
{% endblock %} diff --git a/device/urls.py b/device/urls.py index 7fc0e96..3d0b368 100644 --- a/device/urls.py +++ b/device/urls.py @@ -5,7 +5,7 @@ app_name = 'device' urlpatterns = [ path("add/", views.NewDeviceView.as_view(), name="add"), - path("edit//", views.EditDeviceView.as_view(), name="edit"), - path("/", views.DetailsView.as_view(), name="details"), - path("/annotation/add", views.AddAnnotationView.as_view(), name="add_annotation"), + path("edit//", views.EditDeviceView.as_view(), name="edit"), + path("/", views.DetailsView.as_view(), name="details"), + path("/annotation/add", views.AddAnnotationView.as_view(), name="add_annotation"), ] diff --git a/device/views.py b/device/views.py index 00255dd..012c878 100644 --- a/device/views.py +++ b/device/views.py @@ -6,21 +6,23 @@ from django.utils.translation import gettext_lazy as _ from django.views.generic.edit import ( CreateView, UpdateView, + FormView, ) from django.views.generic.base import TemplateView -from dashboard.mixins import DashboardView, DetailsMixin +from dashboard.mixins import DashboardView from snapshot.models import Annotation from snapshot.xapian import search from lot.models import LotTag from device.models import Device +from device.forms import DeviceFormSet -class NewDeviceView(DashboardView, CreateView): +class NewDeviceView(DashboardView, FormView): template_name = "new_device.html" title = _("New Device") breadcrumb = "Device / New Device" success_url = reverse_lazy('dashboard:unassigned_devices') - model = Device + form_class = DeviceFormSet def form_valid(self, form): form.instance.owner = self.request.user @@ -28,12 +30,41 @@ class NewDeviceView(DashboardView, CreateView): return response +# class AddToLotView(DashboardView, FormView): +# template_name = "list_lots.html" +# title = _("Add to lots") +# breadcrumb = "lot / add to lots" +# success_url = reverse_lazy('dashboard:unassigned_devices') +# form_class = LotsForm + +# def get_context_data(self, **kwargs): +# context = super().get_context_data(**kwargs) +# lots = Lot.objects.filter(owner=self.request.user) +# lot_tags = LotTag.objects.filter(owner=self.request.user) +# context.update({ +# 'lots': lots, +# 'lot_tags':lot_tags, +# }) +# return context + +# def get_form(self): +# form = super().get_form() +# form.fields["lots"].queryset = Lot.objects.filter(owner=self.request.user) +# return form + +# def form_valid(self, form): +# form.devices = self.get_session_devices() +# form.save() +# response = super().form_valid(form) +# return response + + class EditDeviceView(DashboardView, UpdateView): template_name = "new_device.html" title = _("Update Device") breadcrumb = "Device / Update Device" success_url = reverse_lazy('dashboard:unassigned_devices') - model = Device + model = Annotation def get_form_kwargs(self): pk = self.kwargs.get('pk') @@ -43,17 +74,23 @@ class EditDeviceView(DashboardView, UpdateView): return kwargs -class DetailsView(DetailsMixin): +class DetailsView(DashboardView, TemplateView): template_name = "details.html" title = _("Device") breadcrumb = "Device / Details" - model = Device + model = Annotation + + def get(self, request, *args, **kwargs): + self.pk = kwargs['pk'] + self.object = Device(id=self.pk) + return super().get(request, *args, **kwargs) def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) self.object.initial() lot_tags = LotTag.objects.filter(owner=self.request.user) context.update({ + 'object': self.object, 'snapshot': self.object.get_last_snapshot(), 'lot_tags': lot_tags, }) diff --git a/lot/forms.py b/lot/forms.py index 0065960..7b809fd 100644 --- a/lot/forms.py +++ b/lot/forms.py @@ -15,13 +15,14 @@ class LotsForm(forms.Form): def save(self, commit=True): if not commit: return + for dev in self.devices: for lot in self._lots: - lot.devices.add(dev.id) + lot.add(dev.id) return def remove(self): for dev in self.devices: for lot in self._lots: - lot.devices.remove(dev.id) + lot.remove(dev.id) return diff --git a/lot/migrations/0002_remove_lot_devices_devicelot.py b/lot/migrations/0002_remove_lot_devices_devicelot.py new file mode 100644 index 0000000..8496dea --- /dev/null +++ b/lot/migrations/0002_remove_lot_devices_devicelot.py @@ -0,0 +1,39 @@ +# Generated by Django 5.0.6 on 2024-07-18 17:30 + +import django.db.models.deletion +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ("lot", "0001_initial"), + ] + + operations = [ + migrations.RemoveField( + model_name="lot", + name="devices", + ), + migrations.CreateModel( + name="DeviceLot", + fields=[ + ( + "id", + models.BigAutoField( + auto_created=True, + primary_key=True, + serialize=False, + verbose_name="ID", + ), + ), + ("device_id", models.CharField(max_length=256)), + ( + "lot", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="lot.lot" + ), + ), + ], + ), + ] diff --git a/lot/models.py b/lot/models.py index fb34778..9237a27 100644 --- a/lot/models.py +++ b/lot/models.py @@ -7,7 +7,7 @@ from utils.constants import ( ) from user.models import User -from device.models import Device +# from device.models import Device from snapshot.models import Annotation @@ -19,6 +19,11 @@ class LotTag(models.Model): return self.name +class DeviceLot(models.Model): + lot = models.ForeignKey("Lot", on_delete=models.CASCADE) + device_id = models.CharField(max_length=STR_EXTEND_SIZE, blank=False, null=False) + + class Lot(models.Model): created = models.DateTimeField(auto_now_add=True) updated = models.DateTimeField(auto_now=True) @@ -28,4 +33,13 @@ class Lot(models.Model): closed = models.BooleanField(default=True) owner = models.ForeignKey(User, on_delete=models.CASCADE) type = models.ForeignKey(LotTag, on_delete=models.CASCADE) - devices = models.ManyToManyField(Device) + + def add(self, v): + if DeviceLot.objects.filter(lot=self, device_id=v).exists(): + return + DeviceLot.objects.create(lot=self, device_id=v) + + def remove(self, v): + for d in DeviceLot.objects.filter(lot=self, device_id=v): + d.delete() + diff --git a/lot/views.py b/lot/views.py index 44d7266..9549076 100644 --- a/lot/views.py +++ b/lot/views.py @@ -6,7 +6,7 @@ from django.views.generic.edit import ( CreateView, DeleteView, UpdateView, - FormView + FormView, ) from dashboard.mixins import DashboardView from lot.models import Lot, LotTag diff --git a/snapshot/migrations/0002_remove_annotation_device.py b/snapshot/migrations/0002_remove_annotation_device.py new file mode 100644 index 0000000..326eca2 --- /dev/null +++ b/snapshot/migrations/0002_remove_annotation_device.py @@ -0,0 +1,17 @@ +# Generated by Django 5.0.6 on 2024-07-18 17:30 + +from django.db import migrations + + +class Migration(migrations.Migration): + + dependencies = [ + ("snapshot", "0001_initial"), + ] + + operations = [ + migrations.RemoveField( + model_name="annotation", + name="device", + ), + ] diff --git a/snapshot/models.py b/snapshot/models.py index bb6e40e..7c1a942 100644 --- a/snapshot/models.py +++ b/snapshot/models.py @@ -63,7 +63,6 @@ class Annotation(models.Model): type = models.SmallIntegerField(choices=Type) key = models.CharField(max_length=STR_EXTEND_SIZE) value = models.CharField(max_length=STR_EXTEND_SIZE) - device = models.ForeignKey('device.Device', on_delete=models.CASCADE) class Meta: constraints = [ diff --git a/snapshot/parse.py b/snapshot/parse.py index ea7e0a5..8137a86 100644 --- a/snapshot/parse.py +++ b/snapshot/parse.py @@ -7,7 +7,6 @@ import hashlib from datetime import datetime from snapshot.xapian import search, index from snapshot.models import Snapshot, Annotation -from device.models import Device from utils.constants import ALGOS @@ -47,21 +46,10 @@ class Build: value = algorithms['hidalgo1'] ).first() - if annotation: - device = annotation.device - else: - device = Device.objects.create( - type=self.json["device"]["type"], - manufacturer=self.json["device"]["manufacturer"], - model=self.json["device"]["model"], - owner=self.user - ) - for k, v in algorithms.items(): Annotation.objects.create( uuid=self.uuid, owner=self.user, - device=device, type=Annotation.Type.SYSTEM, key=k, value=v diff --git a/snapshot/views.py b/snapshot/views.py index b49615c..6a7f98b 100644 --- a/snapshot/views.py +++ b/snapshot/views.py @@ -1,10 +1,10 @@ from django.utils.translation import gettext_lazy as _ from django.views.generic.base import TemplateView -from django.views.generic.edit import FormView from django.urls import reverse_lazy from django.views.generic.edit import ( CreateView, UpdateView, + FormView, ) from dashboard.mixins import DashboardView