mirror of
https://github.com/lucaspalomodevelop/netbox-acls.git
synced 2026-03-13 07:29:40 +00:00
35 lines
972 B
Python
35 lines
972 B
Python
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
from extras.plugins import PluginTemplateExtension
|
|
from .models import AccessList
|
|
|
|
|
|
class AccessLists(PluginTemplateExtension):
|
|
|
|
def right_page(self):
|
|
obj = self.context['object']
|
|
|
|
access_lists = None
|
|
ctype = ContentType.objects.get_for_model(obj)
|
|
if ctype.model == 'device':
|
|
access_lists = AccessList.objects.filter(device=obj.pk)
|
|
#elif ctype.model == 'virtualmachine':
|
|
# access_lists = AccessList.objects.filter(device=obj.pk)
|
|
|
|
return self.render('inc/device/access_lists.html', extra_context={
|
|
'access_lists': access_lists,
|
|
'type': ctype.model if ctype.model == 'device' else ctype.name.replace(' ', '_'),
|
|
})
|
|
|
|
|
|
class DeviceAccessLists(AccessLists):
|
|
model = 'dcim.device'
|
|
|
|
|
|
#class VMAccessLists(AccessLists):
|
|
# model = 'virtualization.virtualmachine'
|
|
|
|
|
|
template_extensions = [DeviceAccessLists]
|