feat: add tenant groups and required tenancy
This commit is contained in:
@@ -4,7 +4,7 @@ from netbox.plugins import get_plugin_config
|
||||
|
||||
from .models import UtilitiesSettings
|
||||
|
||||
SETTINGS_CACHE_KEY = "netbox_utilities.runtime_settings.v1"
|
||||
SETTINGS_CACHE_KEY = "netbox_utilities.runtime_settings.v2"
|
||||
|
||||
|
||||
def navigation_customization_enabled():
|
||||
@@ -15,18 +15,34 @@ def tenant_filter_enabled():
|
||||
if not get_plugin_config("netbox_utilities", "tenant_filter_enabled"):
|
||||
return False
|
||||
|
||||
enabled = cache.get(SETTINGS_CACHE_KEY)
|
||||
if enabled is None:
|
||||
return bool(_get_database_settings()["tenant_filter_enabled"])
|
||||
|
||||
|
||||
def tenant_required():
|
||||
if not get_plugin_config("netbox_utilities", "tenant_required"):
|
||||
return False
|
||||
|
||||
return bool(_get_database_settings()["tenant_required"])
|
||||
|
||||
|
||||
def _get_database_settings():
|
||||
runtime_settings = cache.get(SETTINGS_CACHE_KEY)
|
||||
if runtime_settings is None:
|
||||
try:
|
||||
enabled = (
|
||||
UtilitiesSettings.objects.filter(singleton_id=1).values_list("tenant_filter_enabled", flat=True).first()
|
||||
runtime_settings = (
|
||||
UtilitiesSettings.objects.filter(singleton_id=1)
|
||||
.values("tenant_filter_enabled", "tenant_required")
|
||||
.first()
|
||||
)
|
||||
except (OperationalError, ProgrammingError):
|
||||
# The middleware can be imported before the first plugin migration.
|
||||
enabled = True
|
||||
enabled = True if enabled is None else enabled
|
||||
cache.set(SETTINGS_CACHE_KEY, enabled, timeout=30)
|
||||
return bool(enabled)
|
||||
runtime_settings = None
|
||||
runtime_settings = runtime_settings or {
|
||||
"tenant_filter_enabled": True,
|
||||
"tenant_required": True,
|
||||
}
|
||||
cache.set(SETTINGS_CACHE_KEY, runtime_settings, timeout=30)
|
||||
return runtime_settings
|
||||
|
||||
|
||||
def clear_runtime_settings_cache():
|
||||
|
||||
Reference in New Issue
Block a user