datamodel-code-generator 0.28.0__tar.gz → 0.28.1__tar.gz

{datamodel_code_generator-0.28.0 → datamodel_code_generator-0.28.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datamodel-code-generator
-Version: 0.28.0
+Version: 0.28.1
 Summary: Datamodel Code Generator
 Project-URL: Homepage, https://github.com/koxudaxi/datamodel-code-generator
 Project-URL: Source, https://github.com/koxudaxi/datamodel-code-generator

{datamodel_code_generator-0.28.0 → datamodel_code_generator-0.28.1}/src/datamodel_code_generator/parser/graphql.py

@@ -386,9 +386,15 @@ class GraphQLParser(Parser):
             elif graphql.is_non_null_type(obj):  # pragma: no cover
                 data_type.is_optional = False
 
-            obj = obj.of_type  # pyright: ignore[reportAttributeAccessIssue]
+            obj = graphql.assert_wrapping_type(obj)
+            obj = obj.of_type
 
-        data_type.type = obj.name  # pyright: ignore[reportAttributeAccessIssue]
+        if graphql.is_enum_type(obj):
+            obj = graphql.assert_enum_type(obj)
+            data_type.reference = self.references[obj.name]
+
+        obj = graphql.assert_named_type(obj)
+        data_type.type = obj.name
 
         required = (not self.force_optional_for_required_fields) and (not final_data_type.is_optional)
 

datamodel_code_generator-0.28.1/tests/data/expected/parser/graphql/field-default-enum.py

@@ -0,0 +1,33 @@
+# generated by datamodel-codegen:
+#   filename:  field-default-enum.graphql
+#   timestamp: 2019-07-26T00:00:00+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import List, Literal, Optional, TypeAlias
+
+from pydantic import BaseModel, Field
+
+Boolean: TypeAlias = bool
+"""
+The `Boolean` scalar type represents `true` or `false`.
+"""
+
+
+String: TypeAlias = str
+"""
+The `String` scalar type represents textual data, represented as UTF-8 character sequences. The String type is most often used by GraphQL to represent free-form human-readable text.
+"""
+
+
+class Color(Enum):
+    BLUE = 'BLUE'
+    GREEN = 'GREEN'
+    RED = 'RED'
+
+
+class Car(BaseModel):
+    colorList: Optional[List[Color]] = [Color.RED]
+    colorOne: Optional[Color] = Color.GREEN
+    typename__: Optional[Literal['Car']] = Field('Car', alias='__typename')

datamodel_code_generator-0.28.1/tests/data/graphql/field-default-enum.graphql

@@ -0,0 +1,10 @@
+enum Color {
+  RED
+  GREEN
+  BLUE
+}
+
+input Car {
+  colorList: [Color!] = [RED]
+  colorOne: Color = GREEN
+}

datamodel_code_generator-0.28.1/tests/parser/test_graphql.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+from freezegun import freeze_time
+
+from datamodel_code_generator.__main__ import Exit, main
+from tests.main.test_main_general import DATA_PATH
+
+GRAPHQL_DATA_PATH: Path = DATA_PATH / "graphql"
+EXPECTED_GRAPHQL_PATH: Path = DATA_PATH / "expected" / "parser" / "graphql"
+
+
+@freeze_time("2019-07-26")
+def test_graphql_field_enum() -> None:
+    with TemporaryDirectory() as output_dir:
+        output_file: Path = Path(output_dir) / "output.py"
+        return_code: Exit = main([
+            "--input",
+            str(GRAPHQL_DATA_PATH / "field-default-enum.graphql"),
+            "--output",
+            str(output_file),
+            "--input-file-type",
+            "graphql",
+            "--set-default-enum-member",
+        ])
+        assert return_code == Exit.OK
+        assert output_file.read_text() == (EXPECTED_GRAPHQL_PATH / "field-default-enum.py").read_text()