diff --git a/netbox/core/models/files.py b/netbox/core/models/files.py index 38d82463e..a9e0e7f00 100644 --- a/netbox/core/models/files.py +++ b/netbox/core/models/files.py @@ -2,6 +2,7 @@ import logging import os from django.conf import settings +from django.core.exceptions import ValidationError from django.db import models from django.urls import reverse from django.utils.translation import gettext as _ @@ -84,6 +85,14 @@ class ManagedFile(SyncedDataMixin, models.Model): self.file_path = os.path.basename(self.data_path) self.data_file.write_to_disk(self.full_path, overwrite=True) + def clean(self): + super().clean() + + # Ensure that the file root and path make a unique pair + if self._meta.model.objects.filter(file_root=self.file_root, file_path=self.file_path).exclude(pk=self.pk).exists(): + raise ValidationError( + f"A {self._meta.verbose_name.lower()} with this file path already exists ({self.file_root}/{self.file_path}).") + def delete(self, *args, **kwargs): # Delete file from disk try: diff --git a/netbox/extras/models/scripts.py b/netbox/extras/models/scripts.py index 443dabdb3..122f56f20 100644 --- a/netbox/extras/models/scripts.py +++ b/netbox/extras/models/scripts.py @@ -2,7 +2,6 @@ import inspect import logging from functools import cached_property -from django.core.exceptions import ValidationError from django.db import models from django.urls import reverse from django.utils.translation import gettext_lazy as _ @@ -77,14 +76,6 @@ class ScriptModule(PythonModuleMixin, JobsMixin, ManagedFile): return scripts - def clean(self): - super().clean() - - # Ensure that the file root and path make a unique pair - if self.file_root and self.file_path: - if ScriptModule.objects.filter(file_root=self.file_root, file_path=self.file_path).exclude(pk=self.pk).exists(): - raise ValidationError(f"A script module with this file path already exists ({self.file_root}/{self.file_path}).") - def save(self, *args, **kwargs): self.file_root = ManagedFileRootPathChoices.SCRIPTS return super().save(*args, **kwargs)