mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
37 lines
808 B
Python
37 lines
808 B
Python
from django.db import models
|
|
from timezone_field import TimeZoneField
|
|
|
|
from netbox.config import ConfigItem
|
|
|
|
|
|
SKIP_FIELDS = (
|
|
TimeZoneField,
|
|
)
|
|
|
|
EXEMPT_ATTRS = (
|
|
'choices',
|
|
'help_text',
|
|
'verbose_name',
|
|
)
|
|
|
|
_deconstruct = models.Field.deconstruct
|
|
|
|
|
|
def custom_deconstruct(field):
|
|
"""
|
|
Imitate the behavior of the stock deconstruct() method, but ignore the field attributes listed above.
|
|
"""
|
|
name, path, args, kwargs = _deconstruct(field)
|
|
|
|
# Remove any ignored attributes
|
|
if field.__class__ not in SKIP_FIELDS:
|
|
for attr in EXEMPT_ATTRS:
|
|
kwargs.pop(attr, None)
|
|
|
|
# Ignore any field defaults which reference a ConfigItem
|
|
kwargs = {
|
|
k: v for k, v in kwargs.items() if not isinstance(v, ConfigItem)
|
|
}
|
|
|
|
return name, path, args, kwargs
|