diff --git a/embark/dashboard/admin.py b/embark/dashboard/admin.py index 24817012..b14cd423 100644 --- a/embark/dashboard/admin.py +++ b/embark/dashboard/admin.py @@ -4,6 +4,8 @@ from django.contrib import admin -from dashboard.models import Result +from dashboard.models import * -admin.register(Result) +admin.site.register(Result) +admin.site.register(Vulnerability) +admin.site.register(SoftwareInfo) diff --git a/embark/dashboard/models.py b/embark/dashboard/models.py index 2c1df10a..066218d8 100644 --- a/embark/dashboard/models.py +++ b/embark/dashboard/models.py @@ -19,7 +19,7 @@ class Vulnerability(models.Model): class SoftwareInfo(models.Model): """ - Many-to-many object for SBOM + Many-to-many object for SBOM entries """ id = models.UUIDField(primary_key=True, default=uuid.uuid4) name = models.CharField(verbose_name="software name", blank=False, editable=True, default="NA", max_length=256) diff --git a/embark/uploader/admin.py b/embark/uploader/admin.py index 14a7187d..d2795598 100644 --- a/embark/uploader/admin.py +++ b/embark/uploader/admin.py @@ -4,6 +4,8 @@ from django.contrib import admin -from uploader.models import FirmwareAnalysis +from uploader.models import FirmwareAnalysis, FirmwareFile, Device admin.site.register(FirmwareAnalysis) +admin.site.register(Device) +admin.site.register(FirmwareFile)