How to access Custom Status Choices in Custom Scripts?

[julianstolp]

How do I access FIELD_CHOICES in custom script?
I specified them in configuration.py but if I try to set the status to an object an KeyError is thrown, because the label is not found.

configuration.py
FIELD_CHOICES = { 'ipam.IPAddress.status+': ( ('offline', 'Offline', 'red'), ('online', 'Online', 'green') ) }

This code works in nbshell() but does not work in a custom script.
ip = IPAddress.objects.get(pk=41558)
ip.status = 'offline'
ip.save()

Script Log Exception

An exception occurred: KeyError: 'offline'

Traceback (most recent call last):
  File "/opt/netbox/netbox/extras/scripts.py", line 464, in _run_script
    script.output = script.run(data=data, commit=commit)
  File "/opt/netbox/netbox/scripts/ping_prefix.py", line 48, in run
    ip_obj.save()
  File "/opt/netbox/netbox/ipam/models/ip.py", line 876, in save
    super().save(*args, **kwargs)
  File "/opt/netbox/venv/lib/python3.10/site-packages/django/db/models/base.py", line 812, in save
    self.save_base(
  File "/opt/netbox/venv/lib/python3.10/site-packages/django/db/models/base.py", line 878, in save_base
    post_save.send(
  File "/opt/netbox/venv/lib/python3.10/site-packages/django/dispatch/dispatcher.py", line 176, in send
    return [
  File "/opt/netbox/venv/lib/python3.10/site-packages/django/dispatch/dispatcher.py", line 177, in <listcomp>
    (receiver, receiver(signal=self, sender=sender, **named))
  File "/opt/netbox/netbox/extras/signals.py", line 87, in handle_changed_object
    enqueue_object(queue, instance, request.user, request.id, action)
  File "/opt/netbox/netbox/extras/webhooks.py", line 72, in enqueue_object
    'data': serialize_for_webhook(instance),
  File "/opt/netbox/netbox/extras/webhooks.py", line 27, in serialize_for_webhook
    return serializer.data
  File "/opt/netbox/venv/lib/python3.10/site-packages/rest_framework/serializers.py", line 555, in data
    ret = super().data
  File "/opt/netbox/venv/lib/python3.10/site-packages/rest_framework/serializers.py", line 253, in data
    self._data = self.to_representation(self.instance)
  File "/opt/netbox/venv/lib/python3.10/site-packages/rest_framework/serializers.py", line 522, in to_representation
    ret[field.field_name] = field.to_representation(attribute)
  File "/opt/netbox/netbox/netbox/api/fields.py", line 51, in to_representation
    'label': self._choices[obj],
KeyError: 'offline'

[julianstolp]

Does someone know how to access Custom Status Choices in Custom Scripts?

[julianstolp]

Nervermind. This ip.status = 'offline' works now. Must have forgotten to restart netbox...

[Kani999]

I've run into the same problem.

NBShell worked
GUI worked
but custom scripts not. Restarting the netbox solved the problem.