mirror of
https://github.com/lucaspalomodevelop/netbox-acls.git
synced 2026-03-12 23:27:23 +00:00
134 lines
3.3 KiB
Python
134 lines
3.3 KiB
Python
from django.contrib.postgres.fields import ArrayField
|
|
from django.db import models
|
|
from django.urls import reverse
|
|
|
|
from netbox.models import NetBoxModel
|
|
from utilities.choices import ChoiceSet
|
|
|
|
|
|
class AccessListActionChoices(ChoiceSet):
|
|
key = 'AccessListRule.action'
|
|
ACTION_DENY = 'deny'
|
|
ACTION_PERMIT = 'permit'
|
|
ACTION_REJECT = 'reject'
|
|
|
|
CHOICES = [
|
|
(ACTION_DENY, 'Deny', 'red'),
|
|
(ACTION_PERMIT, 'Permit', 'green'),
|
|
(ACTION_REJECT, 'Reject (Reset)', 'orange'),
|
|
]
|
|
|
|
|
|
class AccessListTypeChoices(ChoiceSet):
|
|
|
|
CHOICES = [
|
|
('extended', 'Extended', 'purple'),
|
|
('standard', 'Standard', 'blue'),
|
|
]
|
|
|
|
|
|
class AccessListProtocolChoices(ChoiceSet):
|
|
|
|
CHOICES = [
|
|
('icmp', 'ICMP', 'purple'),
|
|
('tcp', 'TCP', 'blue'),
|
|
('udp', 'UDP', 'orange'),
|
|
]
|
|
|
|
|
|
class AccessList(NetBoxModel):
|
|
name = models.CharField(
|
|
max_length=100
|
|
)
|
|
type = models.CharField(
|
|
max_length=30,
|
|
choices=AccessListTypeChoices
|
|
)
|
|
default_action = models.CharField(
|
|
default=AccessListActionChoices.ACTION_DENY,
|
|
max_length=30,
|
|
choices=AccessListActionChoices
|
|
)
|
|
comments = models.TextField(
|
|
blank=True
|
|
)
|
|
|
|
class Meta:
|
|
ordering = ('name',)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('plugins:netbox_access_lists:accesslist', args=[self.pk])
|
|
|
|
def get_default_action_color(self):
|
|
return AccessListActionChoices.colors.get(self.default_action)
|
|
|
|
def get_type_color(self):
|
|
return AccessListTypeChoices.colors.get(self.type)
|
|
|
|
|
|
class AccessListRule(NetBoxModel):
|
|
access_list = models.ForeignKey(
|
|
on_delete=models.CASCADE,
|
|
related_name='rules',
|
|
to=AccessList,
|
|
)
|
|
index = models.PositiveIntegerField()
|
|
protocol = models.CharField(
|
|
blank=True,
|
|
choices=AccessListProtocolChoices,
|
|
max_length=30,
|
|
)
|
|
source_prefix = models.ForeignKey(
|
|
blank=True,
|
|
null=True,
|
|
on_delete=models.PROTECT,
|
|
related_name='+',
|
|
to='ipam.Prefix',
|
|
)
|
|
source_ports = ArrayField(
|
|
base_field=models.PositiveIntegerField(),
|
|
blank=True,
|
|
null=True,
|
|
)
|
|
destination_prefix = models.ForeignKey(
|
|
blank=True,
|
|
null=True,
|
|
on_delete=models.PROTECT,
|
|
related_name='+',
|
|
to='ipam.Prefix',
|
|
)
|
|
destination_ports = ArrayField(
|
|
base_field=models.PositiveIntegerField(),
|
|
blank=True,
|
|
null=True,
|
|
)
|
|
action = models.CharField(
|
|
choices=AccessListActionChoices,
|
|
default=AccessListActionChoices.ACTION_PERMIT,
|
|
max_length=30,
|
|
)
|
|
remark = models.CharField(
|
|
max_length=200,
|
|
blank=True,
|
|
null=True
|
|
)
|
|
|
|
class Meta:
|
|
ordering = ('access_list', 'index')
|
|
unique_together = ('access_list', 'index')
|
|
|
|
def __str__(self):
|
|
return f'{self.access_list}: Rule {self.index}'
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('plugins:netbox_access_lists:accesslistrule', args=[self.pk])
|
|
|
|
def get_protocol_color(self):
|
|
return AccessListProtocolChoices.colors.get(self.protocol)
|
|
|
|
def get_action_color(self):
|
|
return AccessListActionChoices.colors.get(self.action)
|