From 029e69495821728db76b028c835fce5a78efa4c8 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Wed, 18 Sep 2024 18:01:46 +0200 Subject: [PATCH 01/23] add institutions and get new worbench --- dashboard/mixins.py | 8 +-- dashboard/views.py | 2 +- device/models.py | 17 +++---- device/views.py | 16 +++--- evidence/forms.py | 2 +- .../migrations/0004_alter_annotation_owner.py | 22 +++++++++ evidence/models.py | 41 ++++++++++++++-- evidence/parse.py | 49 +++++++++++++++---- evidence/views.py | 4 +- ...wner_alter_lotannotation_owner_and_more.py | 36 ++++++++++++++ lot/models.py | 10 ++-- lot/views.py | 28 +++++------ user/management/commands/add_institution.py | 17 ++++++- user/management/commands/add_user.py | 32 ++++-------- user/views.py | 5 +- utils/constants.py | 18 +++++++ utils/device.py | 2 +- 17 files changed, 224 insertions(+), 85 deletions(-) create mode 100644 evidence/migrations/0004_alter_annotation_owner.py create mode 100644 lot/migrations/0003_alter_lot_owner_alter_lotannotation_owner_and_more.py diff --git a/dashboard/mixins.py b/dashboard/mixins.py index c28dcf8..7e2e236 100644 --- a/dashboard/mixins.py +++ b/dashboard/mixins.py @@ -39,16 +39,16 @@ class DashboardView(LoginRequiredMixin): 'section': self.section, 'path': resolve(self.request.path).url_name, 'user': self.request.user, - 'lot_tags': LotTag.objects.filter(owner=self.request.user) + 'lot_tags': LotTag.objects.filter(owner=self.request.user.institution) }) return context def get_session_devices(self): - # import pdb; pdb.set_trace() dev_ids = self.request.session.pop("devices", []) self._devices = [] - for x in Annotation.objects.filter(value__in=dev_ids).filter(owner=self.request.user).distinct(): + annotation = Annotation.objects.filter(value__in=dev_ids) + for x in annotation.filter(owner=self.request.user.institution).distinct(): self._devices.append(Device(id=x.value)) return self._devices @@ -57,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, owner=self.request.user) + self.object = get_object_or_404(self.model, pk=self.pk, owner=self.request.user.institution) return super().get(request, *args, **kwargs) def get_context_data(self, **kwargs): diff --git a/dashboard/views.py b/dashboard/views.py index 79b9346..bcc4eeb 100644 --- a/dashboard/views.py +++ b/dashboard/views.py @@ -12,7 +12,7 @@ class UnassignedDevicesView(InventaryMixin): def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) - devices = Device.get_unassigned(self.request.user) + devices = Device.get_unassigned(self.request.user.institution) context.update({ 'devices': devices, diff --git a/device/models.py b/device/models.py index f1b1ee7..6e0bc50 100644 --- a/device/models.py +++ b/device/models.py @@ -113,10 +113,10 @@ class Device: self.lots = [x.lot for x in DeviceLot.objects.filter(device_id=self.id)] @classmethod - def get_unassigned(cls, user): - chids = DeviceLot.objects.filter(lot__owner=user).values_list("device_id", flat=True).distinct() + def get_unassigned(cls, institution): + chids = DeviceLot.objects.filter(lot__owner=institution).values_list("device_id", flat=True).distinct() annotations = Annotation.objects.filter( - owner=user, + owner=institution, type=Annotation.Type.SYSTEM, ).exclude(value__in=chids).values_list("value", flat=True).distinct() return [cls(id=x) for x in annotations] @@ -141,22 +141,17 @@ class Device: def manufacturer(self): if not self.last_evidence: self.get_last_evidence() - return self.last_evidence.doc['device']['manufacturer'] + return self.last_evidence.get_manufacturer() @property def type(self): if not self.last_evidence: self.get_last_evidence() - return self.last_evidence.doc['device']['type'] + return self.last_evidence.get_chassis() @property def model(self): if not self.last_evidence: self.get_last_evidence() - return self.last_evidence.doc['device']['model'] + return self.last_evidence.get_model() - @property - def type(self): - if not self.last_evidence: - self.get_last_evidence() - return self.last_evidence.doc['device']['type'] diff --git a/device/views.py b/device/views.py index e89cc3c..0fd9365 100644 --- a/device/views.py +++ b/device/views.py @@ -92,7 +92,7 @@ class DetailsView(DashboardView, TemplateView): def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) self.object.initial() - lot_tags = LotTag.objects.filter(owner=self.request.user) + lot_tags = LotTag.objects.filter(owner=self.request.user.institution) context.update({ 'object': self.object, 'snapshot': self.object.get_last_evidence(), @@ -110,7 +110,7 @@ class AddAnnotationView(DashboardView, CreateView): fields = ("key", "value") def form_valid(self, form): - form.instance.owner = self.request.user + form.instance.owner = self.request.user.institution form.instance.uuid = self.annotation.uuid form.instance.type = Annotation.Type.USER response = super().form_valid(form) @@ -118,11 +118,12 @@ class AddAnnotationView(DashboardView, CreateView): def get_form_kwargs(self): pk = self.kwargs.get('pk') + institution = self.request.user.institution self.annotation = Annotation.objects.filter( - owner=self.request.user, value=pk, type=Annotation.Type.SYSTEM + owner=institution, value=pk, type=Annotation.Type.SYSTEM ).first() if not self.annotation: - get_object_or_404(Annotation, pk=0, owner=self.request.user) + get_object_or_404(Annotation, pk=0, owner=institution) self.success_url = reverse_lazy('device:details', args=[pk]) kwargs = super().get_form_kwargs() return kwargs @@ -137,7 +138,7 @@ class AddDocumentView(DashboardView, CreateView): fields = ("key", "value") def form_valid(self, form): - form.instance.owner = self.request.user + form.instance.owner = self.request.user.institution form.instance.uuid = self.annotation.uuid form.instance.type = Annotation.Type.DOCUMENT response = super().form_valid(form) @@ -145,11 +146,12 @@ class AddDocumentView(DashboardView, CreateView): def get_form_kwargs(self): pk = self.kwargs.get('pk') + institution = self.request.user.institution self.annotation = Annotation.objects.filter( - owner=self.request.user, value=pk, type=Annotation.Type.SYSTEM + owner=institution, value=pk, type=Annotation.Type.SYSTEM ).first() if not self.annotation: - get_object_or_404(Annotation, pk=0, owner=self.request.user) + get_object_or_404(Annotation, pk=0, owner=institution) self.success_url = reverse_lazy('device:details', args=[pk]) kwargs = super().get_form_kwargs() return kwargs diff --git a/evidence/forms.py b/evidence/forms.py index ae2630d..be50ca8 100644 --- a/evidence/forms.py +++ b/evidence/forms.py @@ -71,7 +71,7 @@ class UserTagForm(forms.Form): Annotation.objects.create( uuid=self.uuid, - owner=user, + owner=user.institution, type=Annotation.Type.SYSTEM, key='CUSTOM_ID', value=self.tag diff --git a/evidence/migrations/0004_alter_annotation_owner.py b/evidence/migrations/0004_alter_annotation_owner.py new file mode 100644 index 0000000..26c9bda --- /dev/null +++ b/evidence/migrations/0004_alter_annotation_owner.py @@ -0,0 +1,22 @@ +# Generated by Django 5.0.6 on 2024-09-18 10:55 + +import django.db.models.deletion +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ("evidence", "0003_alter_annotation_type"), + ("user", "0001_initial"), + ] + + operations = [ + migrations.AlterField( + model_name="annotation", + name="owner", + field=models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="user.institution" + ), + ), + ] diff --git a/evidence/models.py b/evidence/models.py index dbb47a0..30625aa 100644 --- a/evidence/models.py +++ b/evidence/models.py @@ -1,10 +1,11 @@ import json +from dmidecode import DMIParse from django.db import models -from utils.constants import STR_SM_SIZE, STR_EXTEND_SIZE +from utils.constants import STR_SM_SIZE, STR_EXTEND_SIZE, CHASSIS_DH from evidence.xapian import search -from user.models import User +from user.models import Institution class Annotation(models.Model): @@ -15,7 +16,7 @@ class Annotation(models.Model): created = models.DateTimeField(auto_now_add=True) uuid = models.UUIDField() - owner = models.ForeignKey(User, on_delete=models.CASCADE) + owner = models.ForeignKey(Institution, on_delete=models.CASCADE) type = models.SmallIntegerField(choices=Type) key = models.CharField(max_length=STR_EXTEND_SIZE) value = models.CharField(max_length=STR_EXTEND_SIZE) @@ -32,6 +33,7 @@ class Evidence: self.owner = None self.doc = None self.created = None + self.dmi = None self.annotations = [] self.get_owner() @@ -58,6 +60,11 @@ class Evidence: for xa in matches: self.doc = json.loads(xa.document.get_data()) + + if self.doc.get("software") == "EreuseWorkbench": + dmidecode_raw = self.doc["data"]["dmidecode"] + self.dmi = DMIParse(dmidecode_raw) + def get_time(self): if not self.doc: @@ -70,9 +77,35 @@ class Evidence: def components(self): return self.doc.get('components', []) + def get_manufacturer(self): + if self.doc.get("software") != "EreuseWorkbench": + return self.doc['device']['manufacturer'] + + return self.dmi.manufacturer().strip() + + def get_model(self): + if self.doc.get("software") != "EreuseWorkbench": + return self.doc['device']['model'] + + return self.dmi.model().strip() + + def get_chassis(self): + if self.doc.get("software") != "EreuseWorkbench": + return self.doc['device']['model'] + + chassis = self.dmi.get("Chassis")[0].get("Type", '_virtual') + lower_type = chassis.lower() + + for k, v in CHASSIS_DH.items(): + if lower_type in v: + return k + return "" + + + @classmethod def get_all(cls, user): return Annotation.objects.filter( - owner=user, + owner=user.institution, type=Annotation.Type.SYSTEM, ).order_by("-created").values_list("uuid", flat=True).distinct() diff --git a/evidence/parse.py b/evidence/parse.py index 614c2ec..d9cfeda 100644 --- a/evidence/parse.py +++ b/evidence/parse.py @@ -4,9 +4,10 @@ import shutil import hashlib from datetime import datetime +from dmidecode import DMIParse from evidence.xapian import search, index from evidence.models import Evidence, Annotation -from utils.constants import ALGOS +from utils.constants import ALGOS, CHASSIS_DH class Build: @@ -33,13 +34,17 @@ class Build: } def get_hid_14(self): - device = self.json['device'] - manufacturer = device.get("manufacturer", '') - model = device.get("model", '') - chassis = device.get("chassis", '') - serial_number = device.get("serialNumber", '') - sku = device.get("sku", '') - hid = f"{manufacturer}{model}{chassis}{serial_number}{sku}" + if self.json.get("software") == "EreuseWorkbench": + hid = self.get_hid(self.json) + else: + device = self.json['device'] + manufacturer = device.get("manufacturer", '') + model = device.get("model", '') + chassis = device.get("chassis", '') + serial_number = device.get("serialNumber", '') + sku = device.get("sku", '') + hid = f"{manufacturer}{model}{chassis}{serial_number}{sku}" + return hashlib.sha3_256(hid.encode()).hexdigest() def create_annotations(self): @@ -47,8 +52,34 @@ class Build: for k, v in self.algorithms.items(): Annotation.objects.create( uuid=self.uuid, - owner=self.user, + owner=self.user.institution, type=Annotation.Type.SYSTEM, key=k, value=v ) + + def get_chassis_dh(self): + chassis = self.get_chassis() + lower_type = chassis.lower() + for k, v in CHASSIS_DH.items(): + if lower_type in v: + return k + return self.default + + def get_sku(self): + return self.dmi.get("System")[0].get("SKU Number", "n/a").strip() + + def get_chassis(self): + return self.dmi.get("Chassis")[0].get("Type", '_virtual') + + def get_hid(self, snapshot): + dmidecode_raw = snapshot["data"]["dmidecode"] + self.dmi = DMIParse(dmidecode_raw) + + manufacturer = self.dmi.manufacturer().strip() + model = self.dmi.model().strip() + chassis = self.get_chassis_dh() + serial_number = self.dmi.serial_number() + sku = self.get_sku() + + return f"{manufacturer}{model}{chassis}{serial_number}{sku}" diff --git a/evidence/views.py b/evidence/views.py index 1e02c7e..d98bddb 100644 --- a/evidence/views.py +++ b/evidence/views.py @@ -89,7 +89,7 @@ class EvidenceView(DashboardView, FormView): def get(self, request, *args, **kwargs): self.pk = kwargs['pk'] self.object = Evidence(self.pk) - if self.object.owner != self.request.user: + if self.object.owner != self.request.user.institution: raise Http403 self.object.get_annotations() @@ -127,7 +127,7 @@ class DownloadEvidenceView(DashboardView, TemplateView): def get(self, request, *args, **kwargs): pk = kwargs['pk'] evidence = Evidence(pk) - if evidence.owner != self.request.user: + if evidence.owner != self.request.user.institution: raise Http403() evidence.get_doc() diff --git a/lot/migrations/0003_alter_lot_owner_alter_lotannotation_owner_and_more.py b/lot/migrations/0003_alter_lot_owner_alter_lotannotation_owner_and_more.py new file mode 100644 index 0000000..806c6f1 --- /dev/null +++ b/lot/migrations/0003_alter_lot_owner_alter_lotannotation_owner_and_more.py @@ -0,0 +1,36 @@ +# Generated by Django 5.0.6 on 2024-09-18 10:55 + +import django.db.models.deletion +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ("lot", "0002_lotannotation"), + ("user", "0001_initial"), + ] + + operations = [ + migrations.AlterField( + model_name="lot", + name="owner", + field=models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="user.institution" + ), + ), + migrations.AlterField( + model_name="lotannotation", + name="owner", + field=models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="user.institution" + ), + ), + migrations.AlterField( + model_name="lottag", + name="owner", + field=models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="user.institution" + ), + ), + ] diff --git a/lot/models.py b/lot/models.py index 9078459..3926f29 100644 --- a/lot/models.py +++ b/lot/models.py @@ -6,14 +6,14 @@ from utils.constants import ( STR_EXTEND_SIZE, ) -from user.models import User +from user.models import Institution # from device.models import Device -from evidence.models import Annotation +# from evidence.models import Annotation class LotTag(models.Model): name = models.CharField(max_length=STR_SIZE, blank=False, null=False) - owner = models.ForeignKey(User, on_delete=models.CASCADE) + owner = models.ForeignKey(Institution, on_delete=models.CASCADE) def __str__(self): return self.name @@ -31,7 +31,7 @@ class Lot(models.Model): code = models.CharField(max_length=STR_SIZE, blank=True, null=True) description = models.CharField(max_length=STR_SIZE, blank=True, null=True) closed = models.BooleanField(default=True) - owner = models.ForeignKey(User, on_delete=models.CASCADE) + owner = models.ForeignKey(Institution, on_delete=models.CASCADE) type = models.ForeignKey(LotTag, on_delete=models.CASCADE) def add(self, v): @@ -52,7 +52,7 @@ class LotAnnotation(models.Model): created = models.DateTimeField(auto_now_add=True) lot = models.ForeignKey(Lot, on_delete=models.CASCADE) - owner = models.ForeignKey(User, on_delete=models.CASCADE) + owner = models.ForeignKey(Institution, on_delete=models.CASCADE) type = models.SmallIntegerField(choices=Type) key = models.CharField(max_length=STR_EXTEND_SIZE) value = models.CharField(max_length=STR_EXTEND_SIZE) diff --git a/lot/views.py b/lot/views.py index 280450e..46252c8 100644 --- a/lot/views.py +++ b/lot/views.py @@ -28,7 +28,7 @@ class NewLotView(DashboardView, CreateView): ) def form_valid(self, form): - form.instance.owner = self.request.user + form.instance.owner = self.request.user.institution response = super().form_valid(form) return response @@ -83,8 +83,8 @@ class AddToLotView(DashboardView, FormView): 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) + lots = Lot.objects.filter(owner=self.request.user.institution) + lot_tags = LotTag.objects.filter(owner=self.request.user.institution) context.update({ 'lots': lots, 'lot_tags':lot_tags, @@ -93,7 +93,7 @@ class AddToLotView(DashboardView, FormView): def get_form(self): form = super().get_form() - form.fields["lots"].queryset = Lot.objects.filter(owner=self.request.user) + form.fields["lots"].queryset = Lot.objects.filter(owner=self.request.user.institution) return form def form_valid(self, form): @@ -123,10 +123,10 @@ class LotsTagsView(DashboardView, TemplateView): def get_context_data(self, **kwargs): self.pk = kwargs.get('pk') context = super().get_context_data(**kwargs) - tag = get_object_or_404(LotTag, owner=self.request.user, id=self.pk) + tag = get_object_or_404(LotTag, owner=self.request.user.institution, id=self.pk) self.title += " {}".format(tag.name) self.breadcrumb += " {}".format(tag.name) - lots = Lot.objects.filter(owner=self.request.user).filter(type=tag) + lots = Lot.objects.filter(owner=self.request.user.institution).filter(type=tag) context.update({ 'lots': lots, 'title': self.title, @@ -144,7 +144,7 @@ class LotAddDocumentView(DashboardView, CreateView): fields = ("key", "value") def form_valid(self, form): - form.instance.owner = self.request.user + form.instance.owner = self.request.user.institution form.instance.lot = self.lot form.instance.type = LotAnnotation.Type.DOCUMENT response = super().form_valid(form) @@ -152,7 +152,7 @@ class LotAddDocumentView(DashboardView, CreateView): def get_form_kwargs(self): pk = self.kwargs.get('pk') - self.lot = get_object_or_404(Lot, pk=pk, owner=self.request.user) + self.lot = get_object_or_404(Lot, pk=pk, owner=self.request.user.institution) self.success_url = reverse_lazy('lot:documents', args=[pk]) kwargs = super().get_form_kwargs() return kwargs @@ -166,10 +166,10 @@ class LotDocumentsView(DashboardView, TemplateView): def get_context_data(self, **kwargs): self.pk = kwargs.get('pk') context = super().get_context_data(**kwargs) - lot = get_object_or_404(Lot, owner=self.request.user, id=self.pk) + lot = get_object_or_404(Lot, owner=self.request.user.institution, id=self.pk) documents = LotAnnotation.objects.filter( lot=lot, - owner=self.request.user, + owner=self.request.user.institution, type=LotAnnotation.Type.DOCUMENT, ) context.update({ @@ -189,10 +189,10 @@ class LotAnnotationsView(DashboardView, TemplateView): def get_context_data(self, **kwargs): self.pk = kwargs.get('pk') context = super().get_context_data(**kwargs) - lot = get_object_or_404(Lot, owner=self.request.user, id=self.pk) + lot = get_object_or_404(Lot, owner=self.request.user.institution, id=self.pk) annotations = LotAnnotation.objects.filter( lot=lot, - owner=self.request.user, + owner=self.request.user.institution, type=LotAnnotation.Type.USER, ) context.update({ @@ -213,7 +213,7 @@ class LotAddAnnotationView(DashboardView, CreateView): fields = ("key", "value") def form_valid(self, form): - form.instance.owner = self.request.user + form.instance.owner = self.request.user.institution form.instance.lot = self.lot form.instance.type = LotAnnotation.Type.USER response = super().form_valid(form) @@ -221,7 +221,7 @@ class LotAddAnnotationView(DashboardView, CreateView): def get_form_kwargs(self): pk = self.kwargs.get('pk') - self.lot = get_object_or_404(Lot, pk=pk, owner=self.request.user) + self.lot = get_object_or_404(Lot, pk=pk, owner=self.request.user.institution) self.success_url = reverse_lazy('lot:annotations', args=[pk]) kwargs = super().get_form_kwargs() return kwargs diff --git a/user/management/commands/add_institution.py b/user/management/commands/add_institution.py index 622dff8..b7a4a26 100644 --- a/user/management/commands/add_institution.py +++ b/user/management/commands/add_institution.py @@ -1,6 +1,6 @@ from django.core.management.base import BaseCommand from user.models import Institution - +from lot.models import LotTag class Command(BaseCommand): help = "Create a new Institution" @@ -9,4 +9,17 @@ class Command(BaseCommand): parser.add_argument('name', type=str, help='institution') def handle(self, *args, **kwargs): - Institution.objects.create(name=kwargs['name']) + self.institution = Institution.objects.create(name=kwargs['name']) + self.create_lot_tags() + + def create_lot_tags(self): + tags = [ + "Entrada", + "Salida", + "Temporal" + ] + for tag in tags: + LotTag.objects.create( + name=tag, + owner=self.institution + ) diff --git a/user/management/commands/add_user.py b/user/management/commands/add_user.py index 7d31ab7..499adfa 100644 --- a/user/management/commands/add_user.py +++ b/user/management/commands/add_user.py @@ -1,6 +1,5 @@ from django.core.management.base import BaseCommand from django.contrib.auth import get_user_model -from lot.models import LotTag from user.models import Institution @@ -16,29 +15,16 @@ class Command(BaseCommand): parser.add_argument('password', type=str, help='password') def handle(self, *args, **kwargs): - email = kwargs['email'] - password = kwargs['password'] - institution = Institution.objects.get(name=kwargs['institution']) - self.create_user(institution, email, password) - self.create_lot_tags() + self.email = kwargs['email'] + self.password = kwargs['password'] + self.institution = Institution.objects.get(name=kwargs['institution']) + self.create_user() - def create_user(self, institution, email, password): + def create_user(self): self.u = User.objects.create( - institution=institution, - email=email, - password=password + institution=self.institution, + email=self.email, + password=self.password ) - self.u.set_password(password) + self.u.set_password(self.password) self.u.save() - - def create_lot_tags(self): - tags = [ - "Entrada", - "Salida", - "Temporal" - ] - for tag in tags: - LotTag.objects.create( - name=tag, - owner=self.u - ) diff --git a/user/views.py b/user/views.py index 91ea44a..e7f3289 100644 --- a/user/views.py +++ b/user/views.py @@ -1,3 +1,6 @@ from django.shortcuts import render +from django.utils.translation import gettext_lazy as _ +from dashboard.mixins import InventaryMixin, DetailsMixin + + -# Create your views here. diff --git a/utils/constants.py b/utils/constants.py index 2fd9a13..e481d6c 100644 --- a/utils/constants.py +++ b/utils/constants.py @@ -20,3 +20,21 @@ HID_ALGO1 = [ ALGOS = { "hidalgo1": HID_ALGO1, } + + +CHASSIS_DH = { + 'Tower': {'desktop', 'low-profile', 'tower', 'server'}, + 'Docking': {'docking'}, + 'AllInOne': {'all-in-one'}, + 'Microtower': {'mini-tower', 'space-saving', 'mini'}, + 'PizzaBox': {'pizzabox'}, + 'Lunchbox': {'lunchbox'}, + 'Stick': {'stick'}, + 'Netbook': {'notebook', 'sub-notebook'}, + 'Handheld': {'handheld'}, + 'Laptop': {'portable', 'laptop'}, + 'Convertible': {'convertible'}, + 'Detachable': {'detachable'}, + 'Tablet': {'tablet'}, + 'Virtual': {'_virtual'}, +} diff --git a/utils/device.py b/utils/device.py index 2e41597..db9f0ce 100644 --- a/utils/device.py +++ b/utils/device.py @@ -69,7 +69,7 @@ def create_annotation(doc, user, commit=False): data = { 'uuid': doc['uuid'], - 'owner': user, + 'owner': user.institution, 'type': Annotation.Type.SYSTEM, 'key': 'CUSTOMER_ID', 'value': doc['CUSTOMER_ID'], -- 2.30.2 From 13c78fb3c1cf3001c01ec903d66882d1b228d8cf Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Thu, 19 Sep 2024 18:22:17 +0200 Subject: [PATCH 02/23] api --- dashboard/templates/base.html | 6 ++++++ dhub/settings.py | 2 ++ dhub/urls.py | 1 + evidence/forms.py | 9 +++++++++ requirements.txt | 3 ++- 5 files changed, 20 insertions(+), 1 deletion(-) diff --git a/dashboard/templates/base.html b/dashboard/templates/base.html index 3d046f1..f4a9307 100644 --- a/dashboard/templates/base.html +++ b/dashboard/templates/base.html @@ -143,6 +143,12 @@ +
  • + + + {% trans 'Tokens' %} + +
    • diff --git a/dhub/settings.py b/dhub/settings.py index 055828a..6e83032 100644 --- a/dhub/settings.py +++ b/dhub/settings.py @@ -43,6 +43,7 @@ INSTALLED_APPS = [ "django.contrib.staticfiles", 'django_extensions', 'django_bootstrap5', + 'django_tables2', "rest_framework", "login", "user", @@ -53,6 +54,7 @@ INSTALLED_APPS = [ "lot", "documents", "dashboard", + "api", ] diff --git a/dhub/urls.py b/dhub/urls.py index 90252d9..9bbd687 100644 --- a/dhub/urls.py +++ b/dhub/urls.py @@ -24,4 +24,5 @@ urlpatterns = [ path("evidence/", include("evidence.urls")), path("device/", include("device.urls")), path("lot/", include("lot.urls")), + path('api/', include('api.urls')), ] diff --git a/evidence/forms.py b/evidence/forms.py index be50ca8..d392f6e 100644 --- a/evidence/forms.py +++ b/evidence/forms.py @@ -56,6 +56,15 @@ class UserTagForm(forms.Form): def __init__(self, *args, **kwargs): self.uuid = kwargs.pop('uuid', None) + annotation = Annotation.objects.filter( + uuid=self.uuid, + type=Annotation.Type.SYSTEM, + key='CUSTOM_ID', + ).first() + + if annotation: + kwargs['initial'].update({'tag': annotation.value}) + super().__init__(*args, **kwargs) def clean(self): diff --git a/requirements.txt b/requirements.txt index 9d4492f..217d120 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,9 +3,10 @@ Django==5.0.6 django-bootstrap5==24.2 django-extensions==3.2.3 djangorestframework==3.15.1 +django-tables2==2.6.0 python-decouple==3.3 py-dmidecode==0.1.3 pandas==2.2.2 xlrd==2.0.1 odfpy==1.4.1 - +pytz==2024.2 -- 2.30.2 From ad59484b035d9ad5c0cb6ba96b534feff189d198 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Thu, 19 Sep 2024 18:35:26 +0200 Subject: [PATCH 03/23] api dir --- api/__init__.py | 0 api/admin.py | 3 + api/apps.py | 6 ++ api/forms.py | 1 + api/migrations/0001_initial.py | 39 +++++++++++ api/migrations/__init__.py | 0 api/models.py | 9 +++ api/tables.py | 67 ++++++++++++++++++ api/templates/custom_table.html | 100 +++++++++++++++++++++++++++ api/templates/token.html | 14 ++++ api/urls.py | 13 ++++ api/views.py | 116 ++++++++++++++++++++++++++++++++ 12 files changed, 368 insertions(+) create mode 100644 api/__init__.py create mode 100644 api/admin.py create mode 100644 api/apps.py create mode 100644 api/forms.py create mode 100644 api/migrations/0001_initial.py create mode 100644 api/migrations/__init__.py create mode 100644 api/models.py create mode 100644 api/tables.py create mode 100644 api/templates/custom_table.html create mode 100644 api/templates/token.html create mode 100644 api/urls.py create mode 100644 api/views.py diff --git a/api/__init__.py b/api/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/api/admin.py b/api/admin.py new file mode 100644 index 0000000..8c38f3f --- /dev/null +++ b/api/admin.py @@ -0,0 +1,3 @@ +from django.contrib import admin + +# Register your models here. diff --git a/api/apps.py b/api/apps.py new file mode 100644 index 0000000..66656fd --- /dev/null +++ b/api/apps.py @@ -0,0 +1,6 @@ +from django.apps import AppConfig + + +class ApiConfig(AppConfig): + default_auto_field = 'django.db.models.BigAutoField' + name = 'api' diff --git a/api/forms.py b/api/forms.py new file mode 100644 index 0000000..8b13789 --- /dev/null +++ b/api/forms.py @@ -0,0 +1 @@ + diff --git a/api/migrations/0001_initial.py b/api/migrations/0001_initial.py new file mode 100644 index 0000000..22c5a1f --- /dev/null +++ b/api/migrations/0001_initial.py @@ -0,0 +1,39 @@ +# Generated by Django 5.0.6 on 2024-09-19 15:09 + +import django.db.models.deletion +from django.conf import settings +from django.db import migrations, models + + +class Migration(migrations.Migration): + + initial = True + + dependencies = [ + migrations.swappable_dependency(settings.AUTH_USER_MODEL), + ] + + operations = [ + migrations.CreateModel( + name="Token", + fields=[ + ( + "id", + models.BigAutoField( + auto_created=True, + primary_key=True, + serialize=False, + verbose_name="ID", + ), + ), + ("token", models.UUIDField()), + ( + "owner", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, + to=settings.AUTH_USER_MODEL, + ), + ), + ], + ), + ] diff --git a/api/migrations/__init__.py b/api/migrations/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/api/models.py b/api/models.py new file mode 100644 index 0000000..b8bbc24 --- /dev/null +++ b/api/models.py @@ -0,0 +1,9 @@ +from django.db import models +from user.models import User + +# Create your models here. + + +class Token(models.Model): + token = models.UUIDField() + owner = models.ForeignKey(User, on_delete=models.CASCADE) diff --git a/api/tables.py b/api/tables.py new file mode 100644 index 0000000..ac1cc7a --- /dev/null +++ b/api/tables.py @@ -0,0 +1,67 @@ +import django_tables2 as tables +from django.utils.html import format_html +from django.utils.translation import gettext_lazy as _ + +from api.models import Token + + +class ButtonColumn(tables.Column): + attrs = { + "a": { + "type": "button", + "class": "text-danger", + "title": "Remove", + } + } + # it makes no sense to order a column of buttons + orderable = False + # django_tables will only call the render function if it doesn't find + # any empty values in the data, so we stop it from matching the data + # to any value considered empty + empty_values = () + + def render(self): + return format_html('') + + +class TokensTable(tables.Table): + delete = ButtonColumn( + verbose_name=_("Delete"), + linkify={ + "viewname": "api:delete_token", + "args": [tables.A("pk")] + }, + orderable=False + ) + + token = tables.Column(verbose_name=_("Token"), empty_values=()) + + def render_view_user(self): + return format_html('') + + # def render_token(self, record): + # return record.get_memberships() + + # def order_membership(self, queryset, is_descending): + # # TODO: Test that this doesn't return more rows than it should + # queryset = queryset.order_by( + # ("-" if is_descending else "") + "memberships__type" + # ) + + # return (queryset, True) + + # def render_role(self, record): + # return record.get_roles() + + # def order_role(self, queryset, is_descending): + # queryset = queryset.order_by( + # ("-" if is_descending else "") + "roles" + # ) + + # return (queryset, True) + + class Meta: + model = Token + template_name = "custom_table.html" + fields = ("token", "view_user") + diff --git a/api/templates/custom_table.html b/api/templates/custom_table.html new file mode 100644 index 0000000..496ddec --- /dev/null +++ b/api/templates/custom_table.html @@ -0,0 +1,100 @@ +{% load django_tables2 %} +{% load i18n %} +{% block table-wrapper %} +
    + {% block table %} + + {% block table.thead %} + {% if table.show_header %} + + + {% for column in table.columns %} + + {% endfor %} + + + {% endif %} + {% endblock table.thead %} + {% block table.tbody %} + + {% for row in table.paginated_rows %} + {% block table.tbody.row %} + + {% for column, cell in row.items %} + + {% endfor %} + + {% endblock table.tbody.row %} + {% empty %} + {% if table.empty_text %} + {% block table.tbody.empty_text %} + + {% endblock table.tbody.empty_text %} + {% endif %} + {% endfor %} + + {% endblock table.tbody %} + {% block table.tfoot %} + {% if table.has_footer %} + + + {% for column in table.columns %} + + {% endfor %} + + + {% endif %} + {% endblock table.tfoot %} +
    + {% if column.orderable %} + {{ column.header }} + {% else %} + {{ column.header }} + {% endif %} +
    {% if column.localize == None %}{{ cell }}{% else %}{% if column.localize %}{{ cell|localize }}{% else %}{{ cell|unlocalize }}{% endif %}{% endif %}
    {{ table.empty_text }}
    {{ column.footer }}
    + {% endblock table %} + + {% block pagination %} + {% if table.page and table.paginator.num_pages > 1 %} + + {% endif %} + {% endblock pagination %} +
    +{% endblock table-wrapper %} diff --git a/api/templates/token.html b/api/templates/token.html new file mode 100644 index 0000000..5185090 --- /dev/null +++ b/api/templates/token.html @@ -0,0 +1,14 @@ +{% extends "base.html" %} +{% load i18n %} +{% load render_table from django_tables2 %} + +{% block content %} +

    + + {{ subtitle }} +

    +{% render_table table %} +
    + {% translate "Generate a new token" %} +
    +{% endblock %} diff --git a/api/urls.py b/api/urls.py new file mode 100644 index 0000000..f3c8028 --- /dev/null +++ b/api/urls.py @@ -0,0 +1,13 @@ +from api import views + +from django.urls import path + + +app_name = 'api' + +urlpatterns = [ + path('snapshot/', views.NewSnapshot, name='new_snapshot'), + path('tokens/', views.TokenView.as_view(), name='tokens'), + path('tokens/new', views.TokenNewView.as_view(), name='new_token'), + path('tokens//del', views.TokenDeleteView.as_view(), name='delete_token'), +] diff --git a/api/views.py b/api/views.py new file mode 100644 index 0000000..5a119f9 --- /dev/null +++ b/api/views.py @@ -0,0 +1,116 @@ +import json + +from django.shortcuts import get_object_or_404, redirect +from django.utils.translation import gettext_lazy as _ +from django.views.decorators.csrf import csrf_exempt +from django.core.exceptions import ValidationError +from django.views.generic.edit import DeleteView +from django.views.generic.base import View +from django.http import JsonResponse +from django_tables2 import SingleTableView +from uuid import uuid4 + +from dashboard.mixins import DashboardView +from evidence.models import Annotation +from evidence.parse import Build +from user.models import User +from api.models import Token +from api.tables import TokensTable + + +def save_in_disk(data, user): + pass + + +@csrf_exempt +def NewSnapshot(request): + # Accept only posts + if request.method != 'POST': + return JsonResponse({'error': 'Invalid request method'}, status=400) + + # Authentication + # auth_header = request.headers.get('Authorization') + # if not auth_header or not auth_header.startswith('Bearer '): + # return JsonResponse({'error': 'Invalid or missing token'}, status=401) + + # token = auth_header.split(' ')[1] + # tk = Token.objects.filter(token=token).first() + # if not tk: + # return JsonResponse({'error': 'Invalid or missing token'}, status=401) + + # Validation snapshot + try: + data = json.loads(request.body) + except json.JSONDecodeError: + return JsonResponse({'error': 'Invalid JSON'}, status=400) + + # try: + # Build(data, None, check=True) + # except Exception: + # return JsonResponse({'error': 'Invalid Snapshot'}, status=400) + + exist_annotation = Annotation.objects.filter( + uuid=data['uuid'] + ).first() + + if exist_annotation: + raise ValidationError("error: the snapshot {} exist".format(data['uuid'])) + + # Process snapshot + # save_in_disk(data, tk.user) + + try: + # Build(data, tk.user) + user = User.objects.get(email="user@example.org") + Build(data, user) + except Exception: + return JsonResponse({'status': 'fail'}, status=200) + + return JsonResponse({'status': 'success'}, status=200) + + + + +class TokenView(DashboardView, SingleTableView): + template_name = "token.html" + title = _("Credential management") + section = "Credential" + subtitle = _('Managament Tokens') + icon = 'bi bi-key' + model = Token + table_class = TokensTable + + def get_queryset(self): + """ + Override the get_queryset method to filter events based on the user type. + """ + return Token.objects.filter().order_by("-id") + + def get_context_data(self, **kwargs): + context = super().get_context_data(**kwargs) + context.update({ + 'tokens': Token.objects, + }) + return context + + +class TokenDeleteView(DashboardView, DeleteView): + model = Token + + def get(self, request, *args, **kwargs): + # self.check_valid_user() + self.pk = kwargs['pk'] + self.object = get_object_or_404(self.model, pk=self.pk) + self.object.delete() + + return redirect('api:tokens') + + +class TokenNewView(DashboardView, View): + + def get(self, request, *args, **kwargs): + # self.check_valid_user() + Token.objects.create(token=uuid4()) + + return redirect('api:tokens') + -- 2.30.2 From a1ffdb5ed694b534f8a604851845a69ec46f513b Mon Sep 17 00:00:00 2001 From: pedro Date: Thu, 19 Sep 2024 14:05:28 -0300 Subject: [PATCH 04/23] docker: add_institution command --- docker/devicehub-django.entrypoint.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docker/devicehub-django.entrypoint.sh b/docker/devicehub-django.entrypoint.sh index 158406e..68e3760 100644 --- a/docker/devicehub-django.entrypoint.sh +++ b/docker/devicehub-django.entrypoint.sh @@ -21,7 +21,9 @@ deploy() { # inspired by https://medium.com/analytics-vidhya/django-with-docker-and-docker-compose-python-part-2-8415976470cc echo "INFO detected NEW deployment" ./manage.py migrate - ./manage.py add_user user@example.org 1234 + ./manage.py add_institution example-org + # TODO: one error on add_user, and you don't add user anymore + ./manage.py add_user example-org user@example.org 1234 fi } -- 2.30.2 From 2f4a85f79bf0ac2cd74a80afcd79ed4139392de4 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 20 Sep 2024 14:23:14 +0200 Subject: [PATCH 05/23] add interface for create tokens by user for user --- api/views.py | 4 +--- dashboard/templates/base.html | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/api/views.py b/api/views.py index 5a119f9..2aeb4de 100644 --- a/api/views.py +++ b/api/views.py @@ -98,7 +98,6 @@ class TokenDeleteView(DashboardView, DeleteView): model = Token def get(self, request, *args, **kwargs): - # self.check_valid_user() self.pk = kwargs['pk'] self.object = get_object_or_404(self.model, pk=self.pk) self.object.delete() @@ -109,8 +108,7 @@ class TokenDeleteView(DashboardView, DeleteView): class TokenNewView(DashboardView, View): def get(self, request, *args, **kwargs): - # self.check_valid_user() - Token.objects.create(token=uuid4()) + Token.objects.create(token=uuid4(), owner=self.request.user) return redirect('api:tokens') diff --git a/dashboard/templates/base.html b/dashboard/templates/base.html index f4a9307..318c890 100644 --- a/dashboard/templates/base.html +++ b/dashboard/templates/base.html @@ -144,7 +144,7 @@
  • - + {% trans 'Tokens' %} -- 2.30.2 From f5f0ea3c35a3c4d142117b9990860d6630785999 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 20 Sep 2024 14:30:31 +0200 Subject: [PATCH 06/23] fix delete token. only user tokens allowed --- api/views.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/views.py b/api/views.py index 2aeb4de..fbccb79 100644 --- a/api/views.py +++ b/api/views.py @@ -99,7 +99,7 @@ class TokenDeleteView(DashboardView, DeleteView): 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) self.object.delete() return redirect('api:tokens') -- 2.30.2 From fa59feaf773f7bf21e008ba6aeca46061eb6a79d Mon Sep 17 00:00:00 2001 From: pedro Date: Fri, 20 Sep 2024 12:09:20 -0300 Subject: [PATCH 07/23] docker: allow config of ALLOWED_HOSTS --- dhub/settings.py | 2 +- docker-compose.yml | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/dhub/settings.py b/dhub/settings.py index 6e83032..6ba2855 100644 --- a/dhub/settings.py +++ b/dhub/settings.py @@ -29,7 +29,7 @@ SECRET_KEY = "django-insecure-1p8rs@qf$$l^!vsbetagojw23kw@1ez(qi8^(s0t!wyh!l3 # SECURITY WARNING: don't run with debug turned on in production! DEBUG = True -ALLOWED_HOSTS = [] +ALLOWED_HOSTS = config('ALLOWED_HOSTS', default='[]', cast=Csv() # Application definition diff --git a/docker-compose.yml b/docker-compose.yml index b6e2d01..6226f1a 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -4,7 +4,8 @@ services: build: dockerfile: docker/devicehub-django.Dockerfile environment: - DEBUG: true + - DEBUG: true + - ALLOWED_HOSTS=* volumes: - .:/opt/devicehub-django ports: -- 2.30.2 From 1834aab36ecea6d88ca9e85f445d727a483f1e92 Mon Sep 17 00:00:00 2001 From: pedro Date: Fri, 20 Sep 2024 12:19:12 -0300 Subject: [PATCH 08/23] add docker-reset with the usual operations to restart everything --- docker-reset.sh | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100755 docker-reset.sh diff --git a/docker-reset.sh b/docker-reset.sh new file mode 100755 index 0000000..b6fc7ce --- /dev/null +++ b/docker-reset.sh @@ -0,0 +1,22 @@ +#!/bin/sh + +# Copyright (c) 2024 Pedro +# SPDX-License-Identifier: AGPL-3.0-or-later + +set -e +set -u +# DEBUG +#set -x + +main() { + # remove old database + sudo rm -v db/* + docker compose down + docker compose build + docker compose up +} + +main "${@}" + +# written in emacs +# -*- mode: shell-script; -*- -- 2.30.2 From d491fdd47e3129c965004556b70837f8ca8954ae Mon Sep 17 00:00:00 2001 From: pedro Date: Fri, 20 Sep 2024 12:20:38 -0300 Subject: [PATCH 09/23] docker-reset.sh: add force --- docker-reset.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker-reset.sh b/docker-reset.sh index b6fc7ce..d75a86c 100755 --- a/docker-reset.sh +++ b/docker-reset.sh @@ -10,7 +10,7 @@ set -u main() { # remove old database - sudo rm -v db/* + sudo rm -vf db/* docker compose down docker compose build docker compose up -- 2.30.2 From ecb0877b36ca61b95480742a33e8accf9d079c6a Mon Sep 17 00:00:00 2001 From: pedro Date: Fri, 20 Sep 2024 12:27:07 -0300 Subject: [PATCH 10/23] docker: bugfix to make it work --- dhub/settings.py | 2 +- docker-compose.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dhub/settings.py b/dhub/settings.py index 6ba2855..3770353 100644 --- a/dhub/settings.py +++ b/dhub/settings.py @@ -29,7 +29,7 @@ SECRET_KEY = "django-insecure-1p8rs@qf$$l^!vsbetagojw23kw@1ez(qi8^(s0t!wyh!l3 # SECURITY WARNING: don't run with debug turned on in production! DEBUG = True -ALLOWED_HOSTS = config('ALLOWED_HOSTS', default='[]', cast=Csv() +ALLOWED_HOSTS = config('ALLOWED_HOSTS', default='[]', cast=Csv()) # Application definition diff --git a/docker-compose.yml b/docker-compose.yml index 6226f1a..a106f6e 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -4,7 +4,7 @@ services: build: dockerfile: docker/devicehub-django.Dockerfile environment: - - DEBUG: true + - DEBUG=true - ALLOWED_HOSTS=* volumes: - .:/opt/devicehub-django -- 2.30.2 From b534d6975077d176787339af7b47430070f4848f Mon Sep 17 00:00:00 2001 From: pedro Date: Fri, 20 Sep 2024 12:28:48 -0300 Subject: [PATCH 11/23] docker-reset.sh: add debug flag --- docker-reset.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker-reset.sh b/docker-reset.sh index d75a86c..9264b93 100755 --- a/docker-reset.sh +++ b/docker-reset.sh @@ -6,7 +6,7 @@ set -e set -u # DEBUG -#set -x +set -x main() { # remove old database -- 2.30.2 From d144cef7b8e68c96bd50436e0a8568692ba03a16 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 20 Sep 2024 18:05:29 +0200 Subject: [PATCH 12/23] add mac --- evidence/parse.py | 26 +++++++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) diff --git a/evidence/parse.py b/evidence/parse.py index d9cfeda..eccb910 100644 --- a/evidence/parse.py +++ b/evidence/parse.py @@ -10,6 +10,28 @@ from evidence.models import Evidence, Annotation from utils.constants import ALGOS, CHASSIS_DH +def get_mac(hwinfo): + + low_ix = None + + nets = [x.split("\n") for x in hwinfo.split("\n\n") + if "network interface" in x and "Attached to" in x] + + for n in nets: + ix = None + if "Attached to:" in n: + for v in c.split(" "): + if "#" in v: + ix = int(v.strip("#")) + if not low_ix: + low_ix = ix + + if "HW Address:" in n: + if low_ix <= ix: + mac = c.split(" ")[-1] + return mac + + class Build: def __init__(self, evidence_json, user, check=False): self.json = evidence_json @@ -81,5 +103,7 @@ class Build: chassis = self.get_chassis_dh() serial_number = self.dmi.serial_number() sku = self.get_sku() + hwinfo_raw = snapshot["data"]["hwinfo"] + mac = get_mac(hwinfo_raw) or "" - return f"{manufacturer}{model}{chassis}{serial_number}{sku}" + return f"{manufacturer}{model}{chassis}{serial_number}{sku}{mac}" -- 2.30.2 From b36636d28e5429966ac69c4a174071b2b482897f Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 20 Sep 2024 18:22:27 +0200 Subject: [PATCH 13/23] print macs --- evidence/parse.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/evidence/parse.py b/evidence/parse.py index eccb910..8eb5e2a 100644 --- a/evidence/parse.py +++ b/evidence/parse.py @@ -29,6 +29,7 @@ def get_mac(hwinfo): if "HW Address:" in n: if low_ix <= ix: mac = c.split(" ")[-1] + print(f"MAC: {mac}") return mac @@ -105,5 +106,8 @@ class Build: sku = self.get_sku() hwinfo_raw = snapshot["data"]["hwinfo"] mac = get_mac(hwinfo_raw) or "" + if not mac: + print("WARNING!! No there are MAC address") + print(f"{manufacturer}{model}{chassis}{serial_number}{sku}{mac}") return f"{manufacturer}{model}{chassis}{serial_number}{sku}{mac}" -- 2.30.2 From bdaaa2740d26f862912f6a940ae8c37ededcce5c Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 20 Sep 2024 19:47:04 +0200 Subject: [PATCH 14/23] fixed geting mac --- evidence/parse.py | 31 +++++++++++++++++++------------ 1 file changed, 19 insertions(+), 12 deletions(-) diff --git a/evidence/parse.py b/evidence/parse.py index 8eb5e2a..2496457 100644 --- a/evidence/parse.py +++ b/evidence/parse.py @@ -13,24 +13,32 @@ from utils.constants import ALGOS, CHASSIS_DH def get_mac(hwinfo): low_ix = None + lnets = [] nets = [x.split("\n") for x in hwinfo.split("\n\n") if "network interface" in x and "Attached to" in x] for n in nets: ix = None - if "Attached to:" in n: - for v in c.split(" "): - if "#" in v: - ix = int(v.strip("#")) - if not low_ix: - low_ix = ix + mac = None + for l in n: + if "Attached to:" in l: + for v in l.split(" "): + if "#" in v: + ix = int(v.strip("#")) + if not low_ix: + low_ix = ix - if "HW Address:" in n: - if low_ix <= ix: - mac = c.split(" ")[-1] - print(f"MAC: {mac}") - return mac + if "HW Address:" in l: + mac = l.split(" ")[-1] + if ix and mac: + lnets.append((ix, mac)) + + if lnets: + lnets.sort() + mac = lnets[0][1] + print(f"MAC: {mac}") + return mac class Build: @@ -108,6 +116,5 @@ class Build: mac = get_mac(hwinfo_raw) or "" if not mac: print("WARNING!! No there are MAC address") - print(f"{manufacturer}{model}{chassis}{serial_number}{sku}{mac}") return f"{manufacturer}{model}{chassis}{serial_number}{sku}{mac}" -- 2.30.2 From ebe80ffea609423c008533bc45daaaa573f1891e Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 20 Sep 2024 19:48:53 +0200 Subject: [PATCH 15/23] add prints --- evidence/parse.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/evidence/parse.py b/evidence/parse.py index 2496457..7122d1e 100644 --- a/evidence/parse.py +++ b/evidence/parse.py @@ -116,5 +116,7 @@ class Build: mac = get_mac(hwinfo_raw) or "" if not mac: print("WARNING!! No there are MAC address") + else: + print(f"{manufacturer}{model}{chassis}{serial_number}{sku}{mac}") return f"{manufacturer}{model}{chassis}{serial_number}{sku}{mac}" -- 2.30.2 From 756570c7b47e531b16972665dd5fb4a915f7f54a Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Mon, 23 Sep 2024 13:44:04 +0200 Subject: [PATCH 16/23] this get the integrate network card --- evidence/parse.py | 24 +++++++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) diff --git a/evidence/parse.py b/evidence/parse.py index 7122d1e..85d6761 100644 --- a/evidence/parse.py +++ b/evidence/parse.py @@ -10,8 +10,9 @@ from evidence.models import Evidence, Annotation from utils.constants import ALGOS, CHASSIS_DH -def get_mac(hwinfo): - +def get_mac2(hwinfo): + # This function get the network card with most lower busid + # but maybe is external or maybe is integrate in motherboard low_ix = None lnets = [] @@ -39,7 +40,24 @@ def get_mac(hwinfo): mac = lnets[0][1] print(f"MAC: {mac}") return mac - + + +def get_network_cards(child, nets): + if child['id'] == 'network': + nets.append(child) + if child.get('children'): + [e(x, nets) for x in child['children']] + + +def get_mac(lshw): + # This funcion get the network card integrated in motherboard + nets = [] + get_network_cards(lshw, nets) + integrate = [x for x in nets if "pci@0000:00:" in x.get('businfo', '')] + + if integrate: + return integrate[0]['serial'] + class Build: def __init__(self, evidence_json, user, check=False): -- 2.30.2 From d277e6cbaa2ed39f884accff43039d4f5dff6cc6 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Mon, 23 Sep 2024 14:14:30 +0200 Subject: [PATCH 17/23] fix bug --- evidence/parse.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/evidence/parse.py b/evidence/parse.py index 85d6761..f01eddf 100644 --- a/evidence/parse.py +++ b/evidence/parse.py @@ -46,13 +46,13 @@ def get_network_cards(child, nets): if child['id'] == 'network': nets.append(child) if child.get('children'): - [e(x, nets) for x in child['children']] + [get_network_cards(x, nets) for x in child['children']] def get_mac(lshw): # This funcion get the network card integrated in motherboard nets = [] - get_network_cards(lshw, nets) + get_network_cards(json.loads(lshw), nets) integrate = [x for x in nets if "pci@0000:00:" in x.get('businfo', '')] if integrate: @@ -130,8 +130,16 @@ class Build: chassis = self.get_chassis_dh() serial_number = self.dmi.serial_number() sku = self.get_sku() + if not snapshot["data"].get('hwinfo'): + return f"{manufacturer}{model}{chassis}{serial_number}{sku}" + + if not snapshot["data"].get('lshw'): + return f"{manufacturer}{model}{chassis}{serial_number}{sku}" + hwinfo_raw = snapshot["data"]["hwinfo"] - mac = get_mac(hwinfo_raw) or "" + lshw = snapshot["data"]["lshw"] + # mac = get_mac2(hwinfo_raw) or "" + mac = get_mac(lshw) or "" if not mac: print("WARNING!! No there are MAC address") else: -- 2.30.2 From 2ed33270ed5a02159d425caf2c3f7478a4046380 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Mon, 23 Sep 2024 14:30:47 +0200 Subject: [PATCH 18/23] get the first network card --- evidence/parse.py | 47 +++++++---------------------------------------- 1 file changed, 7 insertions(+), 40 deletions(-) diff --git a/evidence/parse.py b/evidence/parse.py index f01eddf..2306ac4 100644 --- a/evidence/parse.py +++ b/evidence/parse.py @@ -10,38 +10,6 @@ from evidence.models import Evidence, Annotation from utils.constants import ALGOS, CHASSIS_DH -def get_mac2(hwinfo): - # This function get the network card with most lower busid - # but maybe is external or maybe is integrate in motherboard - low_ix = None - lnets = [] - - nets = [x.split("\n") for x in hwinfo.split("\n\n") - if "network interface" in x and "Attached to" in x] - - for n in nets: - ix = None - mac = None - for l in n: - if "Attached to:" in l: - for v in l.split(" "): - if "#" in v: - ix = int(v.strip("#")) - if not low_ix: - low_ix = ix - - if "HW Address:" in l: - mac = l.split(" ")[-1] - if ix and mac: - lnets.append((ix, mac)) - - if lnets: - lnets.sort() - mac = lnets[0][1] - print(f"MAC: {mac}") - return mac - - def get_network_cards(child, nets): if child['id'] == 'network': nets.append(child) @@ -50,13 +18,15 @@ def get_network_cards(child, nets): def get_mac(lshw): - # This funcion get the network card integrated in motherboard nets = [] - get_network_cards(json.loads(lshw), nets) - integrate = [x for x in nets if "pci@0000:00:" in x.get('businfo', '')] - if integrate: - return integrate[0]['serial'] + get_network_cards(json.loads(lshw), nets) + nets_sorted = sorted(nets, key=lambda x: x['businfo']) + # This funcion get the network card integrated in motherboard + # integrate = [x for x in nets if "pci@0000:00:" in x.get('businfo', '')] + + if nets_sorted: + return nets_sorted[0]['serial'] class Build: @@ -130,13 +100,10 @@ class Build: chassis = self.get_chassis_dh() serial_number = self.dmi.serial_number() sku = self.get_sku() - if not snapshot["data"].get('hwinfo'): - return f"{manufacturer}{model}{chassis}{serial_number}{sku}" if not snapshot["data"].get('lshw'): return f"{manufacturer}{model}{chassis}{serial_number}{sku}" - hwinfo_raw = snapshot["data"]["hwinfo"] lshw = snapshot["data"]["lshw"] # mac = get_mac2(hwinfo_raw) or "" mac = get_mac(lshw) or "" -- 2.30.2 From 36fe3310a36e3d6acca7d6b2a0d1036077f5f154 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Thu, 10 Oct 2024 10:35:27 +0200 Subject: [PATCH 19/23] user panel --- dashboard/templates/base.html | 4 ++-- dhub/urls.py | 1 + user/templates/panel.html | 18 ++++++++++++++++++ user/urls.py | 8 ++++++++ user/views.py | 11 +++++++++-- 5 files changed, 38 insertions(+), 4 deletions(-) create mode 100644 user/templates/panel.html create mode 100644 user/urls.py diff --git a/dashboard/templates/base.html b/dashboard/templates/base.html index c8b0a38..97d090a 100644 --- a/dashboard/templates/base.html +++ b/dashboard/templates/base.html @@ -61,12 +61,12 @@
    DEVICEHUB
    - PANGEA + {{ user.institution.name|upper }}
    - {{ user.email }} + {{ user.email }} diff --git a/dhub/urls.py b/dhub/urls.py index ffc6e4d..e9cec23 100644 --- a/dhub/urls.py +++ b/dhub/urls.py @@ -24,5 +24,6 @@ urlpatterns = [ path("evidence/", include("evidence.urls")), path("device/", include("device.urls")), path("admin/", include("admin.urls")), + path("user/", include("user.urls")), path("lot/", include("lot.urls")), ] diff --git a/user/templates/panel.html b/user/templates/panel.html new file mode 100644 index 0000000..99e22cb --- /dev/null +++ b/user/templates/panel.html @@ -0,0 +1,18 @@ +{% extends "base.html" %} +{% load i18n %} + +{% block content %} +
    +
    +

    {{ subtitle }}

    +
    +
    + {{ user.email }} +
    +
    + +
    + +
    + +{% endblock %} diff --git a/user/urls.py b/user/urls.py new file mode 100644 index 0000000..da4e507 --- /dev/null +++ b/user/urls.py @@ -0,0 +1,8 @@ +from django.urls import path +from user import views + +app_name = 'user' + +urlpatterns = [ + path("panel/", views.PanelView.as_view(), name="panel"), +] diff --git a/user/views.py b/user/views.py index 91ea44a..e0a195a 100644 --- a/user/views.py +++ b/user/views.py @@ -1,3 +1,10 @@ -from django.shortcuts import render +from django.utils.translation import gettext_lazy as _ +from django.views.generic.base import TemplateView +from dashboard.mixins import DashboardView -# Create your views here. + +class PanelView(DashboardView, TemplateView): + template_name = "panel.html" + title = _("User") + breadcrumb = "User / Panel" + subtitle = "User panel" -- 2.30.2 From a6d077aa2d27bc9808835fc84b5a466d2facc942 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Thu, 10 Oct 2024 12:28:03 +0200 Subject: [PATCH 20/23] resolve conflicts --- dashboard/templates/base.html | 10 +++ evidence/migrations/0001_initial.py | 2 +- .../migrations/0004_alter_annotation_owner.py | 22 ------- lot/migrations/0001_initial.py | 66 ++++++++++++++++++- lot/migrations/0002_lotannotation.py | 52 --------------- ...wner_alter_lotannotation_owner_and_more.py | 36 ---------- lot/models.py | 5 +- user/management/commands/add_user.py | 4 +- 8 files changed, 82 insertions(+), 115 deletions(-) delete mode 100644 evidence/migrations/0004_alter_annotation_owner.py delete mode 100644 lot/migrations/0002_lotannotation.py delete mode 100644 lot/migrations/0003_alter_lot_owner_alter_lotannotation_owner_and_more.py diff --git a/dashboard/templates/base.html b/dashboard/templates/base.html index 5c92024..d2c1612 100644 --- a/dashboard/templates/base.html +++ b/dashboard/templates/base.html @@ -185,7 +185,17 @@ {% endblock messages %}

    {{ title }}

    +
    + {% csrf_token %} +
    + + + + +
    +
    +
    {{ breadcrumb }} diff --git a/evidence/migrations/0001_initial.py b/evidence/migrations/0001_initial.py index 1a621d0..9ffbf31 100644 --- a/evidence/migrations/0001_initial.py +++ b/evidence/migrations/0001_initial.py @@ -1,4 +1,4 @@ -# Generated by Django 5.0.6 on 2024-10-07 11:38 +# Generated by Django 5.0.6 on 2024-10-10 09:46 import django.db.models.deletion from django.conf import settings diff --git a/evidence/migrations/0004_alter_annotation_owner.py b/evidence/migrations/0004_alter_annotation_owner.py deleted file mode 100644 index 26c9bda..0000000 --- a/evidence/migrations/0004_alter_annotation_owner.py +++ /dev/null @@ -1,22 +0,0 @@ -# Generated by Django 5.0.6 on 2024-09-18 10:55 - -import django.db.models.deletion -from django.db import migrations, models - - -class Migration(migrations.Migration): - - dependencies = [ - ("evidence", "0003_alter_annotation_type"), - ("user", "0001_initial"), - ] - - operations = [ - migrations.AlterField( - model_name="annotation", - name="owner", - field=models.ForeignKey( - on_delete=django.db.models.deletion.CASCADE, to="user.institution" - ), - ), - ] diff --git a/lot/migrations/0001_initial.py b/lot/migrations/0001_initial.py index 28b3e8e..538ba90 100644 --- a/lot/migrations/0001_initial.py +++ b/lot/migrations/0001_initial.py @@ -1,4 +1,4 @@ -# Generated by Django 5.0.6 on 2024-07-27 16:23 +# Generated by Django 5.0.6 on 2024-10-10 10:14 import django.db.models.deletion from django.conf import settings @@ -10,6 +10,7 @@ class Migration(migrations.Migration): initial = True dependencies = [ + ("user", "0001_initial"), migrations.swappable_dependency(settings.AUTH_USER_MODEL), ] @@ -36,6 +37,15 @@ class Migration(migrations.Migration): "owner", models.ForeignKey( on_delete=django.db.models.deletion.CASCADE, + to="user.institution", + ), + ), + ( + "user", + models.ForeignKey( + blank=True, + null=True, + on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL, ), ), @@ -62,6 +72,51 @@ class Migration(migrations.Migration): ), ], ), + migrations.CreateModel( + name="LotAnnotation", + fields=[ + ( + "id", + models.BigAutoField( + auto_created=True, + primary_key=True, + serialize=False, + verbose_name="ID", + ), + ), + ("created", models.DateTimeField(auto_now_add=True)), + ( + "type", + models.SmallIntegerField( + choices=[(0, "System"), (1, "User"), (2, "Document")] + ), + ), + ("key", models.CharField(max_length=256)), + ("value", models.CharField(max_length=256)), + ( + "lot", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, to="lot.lot" + ), + ), + ( + "owner", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, + to="user.institution", + ), + ), + ( + "user", + models.ForeignKey( + blank=True, + null=True, + on_delete=django.db.models.deletion.SET_NULL, + to=settings.AUTH_USER_MODEL, + ), + ), + ], + ), migrations.CreateModel( name="LotTag", fields=[ @@ -79,6 +134,15 @@ class Migration(migrations.Migration): "owner", models.ForeignKey( on_delete=django.db.models.deletion.CASCADE, + to="user.institution", + ), + ), + ( + "user", + models.ForeignKey( + blank=True, + null=True, + on_delete=django.db.models.deletion.SET_NULL, to=settings.AUTH_USER_MODEL, ), ), diff --git a/lot/migrations/0002_lotannotation.py b/lot/migrations/0002_lotannotation.py deleted file mode 100644 index aca7de1..0000000 --- a/lot/migrations/0002_lotannotation.py +++ /dev/null @@ -1,52 +0,0 @@ -# Generated by Django 5.0.6 on 2024-07-29 15:37 - -import django.db.models.deletion -from django.conf import settings -from django.db import migrations, models - - -class Migration(migrations.Migration): - - dependencies = [ - ("lot", "0001_initial"), - migrations.swappable_dependency(settings.AUTH_USER_MODEL), - ] - - operations = [ - migrations.CreateModel( - name="LotAnnotation", - fields=[ - ( - "id", - models.BigAutoField( - auto_created=True, - primary_key=True, - serialize=False, - verbose_name="ID", - ), - ), - ("created", models.DateTimeField(auto_now_add=True)), - ( - "type", - models.SmallIntegerField( - choices=[(0, "System"), (1, "User"), (2, "Document")] - ), - ), - ("key", models.CharField(max_length=256)), - ("value", models.CharField(max_length=256)), - ( - "lot", - models.ForeignKey( - on_delete=django.db.models.deletion.CASCADE, to="lot.lot" - ), - ), - ( - "owner", - models.ForeignKey( - on_delete=django.db.models.deletion.CASCADE, - to=settings.AUTH_USER_MODEL, - ), - ), - ], - ), - ] diff --git a/lot/migrations/0003_alter_lot_owner_alter_lotannotation_owner_and_more.py b/lot/migrations/0003_alter_lot_owner_alter_lotannotation_owner_and_more.py deleted file mode 100644 index 806c6f1..0000000 --- a/lot/migrations/0003_alter_lot_owner_alter_lotannotation_owner_and_more.py +++ /dev/null @@ -1,36 +0,0 @@ -# Generated by Django 5.0.6 on 2024-09-18 10:55 - -import django.db.models.deletion -from django.db import migrations, models - - -class Migration(migrations.Migration): - - dependencies = [ - ("lot", "0002_lotannotation"), - ("user", "0001_initial"), - ] - - operations = [ - migrations.AlterField( - model_name="lot", - name="owner", - field=models.ForeignKey( - on_delete=django.db.models.deletion.CASCADE, to="user.institution" - ), - ), - migrations.AlterField( - model_name="lotannotation", - name="owner", - field=models.ForeignKey( - on_delete=django.db.models.deletion.CASCADE, to="user.institution" - ), - ), - migrations.AlterField( - model_name="lottag", - name="owner", - field=models.ForeignKey( - on_delete=django.db.models.deletion.CASCADE, to="user.institution" - ), - ), - ] diff --git a/lot/models.py b/lot/models.py index 3926f29..764e134 100644 --- a/lot/models.py +++ b/lot/models.py @@ -6,7 +6,7 @@ from utils.constants import ( STR_EXTEND_SIZE, ) -from user.models import Institution +from user.models import User, Institution # from device.models import Device # from evidence.models import Annotation @@ -14,6 +14,7 @@ from user.models import Institution class LotTag(models.Model): name = models.CharField(max_length=STR_SIZE, blank=False, null=False) owner = models.ForeignKey(Institution, on_delete=models.CASCADE) + user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True) def __str__(self): return self.name @@ -32,6 +33,7 @@ class Lot(models.Model): description = models.CharField(max_length=STR_SIZE, blank=True, null=True) closed = models.BooleanField(default=True) owner = models.ForeignKey(Institution, on_delete=models.CASCADE) + user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True) type = models.ForeignKey(LotTag, on_delete=models.CASCADE) def add(self, v): @@ -53,6 +55,7 @@ class LotAnnotation(models.Model): created = models.DateTimeField(auto_now_add=True) lot = models.ForeignKey(Lot, on_delete=models.CASCADE) owner = models.ForeignKey(Institution, on_delete=models.CASCADE) + user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True) type = models.SmallIntegerField(choices=Type) key = models.CharField(max_length=STR_EXTEND_SIZE) value = models.CharField(max_length=STR_EXTEND_SIZE) diff --git a/user/management/commands/add_user.py b/user/management/commands/add_user.py index d6d3bd8..73385d8 100644 --- a/user/management/commands/add_user.py +++ b/user/management/commands/add_user.py @@ -1,6 +1,7 @@ from django.core.management.base import BaseCommand from django.contrib.auth import get_user_model from user.models import Institution +from lot.models import LotTag User = get_user_model() @@ -21,7 +22,6 @@ class Command(BaseCommand): is_admin = kwargs['is_admin'] institution = Institution.objects.get(name=kwargs['institution']) self.create_user(institution, email, password, is_admin) - self.create_lot_tags() def create_user(self, institution, email, password, is_admin): self.u = User.objects.create( @@ -30,5 +30,5 @@ class Command(BaseCommand): password=password, is_admin=is_admin, ) - self.u.set_password(self.password) + self.u.set_password(password) self.u.save() -- 2.30.2 From 3e967c5a1a2f7d0d3c114815d7e34da857f06b76 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Thu, 10 Oct 2024 14:03:08 +0200 Subject: [PATCH 21/23] token to user panel link and add tags to tokens --- api/migrations/0001_initial.py | 3 +- api/models.py | 3 +- api/tables.py | 22 ++++++++++++-- api/templates/new_token.html | 32 ++++++++++++++++++++ api/templates/token.html | 1 + api/urls.py | 1 + api/views.py | 55 ++++++++++++++++++++++++++++------ dashboard/templates/base.html | 6 ---- user/templates/panel.html | 4 ++- 9 files changed, 106 insertions(+), 21 deletions(-) create mode 100644 api/templates/new_token.html diff --git a/api/migrations/0001_initial.py b/api/migrations/0001_initial.py index 2101a8a..b15c44e 100644 --- a/api/migrations/0001_initial.py +++ b/api/migrations/0001_initial.py @@ -1,4 +1,4 @@ -# Generated by Django 5.0.6 on 2024-10-10 10:35 +# Generated by Django 5.0.6 on 2024-10-10 11:34 import django.db.models.deletion from django.conf import settings @@ -26,6 +26,7 @@ class Migration(migrations.Migration): verbose_name="ID", ), ), + ("tag", models.CharField(max_length=50)), ("token", models.UUIDField()), ( "owner", diff --git a/api/models.py b/api/models.py index b8bbc24..8cdf430 100644 --- a/api/models.py +++ b/api/models.py @@ -1,9 +1,8 @@ from django.db import models from user.models import User -# Create your models here. - class Token(models.Model): + tag = models.CharField(max_length=50) token = models.UUIDField() owner = models.ForeignKey(User, on_delete=models.CASCADE) diff --git a/api/tables.py b/api/tables.py index ac1cc7a..27c8cbf 100644 --- a/api/tables.py +++ b/api/tables.py @@ -33,12 +33,30 @@ class TokensTable(tables.Table): }, orderable=False ) - + edit_token = ButtonColumn( + linkify={ + "viewname": "api:edit_token", + "args": [tables.A("pk")] + }, + attrs = { + "a": { + "type": "button", + "class": "text-primary", + "title": "Remove", + } + }, + orderable=False, + verbose_name="Edit" + ) token = tables.Column(verbose_name=_("Token"), empty_values=()) + tag = tables.Column(verbose_name=_("Tag"), empty_values=()) def render_view_user(self): return format_html('') + def render_edit_token(self): + return format_html('') + # def render_token(self, record): # return record.get_memberships() @@ -63,5 +81,5 @@ class TokensTable(tables.Table): class Meta: model = Token template_name = "custom_table.html" - fields = ("token", "view_user") + fields = ("token", "tag", "edit_token") diff --git a/api/templates/new_token.html b/api/templates/new_token.html new file mode 100644 index 0000000..77e0503 --- /dev/null +++ b/api/templates/new_token.html @@ -0,0 +1,32 @@ +{% extends "base.html" %} +{% load i18n %} + +{% block content %} +
    +
    +

    {{ subtitle }}

    +
    +
    + +{% load django_bootstrap5 %} +
    +{% csrf_token %} +{% if form.errors %} + +{% endif %} +{% bootstrap_form form %} +
    + {% translate "Cancel" %} + +
    + +
    +{% endblock %} diff --git a/api/templates/token.html b/api/templates/token.html index 5185090..2ba2ca7 100644 --- a/api/templates/token.html +++ b/api/templates/token.html @@ -8,6 +8,7 @@ {{ subtitle }} {% render_table table %} +
    {% translate "Generate a new token" %}
    diff --git a/api/urls.py b/api/urls.py index f3c8028..5eac279 100644 --- a/api/urls.py +++ b/api/urls.py @@ -9,5 +9,6 @@ urlpatterns = [ path('snapshot/', views.NewSnapshot, name='new_snapshot'), path('tokens/', views.TokenView.as_view(), name='tokens'), path('tokens/new', views.TokenNewView.as_view(), name='new_token'), + path("tokens//edit", views.EditTokenView.as_view(), name="edit_token"), path('tokens//del', views.TokenDeleteView.as_view(), name='delete_token'), ] diff --git a/api/views.py b/api/views.py index fbccb79..057ede2 100644 --- a/api/views.py +++ b/api/views.py @@ -1,13 +1,18 @@ import json +from django.urls import reverse_lazy from django.shortcuts import get_object_or_404, redirect from django.utils.translation import gettext_lazy as _ from django.views.decorators.csrf import csrf_exempt from django.core.exceptions import ValidationError -from django.views.generic.edit import DeleteView -from django.views.generic.base import View -from django.http import JsonResponse from django_tables2 import SingleTableView +from django.views.generic.base import View +from django.views.generic.edit import ( + CreateView, + DeleteView, + UpdateView, +) +from django.http import JsonResponse from uuid import uuid4 from dashboard.mixins import DashboardView @@ -89,7 +94,7 @@ class TokenView(DashboardView, SingleTableView): def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context.update({ - 'tokens': Token.objects, + 'tokens': Token.objects.all(), }) return context @@ -105,10 +110,42 @@ class TokenDeleteView(DashboardView, DeleteView): return redirect('api:tokens') -class TokenNewView(DashboardView, View): +class TokenNewView(DashboardView, CreateView): + template_name = "new_token.html" + title = _("Credential management") + section = "Credential" + subtitle = _('New Tokens') + icon = 'bi bi-key' + model = Token + success_url = reverse_lazy('api:tokens') + fields = ( + "tag", + ) - def get(self, request, *args, **kwargs): - Token.objects.create(token=uuid4(), owner=self.request.user) + def form_valid(self, form): + form.instance.owner = self.request.user + form.instance.token = uuid4() + return super().form_valid(form) - return redirect('api:tokens') - + +class EditTokenView(DashboardView, UpdateView): + template_name = "new_token.html" + title = _("Credential management") + section = "Credential" + subtitle = _('New Tokens') + icon = 'bi bi-key' + model = Token + success_url = reverse_lazy('api:tokens') + fields = ( + "tag", + ) + + def get_form_kwargs(self): + pk = self.kwargs.get('pk') + self.object = get_object_or_404( + self.model, + owner=self.request.user, + pk=pk, + ) + kwargs = super().get_form_kwargs() + return kwargs diff --git a/dashboard/templates/base.html b/dashboard/templates/base.html index 6aeee6a..40e6658 100644 --- a/dashboard/templates/base.html +++ b/dashboard/templates/base.html @@ -163,12 +163,6 @@
    • -
  • - - - {% trans 'Tokens' %} - -
    • diff --git a/user/templates/panel.html b/user/templates/panel.html index 99e22cb..ca27036 100644 --- a/user/templates/panel.html +++ b/user/templates/panel.html @@ -12,7 +12,9 @@
    - + + {% trans 'Tokens' %} +
    {% endblock %} -- 2.30.2 From e0929c03c7712086ba7aec8af069ca14223d8c8a Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 11 Oct 2024 13:04:37 +0200 Subject: [PATCH 22/23] add settings page for download settings file --- user/forms.py | 20 ++++++++++++++++++++ user/templates/panel.html | 13 ++++++++++--- user/templates/settings.html | 32 ++++++++++++++++++++++++++++++++ user/templates/settings.ini | 3 +++ user/urls.py | 1 + user/views.py | 29 +++++++++++++++++++++++++++++ 6 files changed, 95 insertions(+), 3 deletions(-) create mode 100644 user/forms.py create mode 100644 user/templates/settings.html create mode 100644 user/templates/settings.ini diff --git a/user/forms.py b/user/forms.py new file mode 100644 index 0000000..2dd7e6b --- /dev/null +++ b/user/forms.py @@ -0,0 +1,20 @@ +from django import forms + + +class SettingsForm(forms.Form): + token = forms.ChoiceField( + choices = [] + ) + erasure = forms.ChoiceField( + choices = [(0, 'Not erasure'), + ('erasure1', 'Erasure easy'), + ('erasure2', 'Erasure mediom'), + ('erasure3', 'Erasure hard'), + ], + ) + + def __init__(self, *args, **kwargs): + tokens = kwargs.pop('tokens') + super().__init__(*args, **kwargs) + tk = [(str(x.token), x.tag) for x in tokens] + self.fields['token'].choices = tk diff --git a/user/templates/panel.html b/user/templates/panel.html index ca27036..8686062 100644 --- a/user/templates/panel.html +++ b/user/templates/panel.html @@ -12,9 +12,16 @@
    - - {% trans 'Tokens' %} - +
    + + {% translate 'Admin your Tokens' %} + +
    +
    + + {% translate 'Download a settings file' %} + +
    {% endblock %} diff --git a/user/templates/settings.html b/user/templates/settings.html new file mode 100644 index 0000000..cf93776 --- /dev/null +++ b/user/templates/settings.html @@ -0,0 +1,32 @@ +{% extends "base.html" %} +{% load i18n %} + +{% block content %} +
    +
    +

    {{ subtitle }}

    +
    +
    + +{% load django_bootstrap5 %} +
    +{% csrf_token %} +{% if form.errors %} + +{% endif %} +{% bootstrap_form form %} +
    + {% translate "Cancel" %} + +
    + +
    +{% endblock %} diff --git a/user/templates/settings.ini b/user/templates/settings.ini new file mode 100644 index 0000000..390b650 --- /dev/null +++ b/user/templates/settings.ini @@ -0,0 +1,3 @@ +token = {{ token }} +erasure = {{ erasure }} +legacy = False \ No newline at end of file diff --git a/user/urls.py b/user/urls.py index da4e507..f9e6246 100644 --- a/user/urls.py +++ b/user/urls.py @@ -5,4 +5,5 @@ app_name = 'user' urlpatterns = [ path("panel/", views.PanelView.as_view(), name="panel"), + path("settings/", views.SettingsView.as_view(), name="settings"), ] diff --git a/user/views.py b/user/views.py index e0a195a..36093b8 100644 --- a/user/views.py +++ b/user/views.py @@ -1,6 +1,14 @@ +from django.http import HttpResponse +from django.shortcuts import render from django.utils.translation import gettext_lazy as _ from django.views.generic.base import TemplateView from dashboard.mixins import DashboardView +from django.views.generic.edit import ( + FormView, +) + +from user.forms import SettingsForm +from api.models import Token class PanelView(DashboardView, TemplateView): @@ -8,3 +16,24 @@ class PanelView(DashboardView, TemplateView): title = _("User") breadcrumb = "User / Panel" subtitle = "User panel" + + +class SettingsView(DashboardView, FormView): + template_name = "settings.html" + title = _("Download Settings") + breadcrumb = "user / workbench / settings" + form_class = SettingsForm + + def form_valid(self, form): + form.devices = self.get_session_devices() + data = render(self.request, "settings.ini", form.cleaned_data) + response = HttpResponse(data.content, content_type="application/text") + response['Content-Disposition'] = 'attachment; filename={}'.format("settings.ini") + return response + + def get_form_kwargs(self): + tokens = Token.objects.filter(owner=self.request.user) + kwargs = super().get_form_kwargs() + kwargs['tokens'] = tokens + return kwargs + -- 2.30.2 From 017dc818da602462d55b367300e8e80b1df8297b Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Fri, 11 Oct 2024 13:46:21 +0200 Subject: [PATCH 23/23] add url in settings and legacy --- user/forms.py | 6 +++--- user/templates/settings.ini | 7 +++++-- user/templates/settings_legacy.ini | 6 ++++++ user/views.py | 15 +++++++++++++-- 4 files changed, 27 insertions(+), 7 deletions(-) create mode 100644 user/templates/settings_legacy.ini diff --git a/user/forms.py b/user/forms.py index 2dd7e6b..ec3f17f 100644 --- a/user/forms.py +++ b/user/forms.py @@ -7,9 +7,9 @@ class SettingsForm(forms.Form): ) erasure = forms.ChoiceField( choices = [(0, 'Not erasure'), - ('erasure1', 'Erasure easy'), - ('erasure2', 'Erasure mediom'), - ('erasure3', 'Erasure hard'), + ('basic', 'Erasure Basic'), + ('baseline', 'Erasure Baseline'), + ('enhanced', 'Erasure Enhanced'), ], ) diff --git a/user/templates/settings.ini b/user/templates/settings.ini index 390b650..217e993 100644 --- a/user/templates/settings.ini +++ b/user/templates/settings.ini @@ -1,3 +1,6 @@ +[settings] +url = {{ url }} token = {{ token }} -erasure = {{ erasure }} -legacy = False \ No newline at end of file +erase = {{ erasure }} +legacy = false +# path = /path/to/save \ No newline at end of file diff --git a/user/templates/settings_legacy.ini b/user/templates/settings_legacy.ini new file mode 100644 index 0000000..6407090 --- /dev/null +++ b/user/templates/settings_legacy.ini @@ -0,0 +1,6 @@ +[settings] +url = {{ url }} +token = {{ token }} +legacy = true +# erase = {{ erasure }} +# path = /path/to/save \ No newline at end of file diff --git a/user/views.py b/user/views.py index 36093b8..750b9a3 100644 --- a/user/views.py +++ b/user/views.py @@ -1,3 +1,5 @@ +from decouple import config +from django.urls import reverse from django.http import HttpResponse from django.shortcuts import render from django.utils.translation import gettext_lazy as _ @@ -25,8 +27,17 @@ class SettingsView(DashboardView, FormView): form_class = SettingsForm def form_valid(self, form): - form.devices = self.get_session_devices() - data = render(self.request, "settings.ini", form.cleaned_data) + cleaned_data = form.cleaned_data.copy() + settings_tmpl = "settings.ini" + path = reverse("api:new_snapshot") + cleaned_data['url'] = self.request.build_absolute_uri(path) + + if config("LEGACY", False): + cleaned_data['token'] = config.get('TOKEN_LEGACY', '') + cleaned_data['url'] = config.get('URL_LEGACY', '') + settings_tmpl = "settings_legacy.ini" + + data = render(self.request, settings_tmpl, cleaned_data) response = HttpResponse(data.content, content_type="application/text") response['Content-Disposition'] = 'attachment; filename={}'.format("settings.ini") return response -- 2.30.2