Skip to content

Commit 1dd426c

Browse files
committed
feat: Adding new Destination-Fields to ACLExtended Rule
1 parent 84918ab commit 1dd426c

File tree

1 file changed

+41
-2
lines changed

1 file changed

+41
-2
lines changed

netbox_acls/models/access_list_rules.py

Lines changed: 41 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -192,9 +192,41 @@ class ACLExtendedRule(ACLRule):
192192
null=True,
193193
on_delete=models.PROTECT,
194194
related_name="+",
195-
to="ipam.Prefix",
195+
to=Prefix,
196196
verbose_name="Destination Prefix",
197197
)
198+
destination_iprange = models.ForeignKey(
199+
blank=True,
200+
null=True,
201+
on_delete=models.PROTECT,
202+
related_name="+",
203+
to=IPRange,
204+
verbose_name="Destination IP-Range",
205+
)
206+
destination_ipaddress = models.ForeignKey(
207+
blank=True,
208+
null=True,
209+
on_delete=models.PROTECT,
210+
related_name="+",
211+
to=IPAddress,
212+
verbose_name="Destination IP-Address",
213+
)
214+
destination_aggregate = models.ForeignKey(
215+
blank=True,
216+
null=True,
217+
on_delete=models.PROTECT,
218+
related_name="+",
219+
to=Aggregate,
220+
verbose_name="Destination Aggregate",
221+
)
222+
destination_service = models.ForeignKey(
223+
blank=True,
224+
null=True,
225+
on_delete=models.PROTECT,
226+
related_name="+",
227+
to=Service,
228+
verbose_name="Destination Service",
229+
)
198230
destination_ports = ArrayField(
199231
base_field=models.PositiveIntegerField(),
200232
blank=True,
@@ -219,7 +251,14 @@ def get_protocol_color(self):
219251

220252
@classmethod
221253
def get_prerequisite_models(cls):
222-
return [apps.get_model("ipam.Prefix"), AccessList]
254+
return [
255+
Prefix,
256+
IPRange,
257+
IPAddress,
258+
Aggregate,
259+
Service,
260+
AccessList
261+
]
223262

224263
class Meta(ACLRule.Meta):
225264
"""

0 commit comments

Comments
 (0)