def compile()

in data_validation/query_builder/query_builder.py [0:0]

• 40 lines of code
• 12 McCabe index (conditional complexity)

    def compile(self, validation_type, table):
        """Return an Ibis query object

        Args:
            table (IbisTable): The Ibis Table expression.
        """

        # Build Query Expressions
        compiled_filters = self.compile_filter_fields(table)
        filtered_table = table.filter(compiled_filters) if compiled_filters else table

        if self.calculated_fields:
            depth_limit = max(
                field.config.get(consts.CONFIG_DEPTH, 0)
                for field in self.calculated_fields
            )
            for n in range(0, (depth_limit + 1)):
                filtered_table = filtered_table.mutate(
                    self.compile_calculated_fields(filtered_table, n)
                )

        if (
            validation_type == consts.ROW_VALIDATION
            or validation_type == consts.CUSTOM_QUERY
        ):
            if self.comparison_fields:
                filtered_table = filtered_table.projection(
                    self.compile_comparison_fields(filtered_table)
                )
        else:
            if self.comparison_fields:
                filtered_table = filtered_table.mutate(
                    self.compile_comparison_fields(filtered_table)
                )
        compiled_groups = self.compile_group_fields(filtered_table)
        grouped_table = (
            filtered_table.group_by(compiled_groups)
            if compiled_groups
            else filtered_table
        )
        if self.aggregate_fields:
            query = grouped_table.aggregate(
                self.compile_aggregate_fields(filtered_table)
            )
        else:
            query = grouped_table

        if self.limit:
            query = query.limit(self.limit)

        return query