mirror of
https://github.com/netbox-community/netbox.git
synced 2026-04-26 18:58:54 +02:00
945e7ade0a
- Adopt Ruff `I` (isort) rules for consistent import sorting - Add two `# isort: split` boundaries to keep required imports pinned in `__init__.py` modules
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
from django import forms
|
|
from django.contrib.postgres.forms import SimpleArrayField
|
|
from django.utils.safestring import mark_safe
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from utilities.data import ranges_to_string, string_to_ranges
|
|
|
|
from ..utils import parse_numeric_range
|
|
|
|
__all__ = (
|
|
'NumericArrayField',
|
|
'NumericRangeArrayField',
|
|
)
|
|
|
|
|
|
class NumericArrayField(SimpleArrayField):
|
|
|
|
def clean(self, value):
|
|
if value and not self.to_python(value):
|
|
raise forms.ValidationError(
|
|
_("Invalid list ({value}). Must be numeric and ranges must be in ascending order.").format(value=value)
|
|
)
|
|
return super().clean(value)
|
|
|
|
def to_python(self, value):
|
|
if not value:
|
|
return []
|
|
if isinstance(value, str):
|
|
value = ','.join([str(n) for n in parse_numeric_range(value)])
|
|
return super().to_python(value)
|
|
|
|
|
|
class NumericRangeArrayField(forms.CharField):
|
|
"""
|
|
A field which allows for array of numeric ranges:
|
|
Example: 1-5,10,20-30
|
|
"""
|
|
def __init__(self, *args, help_text='', **kwargs):
|
|
if not help_text:
|
|
help_text = mark_safe(
|
|
_(
|
|
"Specify one or more individual numbers or numeric ranges separated by commas. Example: {example}"
|
|
).format(example="<code>1-5,10,20-30</code>")
|
|
)
|
|
super().__init__(*args, help_text=help_text, **kwargs)
|
|
|
|
def clean(self, value):
|
|
if value and not self.to_python(value):
|
|
raise forms.ValidationError(
|
|
_("Invalid ranges ({value}). Must be a range of integers in ascending order.").format(value=value)
|
|
)
|
|
return super().clean(value)
|
|
|
|
def prepare_value(self, value):
|
|
if isinstance(value, str):
|
|
return value
|
|
return ranges_to_string(value)
|
|
|
|
def to_python(self, value):
|
|
return string_to_ranges(value)
|