def visit_aggregate_filter(self, node, children):
(negation, search_key, _, operator, search_value) = children
operator = self.handle_negation(negation, operator)
search_value = search_value[0] if not isinstance(search_value, RegexNode) else search_value
try:
aggregate_value = None
if search_value.expr_name in ["duration_format", "percentage_format"]:
# Even if the search value matches duration format, only act as duration for certain columns
function = resolve_field(
search_key.name, self.params, functions_acl=FUNCTIONS.keys()
)
if function.aggregate is not None:
if search_value.expr_name == "percentage_format" and self.is_percentage_key(
function.aggregate[0]
):
aggregate_value = parse_percentage(*search_value.match.groups())
# Extract column and function name out so we can check if we should parse as duration
elif search_value.expr_name == "duration_format" and self.is_duration_key(
function.aggregate[1]
):
aggregate_value = parse_duration(*search_value.match.groups())
if aggregate_value is None:
aggregate_value = parse_numeric_value(*search_value.match.groups())
except ValueError:
raise InvalidSearchQuery(f"Invalid aggregate query condition: {search_key}")
except InvalidQuery as exc:
raise InvalidSearchQuery(str(exc))
return AggregateFilter(search_key, operator, SearchValue(aggregate_value))
def visit_aggregate_percentage_filter(self, node, children):
(negation, search_key, _, operator, search_value) = children
operator = handle_negation(negation, operator)
aggregate_value = None
try:
# Even if the search value matches percentage format, only act as
# percentage for certain columns
function = resolve_field(search_key.name,
self.params,
functions_acl=FUNCTIONS.keys())
if function.aggregate is not None and self.is_percentage_key(
function.aggregate[0]):
aggregate_value = parse_percentage(search_value)
except ValueError:
raise InvalidSearchQuery(
f"Invalid aggregate query condition: {search_key}")
except InvalidQuery as exc:
raise InvalidSearchQuery(str(exc))
if aggregate_value is not None:
return AggregateFilter(search_key, operator,
SearchValue(aggregate_value))
# Invalid formats fall back to text match
search_value = operator + search_value if operator != "=" else search_value
return AggregateFilter(search_key, "=", SearchValue(search_value))