def generate_data_class_for_shape()

in src/sagemaker_core/tools/shapes_codegen.py [0:0]

• 16 lines of code
• 2 McCabe index (conditional complexity)

    def generate_data_class_for_shape(self, shape):
        """
        Generates a data class for a given shape.

        :param shape: The name of the shape.
        :return: The generated data class as a string.
        """
        class_name = shape
        init_data = self.shapes_extractor.generate_data_shape_string_body(
            shape, self.resources_plan, add_shapes_prefix=False
        )
        try:
            data_class_members = add_indent(init_data, 4)
        except Exception:
            print("DEBUG HELP\n", init_data)
            raise
        return SHAPE_CLASS_TEMPLATE.format(
            class_name=class_name + "(Base)",
            data_class_members=data_class_members,
            docstring=self._generate_doc_string_for_shape(shape),
            class_name_snake=pascal_to_snake(class_name),
        )