text: from typing import Union, List, Optional
  
  from pyspark.sql.types import StructType, StructField, StringType, ArrayType, DataType
  
  
  # This file is auto-generated by generate_schema so do not edit manually
  # noinspection PyPep8Naming
  class GraphDefinition_CompartmentSchema:
      """
      A formal computable definition of a graph of resources - that is, a coherent
      set of resources that form a graph by following references. The Graph
      Definition resource defines a set and makes rules about the set.
      """
  
      # noinspection PyDefaultArgument
      @staticmethod
      def get_schema(
          max_nesting_depth: Optional[int] = 6,
          nesting_depth: int = 0,
          nesting_list: List[str] = [],
          max_recursion_limit: Optional[int] = 2,
          include_extension: Optional[bool] = False,
          extension_fields: Optional[List[str]] = [
              "valueBoolean",
              "valueCode",
              "valueDate",
              "valueDateTime",
              "valueDecimal",
              "valueId",
              "valueInteger",
              "valuePositiveInt",
              "valueString",
              "valueTime",
              "valueUnsignedInt",
              "valueUri",
              "valueQuantity",
          ],
          extension_depth: int = 0,
          max_extension_depth: Optional[int] = 2,
      ) -> Union[StructType, DataType]:
          """
          A formal computable definition of a graph of resources - that is, a coherent
          set of resources that form a graph by following references. The Graph
          Definition resource defines a set and makes rules about the set.
  
  
          id: unique id for the element within a resource (for internal references). This
              may be any string value that does not contain spaces.
  
          extension: May be used to represent additional information that is not part of the basic
              definition of the element. In order to make the use of extensions safe and
              manageable, there is a strict set of governance  applied to the definition and
              use of extensions. Though any implementer is allowed to define an extension,
              there is a set of requirements that SHALL be met as part of the definition of
              the extension.
  
          code: Identifies the compartment.
  
          rule: identical | matching | different | no-rule | custom.
  
          expression: Custom rule, as a FHIRPath expression.
  
          description: Documentation for FHIRPath expression.
  
          """
          from spark_fhir_schemas.stu3.complex_types.extension import ExtensionSchema
  
          if (
              max_recursion_limit
              and nesting_list.count("GraphDefinition_Compartment") >= max_recursion_limit
          ) or (max_nesting_depth and nesting_depth >= max_nesting_depth):
              return StructType([StructField("id", StringType(), True)])
          # add my name to recursion list for later
          my_nesting_list: List[str] = nesting_list + ["GraphDefinition_Compartment"]
          schema = StructType(
              [
                  # unique id for the element within a resource (for internal references). This
                  # may be any string value that does not contain spaces.
                  StructField("id", StringType(), True),
                  # May be used to represent additional information that is not part of the basic
                  # definition of the element. In order to make the use of extensions safe and
                  # manageable, there is a strict set of governance  applied to the definition and
                  # use of extensions. Though any implementer is allowed to define an extension,
                  # there is a set of requirements that SHALL be met as part of the definition of
                  # the extension.
                  StructField(
                      "extension",
                      ArrayType(
                          ExtensionSchema.get_schema(
                              max_nesting_depth=max_nesting_depth,
                              nesting_depth=nesting_depth + 1,
                              nesting_list=my_nesting_list,
                              max_recursion_limit=max_recursion_limit,
                              include_extension=include_extension,
                              extension_fields=extension_fields,
                              extension_depth=extension_depth,
                              max_extension_depth=max_extension_depth,
                          )
                      ),
                      True,
                  ),
                  # Identifies the compartment.
                  StructField("code", StringType(), True),
                  # identical | matching | different | no-rule | custom.
                  StructField("rule", StringType(), True),
                  # Custom rule, as a FHIRPath expression.
                  StructField("expression", StringType(), True),
                  # Documentation for FHIRPath expression.
                  StructField("description", StringType(), True),
              ]
          )
          if not include_extension:
              schema.fields = [
                  c
                  if c.name != "extension"
                  else StructField("extension", StringType(), True)
                  for c in schema.fields
              ]
  
          return schema
  