Skip to content

Commit

Permalink
#41: Add implementation to filter
Browse files Browse the repository at this point in the history
  • Loading branch information
keiranjprice101 committed Sep 11, 2019
1 parent cc2459a commit 2a496a0
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion common/database_helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,8 +152,16 @@ def apply_filter(self, query):
class DistinctFieldFilter(QueryFilter):
precedence = 0

def __init__(self, fields):
self.fields = fields if type(fields) is list else [fields] # This allows single string distinct filters

def apply_filter(self, query):
pass
query.is_distinct_fields_query = True
try:
self.fields = [getattr(query.table, field) for field in self.fields]
except AttributeError:
raise BadFilterError("Bad field requested")
query.base_query = query.session.query(*self.fields).distinct()


class OrderFilter(QueryFilter):
Expand Down

0 comments on commit 2a496a0

Please sign in to comment.