update generate.py to handle query params
This commit is contained in:
@ -231,7 +231,9 @@ response: Response[""" + success_type + """] = await """ + fn_name + """.asyncio
|
|||||||
f.write("\t*,\n")
|
f.write("\t*,\n")
|
||||||
f.write("\tclient: Client,\n")
|
f.write("\tclient: Client,\n")
|
||||||
f.write(") -> Dict[str, Any]:\n")
|
f.write(") -> Dict[str, Any]:\n")
|
||||||
f.write("\turl = \"{}" + name + "\".format(client.base_url")
|
templateUrl = "{}" + name
|
||||||
|
formatTemplate = ".format(client.base_url"
|
||||||
|
queryTemplate = ""
|
||||||
# Iterate over the parameters.
|
# Iterate over the parameters.
|
||||||
if 'parameters' in endpoint:
|
if 'parameters' in endpoint:
|
||||||
parameters = endpoint['parameters']
|
parameters = endpoint['parameters']
|
||||||
@ -246,12 +248,20 @@ response: Response[""" + success_type + """] = await """ + fn_name + """.asyncio
|
|||||||
else:
|
else:
|
||||||
print(" parameter: ", parameter)
|
print(" parameter: ", parameter)
|
||||||
raise Exception("Unknown parameter type")
|
raise Exception("Unknown parameter type")
|
||||||
f.write(
|
if parameter['in'] == 'path':
|
||||||
", " +
|
formatTemplate = formatTemplate + ", " + \
|
||||||
parameter_name +
|
parameter_name + "=" + camel_to_snake(parameter_name)
|
||||||
"=" +
|
elif parameter['in'] == 'query':
|
||||||
camel_to_snake(parameter_name))
|
formatTemplate = formatTemplate + ", " + \
|
||||||
f.write(")\n")
|
parameter_name + "=" + camel_to_snake(parameter_name)
|
||||||
|
queryTemplate = queryTemplate + "&" + \
|
||||||
|
camel_to_snake(parameter_name) + "=" + "{" + camel_to_snake(parameter_name) + "}"
|
||||||
|
if queryTemplate[1:].__len__() > 0:
|
||||||
|
f.write("\turl = \"" + templateUrl + "?" +
|
||||||
|
queryTemplate[1:] + "\"" + formatTemplate + ")\n")
|
||||||
|
else:
|
||||||
|
f.write("\turl = \"" + templateUrl + "\"" + formatTemplate + ")\n")
|
||||||
|
|
||||||
f.write("\n")
|
f.write("\n")
|
||||||
f.write("\theaders: Dict[str, Any] = client.get_headers()\n")
|
f.write("\theaders: Dict[str, Any] = client.get_headers()\n")
|
||||||
f.write("\tcookies: Dict[str, Any] = client.get_cookies()\n")
|
f.write("\tcookies: Dict[str, Any] = client.get_cookies()\n")
|
||||||
|
Reference in New Issue
Block a user