NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-24 17:38:37 -06:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

update openapi field type for choice field value fields to string

Browse Source
This commit is contained in:
John Anderson 2019-12-09 00:41:43 -05:00
parent 88c7d95b08
commit 6fe1b9b37c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
netbox/utilities/custom_inspectors.py
View File

@ -75,7 +75,7 @@ class CustomChoiceFieldInspector(FieldInspector):
SwaggerType, _ = self._get_partial_types(field, swagger_object_type, use_references, **kwargs) SwaggerType, _ = self._get_partial_types(field, swagger_object_type, use_references, **kwargs)
if isinstance(field, ChoiceField): if isinstance(field, ChoiceField):
value_schema = openapi.Schema(type=openapi.TYPE_INTEGER) value_schema = openapi.Schema(type=openapi.TYPE_STRING)
choices = list(field._choices.keys()) choices = list(field._choices.keys())
if set([None] + choices) == {None, True, False}: if set([None] + choices) == {None, True, False}:
@ -83,7 +83,7 @@ class CustomChoiceFieldInspector(FieldInspector):
# differentiated since they each have subtly different values in their choice keys. # differentiated since they each have subtly different values in their choice keys.
# - subdevice_role and connection_status are booleans, although subdevice_role includes None # - subdevice_role and connection_status are booleans, although subdevice_role includes None
# - face is an integer set {0, 1} which is easily confused with {False, True} # - face is an integer set {0, 1} which is easily confused with {False, True}
schema_type = openapi.TYPE_INTEGER schema_type = openapi.TYPE_STRING
if all(type(x) == bool for x in [c for c in choices if c is not None]): if all(type(x) == bool for x in [c for c in choices if c is not None]):
schema_type = openapi.TYPE_BOOLEAN schema_type = openapi.TYPE_BOOLEAN
value_schema = openapi.Schema(type=schema_type) value_schema = openapi.Schema(type=schema_type)