Merge pull request #13 from k01ek/develop

add routing policies;

Author: k01ek

netbox_bgp/admin.py

@@ -1,17 +1,30 @@
 from django.contrib import admin
-from .models import ASN, Community, BGPSession
+from .models import ASN, Community, BGPSession, BGPPeerGroup, RoutingPolicy
 
 
 @admin.register(ASN)
 class ASNAdmin(admin.ModelAdmin):
-    fields = ('number', 'name', 'status', 'description')
+    fields = ('number', 'status', 'description')
 
 
 @admin.register(Community)
 class CommunityAdmin(admin.ModelAdmin):
-    fields = ('value', 'name', 'status', 'description')
+    fields = ('value', 'status', 'description')
 
 
 @admin.register(BGPSession)
 class BGPSessionAdmin(admin.ModelAdmin):
-    fields = ('name', 'local_address', 'local_as', 'remote_address', 'remote_as', 'description')
+    fields = (
+        'name', 'local_address', 'local_as', 'remote_address',
+        'remote_as', 'description', 'import_policies', 'export_policies'
+    )
+
+
+@admin.register(BGPPeerGroup)
+class BGPpeerGroupAdmin(admin.ModelAdmin):
+    fields = ('name', 'description', 'import_policies', 'export_policies')
+
+
+@admin.register(RoutingPolicy)
+class RoutingPolicyAdmin(admin.ModelAdmin):
+    fields = ('name', 'description')

netbox_bgp/api/serializers.py

@@ -10,7 +10,7 @@
     from netbox.api.serializers import CustomFieldModelSerializer
 
 from extras.models import Tag
-from netbox_bgp.models import ASN, ASNStatusChoices, BGPSession, SessionStatusChoices
+from netbox_bgp.models import ASN, ASNStatusChoices, BGPSession, SessionStatusChoices, RoutingPolicy
 
 
 class NestedTagSerializer(WritableNestedSerializer):
@@ -68,3 +68,9 @@ class BGPSessionSerializer(TaggedObjectSerializer, CustomFieldModelSerializer):
     class Meta:
         model = BGPSession
         fields = '__all__'
+
+
+class RoutingPolicySerializer(TaggedObjectSerializer, CustomFieldModelSerializer):
+    class Meta:
+        model = RoutingPolicy
+        fields = '__all__'

netbox_bgp/api/urls.py

@@ -1,9 +1,11 @@
 from rest_framework import routers
 
-from .views import ASNViewSet, BGPSessionViewSet
+from .views import ASNViewSet, BGPSessionViewSet, RoutingPolicyViewSet
 
 router = routers.DefaultRouter()
 router.register('asn', ASNViewSet)
-router.register('session', BGPSessionViewSet)
+router.register('session', BGPSessionViewSet),
+router.register('routing_policy', RoutingPolicyViewSet)
+
 
 urlpatterns = router.urls

netbox_bgp/api/views.py

@@ -1,8 +1,8 @@
 from rest_framework.viewsets import ModelViewSet
 
-from .serializers import ASNSerializer, BGPSessionSerializer
-from netbox_bgp.models import ASN, BGPSession
-from netbox_bgp.filters import ASNFilterSet, BGPSessionFilterSet
+from .serializers import ASNSerializer, BGPSessionSerializer, RoutingPolicySerializer
+from netbox_bgp.models import ASN, BGPSession, RoutingPolicy
+from netbox_bgp.filters import ASNFilterSet, BGPSessionFilterSet, RoutingPolicyFilterSet
 
 
 class ASNViewSet(ModelViewSet):
@@ -15,3 +15,9 @@ class BGPSessionViewSet(ModelViewSet):
     queryset = BGPSession.objects.all()
     serializer_class = BGPSessionSerializer
     filterset_class = BGPSessionFilterSet
+
+
+class RoutingPolicyViewSet(ModelViewSet):
+    queryset = RoutingPolicy.objects.all()
+    serializer_class = RoutingPolicySerializer
+    filterset_class = RoutingPolicyFilterSet

netbox_bgp/filters.py

@@ -3,7 +3,7 @@
 
 from utilities.filters import TagFilter
 
-from .models import ASN, Community, BGPSession
+from .models import ASN, Community, BGPSession, RoutingPolicy
 
 
 class ASNFilterSet(django_filters.FilterSet):
@@ -77,3 +77,25 @@ def search(self, queryset, name, value):
             | Q(description__icontains=value)
         )
         return queryset.filter(qs_filter)
+
+
+class RoutingPolicyFilterSet(django_filters.FilterSet):
+    q = django_filters.CharFilter(
+        method='search',
+        label='Search',
+    )
+    tag = TagFilter()
+
+    class Meta:
+        model = RoutingPolicy
+        fields = ['name', 'description']
+
+    def search(self, queryset, name, value):
+        """Perform the filtered search."""
+        if not value.strip():
+            return queryset
+        qs_filter = (
+            Q(name__icontains=value)
+            | Q(description__icontains=value)
+        )
+        return queryset.filter(qs_filter)

netbox_bgp/forms.py

@@ -12,7 +12,10 @@
     APISelect, APISelectMultiple, StaticSelect2Multiple, TagFilterField
 )
 
