Andreas/wekan
1
0
Fork 0
mirror of https://github.com/wekan/wekan.git synced 2025-09-22 01:50:48 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

generate_openapi: fix enums when they are declared as const

Browse source 
Fixes: #2781
This commit is contained in:
Benjamin Tissoires 2019-11-05 12:56:48 +01:00
parent 8cb3974eff
commit 3f059ec1e0
1 changed files with 35 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

36
openapi/generate_openapi.py
View file

@ -474,7 +474,41 @@ class SchemaProperty(object):
elif p.key.name == 'allowedValues':
self.type = 'enum'
self.enum = [e.value.lower() for e in p.value.elements]
if p.value.type == 'ArrayExpression':
self.enum = [e.value.lower() for e in p.value.elements]
elif p.value.type == 'Identifier':
# tree wide lookout for the identifier
def find_variable(elem, match):
if isinstance(elem, list):
for value in elem:
ret = find_variable(value, match)
if ret is not None:
return ret
try:
items = elem.items()
except AttributeError:
return None
except TypeError:
return None
if (elem.type == 'VariableDeclarator' and
elem.id.name == match):
return elem
for type, value in items:
ret = find_variable(value, match)
if ret is not None:
return ret
return None
elem = find_variable(context.program.body, p.value.name)
if elem.init.type != 'ArrayExpression':
raise TypeError('can not find "{}"'.format(p.value.name))
self.enum = [e.value.lower() for e in elem.init.elements]
elif p.key.name == 'blackbox':
self.blackbox = True