diff --git a/netbox_vault_secrets/template_content.py b/netbox_vault_secrets/template_content.py index 0bd08f9..cc3265d 100644 --- a/netbox_vault_secrets/template_content.py +++ b/netbox_vault_secrets/template_content.py @@ -30,4 +30,13 @@ class VaultVMSecrets(VaultSecretInserter): model = 'virtualization.virtualmachine' -template_extensions = [VaultDeviceSecrets, VaultServiceSecrets, VaultVMSecrets] +class VaultTenantSecrets(VaultSecretInserter): + vault_path_slug = "tenant" + model = "tenancy.tenant" + +class VaultSiteSecrets(VaultSecretInserter): + vault_path_slug = "site" + model = "dcim.site" + +template_extensions = [VaultDeviceSecrets, VaultServiceSecrets, VaultVMSecrets,VaultTenantSecrets,VaultSiteSecrets] +