|
3 | 3 | from netbox.plugins import PluginTemplateExtension |
4 | 4 | from netbox.views.generic import ObjectChildrenView |
5 | 5 | from utilities.views import ViewTab, register_model_view |
| 6 | +from virtualization.models import VirtualMachine |
6 | 7 |
|
7 | 8 | from .filtersets import BGPSessionFilterSet |
8 | 9 | from .models import BGPSession |
@@ -112,6 +113,30 @@ def get_children( |
112 | 113 | return BGPSession.objects.filter(tenant=parent) |
113 | 114 |
|
114 | 115 |
|
| 116 | +@register_model_view(VirtualMachine, name="bgp-sessions", path="bgp-sessions") |
| 117 | +class VirtualMachineBGPSessionsView(generic.ObjectChildrenView): |
| 118 | + """View to display BGP sessions associated with a virtual machine.""" |
| 119 | + |
| 120 | + queryset = VirtualMachine.objects.all() |
| 121 | + child_model = BGPSession |
| 122 | + filterset = BGPSessionFilterSet |
| 123 | + table = BGPSessionTable |
| 124 | + template_name = "generic/object_children.html" |
| 125 | + hide_if_empty = False |
| 126 | + |
| 127 | + tab = ViewTab( |
| 128 | + label="BGP Sessions", |
| 129 | + badge=lambda obj: BGPSession.objects.filter(virtualmachine=obj).count(), |
| 130 | + permission="netbox_bgp.view_bgpsession", |
| 131 | + ) |
| 132 | + |
| 133 | + def get_children( |
| 134 | + self, request: HttpRequest, parent: VirtualMachine |
| 135 | + ) -> QuerySet[BGPSession]: |
| 136 | + """Get BGP sessions for the virtual machine.""" |
| 137 | + return BGPSession.objects.filter(virtualmachine=parent) |
| 138 | + |
| 139 | + |
115 | 140 | # Register only when device_ext_page is set to 'tab'; |
116 | 141 | class DeviceBGPSessionsView(generic.ObjectChildrenView): |
117 | 142 | """View to display BGP sessions associated with a device.""" |
|
0 commit comments