mirror of
https://github.com/netbox-community/netbox.git
synced 2025-12-21 11:46:37 +00:00
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from utilities.forms import unpack_grouped_choices
|
|
|
|
|
|
class ChoiceSetMeta(type):
|
|
"""
|
|
Metaclass for ChoiceSet
|
|
"""
|
|
def __call__(cls, *args, **kwargs):
|
|
# Django will check if a 'choices' value is callable, and if so assume that it returns an iterable
|
|
return getattr(cls, 'CHOICES', ())
|
|
|
|
def __iter__(cls):
|
|
choices = getattr(cls, 'CHOICES', ())
|
|
return iter(choices)
|
|
|
|
|
|
class ChoiceSet(metaclass=ChoiceSetMeta):
|
|
|
|
CHOICES = list()
|
|
LEGACY_MAP = dict()
|
|
|
|
@classmethod
|
|
def values(cls):
|
|
return [c[0] for c in cls.CHOICES]
|
|
|
|
@classmethod
|
|
def as_dict(cls):
|
|
# Unpack grouped choices before casting as a dict
|
|
return dict(unpack_grouped_choices(cls.CHOICES))
|
|
|
|
@classmethod
|
|
def slug_to_id(cls, slug):
|
|
"""
|
|
Return the legacy integer value corresponding to a slug.
|
|
"""
|
|
return cls.LEGACY_MAP.get(slug)
|
|
|
|
@classmethod
|
|
def id_to_slug(cls, legacy_id):
|
|
"""
|
|
Return the slug value corresponding to a legacy integer value.
|
|
"""
|
|
if legacy_id in cls.LEGACY_MAP.values():
|
|
# Invert the legacy map to allow lookup by integer
|
|
legacy_map = dict([
|
|
(id, slug) for slug, id in cls.LEGACY_MAP.items()
|
|
])
|
|
return legacy_map.get(legacy_id)
|