Feature 15492 - Copying of Admin->Authentication->Permissions

netbox/users/models/permissions.py

@@ -1,9 +1,12 @@
+import json
+
 from django.contrib.postgres.fields import ArrayField
 from django.db import models
 from django.urls import reverse
 from django.utils.translation import gettext_lazy as _
 
 from users.constants import OBJECTPERMISSION_OBJECT_TYPES
+
 from utilities.querysets import RestrictedQuerySet
 
 __all__ = (
@@ -47,6 +50,8 @@ class ObjectPermission(models.Model):
 
     objects = RestrictedQuerySet.as_manager()
 
+    clone_fields = ['name', 'description', 'enabled', 'object_types', 'actions', 'constraints']
+
     class Meta:
         ordering = ['name']
         verbose_name = _('permission')
@@ -81,3 +86,18 @@ def list_constraints(self):
 
     def get_absolute_url(self):
         return reverse('users:objectpermission', args=[self.pk])
+
+    def clone(self):
+        attrs = {}
+
+        for field_name in getattr(self, 'clone_fields', []):
+            field = self._meta.get_field(field_name)
+            field_value = field.value_from_object(self)
+            if field_value and isinstance(field, models.ManyToManyField):
+                attrs[field_name] = [v.pk for v in field_value]
+            elif field_value and isinstance(field, models.JSONField):
+                attrs[field_name] = json.dumps(field_value)
+            elif field_value not in (None, ''):
+                attrs[field_name] = field_value
+
+        return attrs