Kaynağa Gözat

Fix inclusion of legacy IDs on choice fields

Jeremy Stretch 6 yıl önce
ebeveyn
işleme
f0a6c881bc
1 değiştirilmiş dosya ile 2 ekleme ve 1 silme
  1. 2 1
      netbox/utilities/api.py

+ 2 - 1
netbox/utilities/api.py

@@ -86,8 +86,9 @@ class ChoiceField(Field):
             ('label', self._choices[obj])
             ('label', self._choices[obj])
         ])
         ])
 
 
+        # TODO: Remove in v2.8
         # Include legacy numeric ID (where applicable)
         # Include legacy numeric ID (where applicable)
-        if type(self.choiceset) is ChoiceSet and obj in self.choiceset.LEGACY_MAP:
+        if hasattr(self.choiceset, 'LEGACY_MAP') and obj in self.choiceset.LEGACY_MAP:
             data['id'] = self.choiceset.LEGACY_MAP.get(obj)
             data['id'] = self.choiceset.LEGACY_MAP.get(obj)
 
 
         return data
         return data