From 70f8463267b522a81a9f08b0ea6376c61857e478 Mon Sep 17 00:00:00 2001 From: jeremystretch Date: Thu, 17 Mar 2022 14:57:56 -0400 Subject: [PATCH 1/2] Add choice field color methods --- netbox_access_lists/models.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/netbox_access_lists/models.py b/netbox_access_lists/models.py index 057f4f9..82ab04d 100644 --- a/netbox_access_lists/models.py +++ b/netbox_access_lists/models.py @@ -42,6 +42,9 @@ class AccessList(NetBoxModel): def __str__(self): return self.name + def get_default_action_color(self): + return ActionChoices.colors.get(self.default_action) + class AccessListRule(NetBoxModel): access_list = models.ForeignKey( @@ -94,3 +97,9 @@ class AccessListRule(NetBoxModel): def __str__(self): return f'{self.access_list}: Rule {self.index}' + + def get_protocol_color(self): + return ProtocolChoices.colors.get(self.protocol) + + def get_action_color(self): + return ActionChoices.colors.get(self.action) From 890a91b1a4714ea04e9d7d37a40ef1cc9e211bd6 Mon Sep 17 00:00:00 2001 From: jeremystretch Date: Thu, 17 Mar 2022 15:00:31 -0400 Subject: [PATCH 2/2] Add ChoiceFieldColumns --- netbox_access_lists/tables.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/netbox_access_lists/tables.py b/netbox_access_lists/tables.py index cd0bbc2..754a7c1 100644 --- a/netbox_access_lists/tables.py +++ b/netbox_access_lists/tables.py @@ -1,6 +1,6 @@ import django_tables2 as tables -from netbox.tables import NetBoxTable +from netbox.tables import NetBoxTable, ChoiceFieldColumn from .models import AccessList, AccessListRule @@ -8,6 +8,7 @@ class AccessListTable(NetBoxTable): name = tables.Column( linkify=True ) + default_action = ChoiceFieldColumn() rule_count = tables.Column() class Meta(NetBoxTable.Meta): @@ -23,6 +24,8 @@ class AccessListRuleTable(NetBoxTable): index = tables.Column( linkify=True ) + protocol = ChoiceFieldColumn() + action = ChoiceFieldColumn() class Meta(NetBoxTable.Meta): model = AccessListRule