예제 #1
0
    def filter(self, queryset, values, operator):
        value = values.get('value', None)

        if operator == '=':
            return queryset.filter(values__value=bool(value))
        else:
            raise InvalidOperatorError()
예제 #2
0
    def filter(self, queryset, values, operator):
        value = values.get('value', None)
        if value is None:
            raise InvalidFieldInputError()

        try:
            float_value = float(value)
        except (ValueError, TypeError):
            raise InvalidFieldInputError()

        if operator == '=':
            return queryset.filter(values__value=float_value)

        elif operator == '<':
            return queryset.filter(values__value__lt=float_value)

        elif operator == '<=':
            return queryset.filter(values__value__lte=float_value)

        elif operator == '>':
            return queryset.filter(values__value__gt=float_value)

        elif operator == '>=':
            return queryset.filter(values__value__gte=float_value)

        else:
            raise InvalidOperatorError()
예제 #3
0
    def filter(self, queryset, values, operator):
        date = values.get('value', None)
        if date is None:
            raise InvalidFieldInputError()

        try:
            date_value = parse_date(date).isoformat()
        except (ValueError, TypeError):
            raise InvalidFieldInputError()

        if operator == '=':
            # Exact match when the value in the db starts with the truncated input value.
            return queryset.filter(values__value__startswith=date_value)

        elif operator == '<':
            # This works as before. The string comparison stops at the latest when a lower minute value is found.
            return queryset.filter(values__value__lt=date_value)

        elif operator == '<=':
            # Using Q to filter for a lower value OR an exact match using the truncated input value.
            return queryset.filter(Q(values__value__lt=date_value) | Q(values__value__startswith=date_value))

        elif operator == '>':
            # Excluding the truncated input value to avoid exact (=) matches.
            return queryset.filter(values__value__gt=date_value).exclude(values__value__startswith=date_value)

        elif operator == '>=':
            # Using Q to filter for a greater value OR an exact match using the truncated input value.
            return queryset.filter(Q(values__value__gt=date_value) | Q(values__value__startswith=date_value))

        else:
            raise InvalidOperatorError()
예제 #4
0
    def filter(self, queryset, values, operator):
        if operator != '=':
            raise InvalidOperatorError()

        if not values:
            raise InvalidFieldInputError()

        answers = values.get('answers', None)
        single_selected = values.get('single_selected', None)
        custom_input = values.get('custom_input', None)

        # Needed if a user reselects the default value
        if single_selected == '':
            raise InvalidFieldInputError()

        # Iterate over answers to delete selected: false values.
        # If a user selects and unselects a checkbox the selected: false value is set, which is not in the db
        if not single_selected:
            for answer in answers:
                selected = answer.get('selected', None)
                if selected is False:
                    answer.pop('selected')

        if answers and custom_input:
            return queryset \
                .filter(values__answers=answers) \
                .filter(values__custom_input__icontains=custom_input)
        elif answers and not custom_input:
            return queryset.filter(values__answers=answers)
        elif single_selected and custom_input:
            return queryset \
                .filter(values__single_selected=single_selected) \
                .filter(values__custom_input__icontains=custom_input)
        elif single_selected and not custom_input:
            return queryset.filter(values__single_selected=single_selected)
예제 #5
0
    def filter(self, queryset, values, operator):
        # allow equality operator only
        if operator != '=':
            raise InvalidOperatorError()

        value = values.get('value', '')
        if value == '':
            raise InvalidFieldInputError()

        queryset = queryset \
            .annotate(text_value=KeyTextTransform('value', 'values')) \
            .filter(text_value__icontains=value)

        return queryset
예제 #6
0
    def filter(self, queryset, values, operator):
        if operator != '=':
            raise InvalidOperatorError()

        x = values.get('x', '')
        y = values.get('y', '')

        if x == '' and y == '':
            raise InvalidFieldInputError()

        return queryset \
            .annotate(x=KeyTransform('x', 'values')) \
            .annotate(y=KeyTransform('y', 'values')) \
            .filter(x__iexact=x) \
            .filter(y__iexact=y)
예제 #7
0
    def filter(self, queryset, values, operator):
        numerator = values.get('numerator', None)
        denominator = values.get('denominator', None)
        if numerator is None or denominator is None:
            raise InvalidFieldInputError()

        try:
            numerator = float(numerator)
            denominator = float(denominator)
        except ValueError:
            raise InvalidFieldInputError()

        if denominator == 0:
            raise InvalidFieldInputError()

        # Numerator and denominator must be transformed to decimals, otherwise the SQL division will return an integer.
        # The input fraction must be computed in SQL, to avoid problems caused by different precision levels.
        queryset = queryset \
            .annotate(numerator=KeyDecimalTransform('numerator', 'values')) \
            .annotate(denominator=KeyDecimalTransform('denominator', 'values')) \
            .filter(~Q(denominator=0)) \
            .annotate(fraction=F('numerator') / F('denominator')) \
            .annotate(input_fraction=ExpressionWrapper(Value(numerator) / Value(denominator),
                                                       output_field=FloatField()))

        if operator == '=':
            return queryset.filter(fraction=F('input_fraction'))

        elif operator == '<':
            return queryset.filter(fraction__lt=F('input_fraction'))

        elif operator == '<=':
            return queryset.filter(fraction__lte=F('input_fraction'))

        elif operator == '>':
            return queryset.filter(fraction__gt=F('input_fraction'))

        elif operator == '>=':
            return queryset.filter(fraction__gte=F('input_fraction'))

        else:
            raise InvalidOperatorError()