feat(core): function to fetch history entry item

required for the checking of user permissions against the organization

!18 #48 #15

app/core/views/history.py

@@ -10,8 +10,6 @@ from access.mixin import OrganizationPermission
 
 from core.models.history import History
 
-from itam.models.device import Device, DeviceSoftware, DeviceOperatingSystem
-from itam.models.software import Software
 
 
 class View(OrganizationPermission, generic.View):
@@ -23,6 +21,67 @@ class View(OrganizationPermission, generic.View):
     template_name = 'history.html.j2'
 
 
+    def get_object(self):
+        """ Get the history entry model
+
+        function required to check permissions, in particular that the user is in the same organization.
+
+        Raises:
+            Exception: if the model can't be found.
+
+        Returns:
+            Model: Returns the model Object
+        """
+        from access.models import Organization, Team
+
+        from itam.models.device import Device, DeviceSoftware, DeviceModel, DeviceOperatingSystem
+        from itam.models.operating_system import OperatingSystem
+        from itam.models.software import Software
+
+        from config_management.models.groups import ConfigGroups
+
+        if not hasattr(self, 'model'):
+
+            match self.kwargs['model_name']:
+
+                case 'configgroups':
+
+                    self.model = ConfigGroups
+
+                case 'device':
+
+                    self.model = Device
+
+                case 'devicemodel':
+
+                    self.model = DeviceModel
+
+                case 'software':
+
+                    self.model = Software
+
+                case 'operatingsystem':
+
+                    self.model = OperatingSystem
+
+                case 'organization':
+
+                    self.model = Organization
+
+                case 'team':
+
+                    self.model = Team
+
+                case _:
+                    raise Exception('Unable to determine history items model')
+
+        if not hasattr(self, 'obj'):
+
+            self.obj = self.model.objects.get(id=self.kwargs['model_pk'])
+
+        return self.obj
+
+
     def get(self, request, model_name, model_pk):
         if not request.user.is_authenticated and settings.LOGIN_REQUIRED:
             return redirect(f"{settings.LOGIN_URL}?next={request.path}")