added tenants (#94)

netbox_topology_views/__init__.py

@@ -4,7 +4,7 @@ class TopologyViewsConfig(PluginConfig):
     name = 'netbox_topology_views'
     verbose_name = 'Topology views'
     description = 'An plugin to render topology maps'
-    version = '1.0.2'
+    version = '1.0.3'
     author = 'Mattijs Vanhaverbeke'
     author_email = 'author@example.com'
     base_url = 'netbox_topology_views'

netbox_topology_views/filters.py

@@ -1,10 +1,12 @@
 import django_filters
 from extras.filters import TagFilter
 from dcim.models import Device, DeviceRole, Region, Site, Location
+from tenancy.models import TenantGroup, Tenant
+from tenancy.filtersets import TenancyFilterSet
 from utilities.filters import TreeNodeMultipleChoiceFilter
 from django.db.models import Q
 
-class DeviceFilterSet(django_filters.FilterSet):
+class DeviceFilterSet(TenancyFilterSet, django_filters.FilterSet):
     q = django_filters.CharFilter(
         method='search',
         label='Search',

netbox_topology_views/forms.py

@@ -1,18 +1,22 @@
+import imp
 from django import forms
 from django.conf import settings
 from django.utils.translation import gettext as _
 
 from extras.models import Tag
 from dcim.models import Device, Site, Region, DeviceRole, Location
+from tenancy.models import TenantGroup, Tenant
+from tenancy.forms import TenancyFilterForm
 from django.conf import settings
 
 from utilities.forms import (TagFilterField,  DynamicModelMultipleChoiceField, FilterForm)
 allow_coordinates_saving = settings.PLUGINS_CONFIG["netbox_topology_views"]["allow_coordinates_saving"]
 
-class DeviceFilterForm(FilterForm):
+class DeviceFilterForm(TenancyFilterForm, FilterForm):
     model = Device
     field_groups = [
         ['q', 'hide_unconnected', 'save_coords'],
+        ['tenant_group_id', 'tenant_id'],
         ['region_id', 'site_id', 'location_id'],
         ['device_role_id'],
         ['tag'],