Merge pull request #14 from koxudaxi/fix_invalid_required

fix invalid required
koxudaxi · Jun 18, 2020 · f8dfd3a · f8dfd3a
2 parents ee6edcf + bb4fc92
commit f8dfd3a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/fastapi_code_generator/parser.py b/fastapi_code_generator/parser.py
@@ -137,7 +137,7 @@ def request_objects(self) -> List[Request]:
                 Request(
                     description=self.requestBody.get("description"),
                     contents=contents,
-                    required=self.requestBody.get("required") == "true",
+                    required=self.requestBody.get("required") is True,
                 )
             )
         return requests
@@ -219,8 +219,7 @@ def get_parameter_type(
         field = DataModelField(
             name=name,
             data_types=[type_map[type_]],
-            required=parameter.get("required") == "true"
-            or parameter.get("in") == "path",
+            required=parameter.get("required") or parameter.get("in") == "path",
             default=schema.typed_default,
         )
         self.imports.extend(field.imports)