Support unique_items (#746)

* Support unique_items * Fix format
koxudaxi · Apr 16, 2022 · 42e99fe · 42e99fe
1 parent e15e74d
commit 42e99fe
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 5 deletions.
diff --git a/datamodel_code_generator/model/pydantic/base_model.py b/datamodel_code_generator/model/pydantic/base_model.py
@@ -25,6 +25,7 @@ class Constraints(ConstraintsBase):
     min_length: Optional[int] = Field(None, alias='minLength')
     max_length: Optional[int] = Field(None, alias='maxLength')
     regex: Optional[str] = Field(None, alias='pattern')
+    unique_items: Optional[bool] = Field(None, alias='uniqueItems')
 
 
 class DataModelField(DataModelFieldBase):

diff --git a/datamodel_code_generator/parser/jsonschema.py b/datamodel_code_generator/parser/jsonschema.py
@@ -124,6 +124,7 @@ class JsonSchemaObject(BaseModel):
         'minLength',
         'maxLength',
         'pattern',
+        'uniqueItems',
     }
     __extra_key__: str = SPECIAL_PATH_FORMAT.format('extras')
 
@@ -157,7 +158,7 @@ def validate_ref(cls, value: Any) -> Any:
         return value
 
     items: Union[List['JsonSchemaObject'], 'JsonSchemaObject', None]
-    uniqueItem: Optional[bool]
+    uniqueItems: Optional[bool]
     type: Union[str, List[str], None]
     format: Optional[str]
     pattern: Optional[str]

diff --git a/tests/data/expected/main/main_disable_appending_item_suffix/output.py b/tests/data/expected/main/main_disable_appending_item_suffix/output.py
@@ -16,7 +16,7 @@ class Pet(BaseModel):
 
 
 class Pets(BaseModel):
-    __root__: List[Pet] = Field(..., max_items=10, min_items=1)
+    __root__: List[Pet] = Field(..., max_items=10, min_items=1, unique_items=True)
 
 
 class UID(BaseModel):

diff --git a/tests/data/expected/main/main_use_annotated_with_field_constraints/output.py b/tests/data/expected/main/main_use_annotated_with_field_constraints/output.py
@@ -16,7 +16,7 @@ class Pet(BaseModel):
 
 
 class Pets(BaseModel):
-    __root__: Annotated[List[Pet], Field(max_items=10, min_items=1)]
+    __root__: Annotated[List[Pet], Field(max_items=10, min_items=1, unique_items=True)]
 
 
 class UID(BaseModel):

diff --git a/tests/data/expected/main/main_with_field_constraints/output.py b/tests/data/expected/main/main_with_field_constraints/output.py
@@ -16,7 +16,7 @@ class Pet(BaseModel):
 
 
 class Pets(BaseModel):
-    __root__: List[Pet] = Field(..., max_items=10, min_items=1)
+    __root__: List[Pet] = Field(..., max_items=10, min_items=1, unique_items=True)
 
 
 class UID(BaseModel):

diff --git a/tests/data/expected/main/main_without_field_constraints/output.py b/tests/data/expected/main/main_without_field_constraints/output.py
@@ -16,7 +16,7 @@ class Pet(BaseModel):
 
 
 class Pets(BaseModel):
-    __root__: List[Pet] = Field(..., max_items=10, min_items=1)
+    __root__: List[Pet] = Field(..., max_items=10, min_items=1, unique_items=True)
 
 
 class UID(BaseModel):

diff --git a/tests/data/openapi/api_constrained.yaml b/tests/data/openapi/api_constrained.yaml
@@ -122,6 +122,7 @@ components:
         $ref: "#/components/schemas/Pet"
       maxItems: 10
       minItems: 1
+      uniqueItems: true
     UID:
       type: integer
       minimum: 0