add models to admin pages

e-m-b-a · Nov 4, 2024 · 77fc120 · 77fc120
1 parent 201e1cc
commit 77fc120
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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)