NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-23 07:56:44 -06:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

14025 backout changes to model_forms

Browse Source
This commit is contained in:
Arthur 2023-10-17 07:27:18 -07:00
parent 1c942d0538
commit 0d2450868b
1 changed files with 2 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
netbox/core/forms/model_forms.py
View File

@ -1,14 +1,12 @@
import copy import copy
from django import forms from django import forms
from django.core.exceptions import ValidationError
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
from core.forms.mixins import SyncedDataMixin from core.forms.mixins import SyncedDataMixin
from core.models import * from core.models import *
from netbox.forms import NetBoxModelForm from netbox.forms import NetBoxModelForm
from netbox.registry import registry from netbox.registry import registry
from pathlib import Path
from utilities.forms import get_field_value from utilities.forms import get_field_value
from utilities.forms.fields import CommentField from utilities.forms.fields import CommentField
from utilities.forms.widgets import HTMXSelect from utilities.forms.widgets import HTMXSelect
@ -90,34 +88,14 @@ class ManagedFileForm(SyncedDataMixin, NetBoxModelForm):
model = ManagedFile model = ManagedFile
fields = ('data_source', 'data_file', 'auto_sync_enabled') fields = ('data_source', 'data_file', 'auto_sync_enabled')
def _validate_file_extension(self, file):
if file:
extension = Path(file.name).suffix[1:].lower()
if extension != "py":
raise ValidationError(
"File extension “%(extension)s” is not allowed .py extension is required",
code="invalid_extension",
params={
"extension": extension,
},
)
def clean(self): def clean(self):
super().clean() super().clean()
upload_file = self.cleaned_data['upload_file'] if self.cleaned_data.get('upload_file') and self.cleaned_data.get('data_file'):
data_file = self.cleaned_data['data_file']
if upload_file and data_file:
extension = Path(upload_file.name).suffix[1:].lower()
raise forms.ValidationError("Cannot upload a file and sync from an existing file") raise forms.ValidationError("Cannot upload a file and sync from an existing file")
if not upload_file and not data_file: if not self.cleaned_data.get('upload_file') and not self.cleaned_data.get('data_file'):
raise forms.ValidationError("Must upload a file or select a data file to sync") raise forms.ValidationError("Must upload a file or select a data file to sync")
self._validate_file_extension(upload_file)
self._validate_file_extension(data_file)
return self.cleaned_data return self.cleaned_data
def save(self, *args, **kwargs): def save(self, *args, **kwargs):