def check_none_types()

in cli/src/graphar_cli/config.py [0:0]

• 41 lines of code
• 18 McCabe index (conditional complexity)

    def check_none_types(self) -> Self:
        for vertex in self.import_schema.vertices:
            if vertex.chunk_size is None:
                vertex.chunk_size = self.graphar.vertex_chunk_size
            if vertex.validate_level is None:
                vertex.validate_level = self.graphar.validate_level
            for property_group in vertex.property_groups:
                if property_group.file_type is None:
                    property_group.file_type = self.graphar.file_type
        for edge in self.import_schema.edges:
            if edge.chunk_size is None:
                edge.chunk_size = self.graphar.edge_chunk_size
            if edge.validate_level is None:
                edge.validate_level = self.graphar.validate_level
            if len(edge.adj_lists) == 0:
                if self.graphar.adj_list_type == "ordered_by_source":
                    edge.adj_lists.append(
                        AdjList(ordered=True, aligned_by="src", file_type=self.graphar.file_type)
                    )
                elif self.graphar.adj_list_type == "ordered_by_dest":
                    edge.adj_lists.append(
                        AdjList(ordered=True, aligned_by="dst", file_type=self.graphar.file_type)
                    )
                elif self.graphar.adj_list_type == "unordered_by_source":
                    edge.adj_lists.append(
                        AdjList(ordered=False, aligned_by="src", file_type=self.graphar.file_type)
                    )
                elif self.graphar.adj_list_type == "unordered_by_dest":
                    edge.adj_lists.append(
                        AdjList(ordered=False, aligned_by="dst", file_type=self.graphar.file_type)
                    )
                else:
                    msg = f"Invalid adj_list_type '{self.graphar.adj_list_type}'"
                    raise ValueError(msg)
            for adj_list in edge.adj_lists:
                if adj_list.file_type is None:
                    adj_list.file_type = self.graphar.file_type
            for property_group in edge.property_groups:
                if property_group.file_type is None:
                    property_group.file_type = self.graphar.file_type

        return self