-from .models import ASN, ASNStatusChoices, Community, BGPSession, SessionStatusChoices
+from .models import (
+    ASN, ASNStatusChoices, Community, BGPSession,
+    SessionStatusChoices, RoutingPolicy, BGPPeerGroup
+)
 
 
 class ASNFilterForm(BootstrapMixin, forms.ModelForm):
@@ -208,14 +211,34 @@ class BGPSessionForm(BootstrapMixin, forms.ModelForm):
             'device_id': '$device'
         }
     )
+    import_policies = DynamicModelMultipleChoiceField(
+        queryset=RoutingPolicy.objects.all(),
+        required=False,
+        widget=APISelectMultiple(
+            api_url='/api/plugins/bgp/routing_policy/'
+        )
+    )
+    export_policies = DynamicModelMultipleChoiceField(
+        queryset=RoutingPolicy.objects.all(),
+        required=False,
+        widget=APISelectMultiple(
+            api_url='/api/plugins/bgp/routing_policy/'
+        )
+    )
 
     class Meta:
         model = BGPSession
         fields = [
             'name', 'site', 'device',
             'local_as', 'remote_as', 'local_address', 'remote_address',
-            'description', 'status', 'tenant', 'tags',
+            'description', 'status', 'tenant', 'tags', 'import_policies', 'export_policies'
         ]
+        fieldsets = (
+            ('Session', ('name', 'site', 'device', 'description', 'status', 'tenant', 'tags')),
+            ('Remote', ('remote_as', 'remote_address')),
+            ('Local', ('local_as', 'local_address')),
+            ('Policies', ('import_policies', 'export_policies'))
+        )
 
 
 class BGPSessionAddForm(BGPSessionForm):
@@ -268,3 +291,27 @@ class BGPSessionFilterForm(BootstrapMixin, forms.ModelForm):
     class Meta:
         model = BGPSession
         fields = ['q', 'status', 'tenant', 'remote_as', 'local_as']
+
+
+class RoutingPolicyFilterForm(BootstrapMixin, forms.ModelForm):
+    q = forms.CharField(
+        required=False,
+        label='Search'
+    )
+
+    tag = TagFilterField(RoutingPolicy)
+
+    class Meta:
+        model = RoutingPolicy
+        fields = ['q']
+
+
+class RoutingPolicyForm(BootstrapMixin, forms.ModelForm):
+    tags = DynamicModelMultipleChoiceField(
+        queryset=Tag.objects.all(),
+        required=False
+    )
+
+    class Meta:
+        model = RoutingPolicy
+        fields = ['name', 'description']

netbox_bgp/migrations/0011_netbox_bgp.py

@@ -0,0 +1,50 @@
+# Generated by Django 3.2 on 2021-04-27 06:18
+
+from django.db import migrations, models
+import taggit.managers
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('extras', '0059_exporttemplate_as_attachment'),
+        ('netbox_bgp', '0010_netbox_bgp'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='RoutingPolicy',
+            fields=[
+                ('created', models.DateField(auto_now_add=True, null=True)),
+                ('last_updated', models.DateTimeField(auto_now=True, null=True)),
+                ('id', models.BigAutoField(primary_key=True, serialize=False)),
+                ('name', models.CharField(max_length=100)),
+                ('slug', models.SlugField(max_length=100)),
+                ('description', models.CharField(blank=True, max_length=200)),
+                ('tags', taggit.managers.TaggableManager(through='extras.TaggedItem', to='extras.Tag')),
+            ],
+            options={
+                'verbose_name_plural': 'Routing policies',
+            },
+        ),
+        migrations.AddField(
+            model_name='bgppeergroup',
+            name='export_policies',
+            field=models.ManyToManyField(blank=True, related_name='group_export_policies', to='netbox_bgp.RoutingPolicy'),
+        ),
+        migrations.AddField(
+            model_name='bgppeergroup',
+            name='import_policies',
+            field=models.ManyToManyField(blank=True, related_name='group_import_policies', to='netbox_bgp.RoutingPolicy'),
+        ),
+        migrations.AddField(
+            model_name='bgpsession',
+            name='export_policies',
+            field=models.ManyToManyField(blank=True, related_name='session_export_policies', to='netbox_bgp.RoutingPolicy'),
+        ),
+        migrations.AddField(
+            model_name='bgpsession',
+            name='import_policies',
+            field=models.ManyToManyField(blank=True, related_name='session_import_policies', to='netbox_bgp.RoutingPolicy'),
+        ),
+    ]

netbox_bgp/migrations/0012_netbox_bgp.py

@@ -0,0 +1,22 @@
+# Generated by Django 3.2 on 2021-04-27 11:08
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('netbox_bgp', '0011_netbox_bgp'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='routingpolicy',
+            options={'verbose_name_plural': 'Routing Policies'},
+        ),
+        migrations.AlterField(
+            model_name='bgppeergroup',
+            name='slug',
+            field=models.SlugField(blank=True, max_length=100),
+        ),
+    ]

netbox_bgp/migrations/0013_netbox_bgp.py

@@ -0,0 +1,21 @@
+# Generated by Django 3.2 on 2021-04-27 13:17
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('netbox_bgp', '0012_netbox_bgp'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='bgppeergroup',
+            name='slug',
+        ),
+        migrations.RemoveField(
+            model_name='routingpolicy',
+            name='slug',
+        ),
+    ]