@ -85,14 +85,12 @@ def generateType(path: str, name: str, schema: dict):
|
||||
elif property_type == 'boolean':
|
||||
f.write("\t"+property_name+": Union[Unset, bool] = False\n")
|
||||
else:
|
||||
print(" unknown type: ", property_type)
|
||||
raise
|
||||
raise (" unknown type: ", property_type)
|
||||
elif '$ref' in property_schema:
|
||||
ref = property_schema['$ref'].replace('#/components/schemas/', '')
|
||||
f.write("\t"+property_name+": Union[Unset, "+ref+"] = UNSET\n")
|
||||
else:
|
||||
print(" unknown schema: ", property_schema)
|
||||
raise
|
||||
raise (" unknown schema: ", property_schema)
|
||||
|
||||
# Finish writing the class.
|
||||
f.write("\n")
|
||||
@ -124,16 +122,14 @@ def generateType(path: str, name: str, schema: dict):
|
||||
elif property_type == 'boolean':
|
||||
f.write("\t"+property_name+" = self."+property_name+"\n")
|
||||
else:
|
||||
print(" unknown type: ", property_type)
|
||||
raise
|
||||
raise (" unknown type: ", property_type)
|
||||
elif '$ref' in property_schema:
|
||||
ref = property_schema['$ref'].replace('#/components/schemas/', '')
|
||||
f.write("\t\t"+property_name+": Union[Unset, str] = UNSET\n")
|
||||
f.write("\t\tif not isinstance(self."+property_name+", Unset):\n")
|
||||
f.write("\t\t\t"+property_name+" = self."+property_name+".value\n")
|
||||
else:
|
||||
print(" unknown schema: ", property_schema)
|
||||
raise
|
||||
raise (" unknown schema: ", property_schema)
|
||||
|
||||
# Finish writing the to_dict method.
|
||||
f.write("\n")
|
||||
|
Reference in New Issue
Block a user