diff --git a/gracedb/superevents/api/fields.py b/gracedb/superevents/api/fields.py index f2a4b7f1d9bb1164338ec44e5af69856331dc98b..2cce706c616fb330aaa11b443978a9467d7794c9 100644 --- a/gracedb/superevents/api/fields.py +++ b/gracedb/superevents/api/fields.py @@ -64,3 +64,12 @@ class CommaSeparatedOrListField(fields.ListField): data = data[0].split(',') return super(CommaSeparatedOrListField, self).to_internal_value(data) + +class ChoiceDisplayField(fields.ChoiceField): + """ + Same as standard choice field, but return a choice's display_value + instead of the key when serializing the field. + """ + + def to_representation(self, value): + return self._choices[value]