Merge pull request #17 from k01ek/develop

add customfield model to asn,session,policy

Author: k01ek

netbox_bgp/migrations/0015_netbox_bgp.py

@@ -0,0 +1,29 @@
+# Generated by Django 3.2 on 2021-05-05 06:56
+
+import django.core.serializers.json
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('netbox_bgp', '0014_netbox_bgp'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='asn',
+            name='custom_field_data',
+            field=models.JSONField(blank=True, default=dict, encoder=django.core.serializers.json.DjangoJSONEncoder),
+        ),
+        migrations.AddField(
+            model_name='bgpsession',
+            name='custom_field_data',
+            field=models.JSONField(blank=True, default=dict, encoder=django.core.serializers.json.DjangoJSONEncoder),
+        ),
+        migrations.AddField(
+            model_name='routingpolicy',
+            name='custom_field_data',
+            field=models.JSONField(blank=True, default=dict, encoder=django.core.serializers.json.DjangoJSONEncoder),
+        ),
+    ]

netbox_bgp/models.py

@@ -9,6 +9,12 @@
     from extras.models import ChangeLoggedModel
 except ImportError:
     from netbox.models import ChangeLoggedModel
+
+try:
+    from extras.models import CustomFieldModel
+except ImportError:
+    from netbox.models import CustomFieldsMixin as CustomFieldModel
+
 from extras.models import TaggedItem
 from extras.utils import extras_features
 
@@ -76,7 +82,7 @@ def __str__(self):
 
 
 @extras_features('custom_fields', 'export_templates', 'webhooks')
-class RoutingPolicy(ChangeLoggedModel):
+class RoutingPolicy(ChangeLoggedModel, CustomFieldModel):
     """
     """
     name = models.CharField(
@@ -171,7 +177,7 @@ class Meta:
 
 
 @extras_features('export_templates', 'webhooks')
-class ASN(BGPBase):
+class ASN(BGPBase, CustomFieldModel):
 
     group = models.ForeignKey(
         ASNGroup,
@@ -218,7 +224,7 @@ def get_absolute_url(self):
 
 
 @extras_features('export_templates', 'webhooks')
-class BGPSession(ChangeLoggedModel):
+class BGPSession(ChangeLoggedModel, CustomFieldModel):
     name = models.CharField(
         max_length=64,
         blank=True,

netbox_bgp/version.py

@@ -1 +1 @@
-__version__ = "0.3.1"
+__version__ = "0.3.2"