mirror of https://github.com/snachodog/mybuddy.git
Ensure API `OPTIONS` captures class-based filters
This commit is contained in:
parent
54a24423c0
commit
507e0c383d
|
@ -3,64 +3,90 @@ from django_filters import rest_framework as filters
|
||||||
from core import models
|
from core import models
|
||||||
|
|
||||||
|
|
||||||
class TimeFieldFilter(filters.FilterSet):
|
class ChildFieldFilter(filters.FilterSet):
|
||||||
|
class Meta:
|
||||||
|
abstract = True
|
||||||
|
fields = ['child']
|
||||||
|
|
||||||
|
|
||||||
|
class TimeFieldFilter(ChildFieldFilter):
|
||||||
date = filters.DateFilter(field_name='time__date', label='Date')
|
date = filters.DateFilter(field_name='time__date', label='Date')
|
||||||
date_min = filters.DateFilter(field_name='time__date', label='Min. Date',
|
|
||||||
lookup_expr='gte')
|
|
||||||
date_max = filters.DateFilter(field_name='time__date', label='Max. Date',
|
date_max = filters.DateFilter(field_name='time__date', label='Max. Date',
|
||||||
lookup_expr='lte')
|
lookup_expr='lte')
|
||||||
|
date_min = filters.DateFilter(field_name='time__date', label='Min. Date',
|
||||||
|
lookup_expr='gte')
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
abstract = True
|
||||||
|
fields = sorted(ChildFieldFilter.Meta.fields + [
|
||||||
|
'date',
|
||||||
|
'date_max',
|
||||||
|
'date_min'
|
||||||
|
])
|
||||||
|
|
||||||
|
|
||||||
class StartEndFieldFilter(filters.FilterSet):
|
class StartEndFieldFilter(ChildFieldFilter):
|
||||||
end = filters.DateFilter(field_name='end__date', label='End Date')
|
end = filters.DateFilter(field_name='end__date', label='End Date')
|
||||||
end_min = filters.DateFilter(field_name='end__date', label='Min. End Date',
|
|
||||||
lookup_expr='gte')
|
|
||||||
end_max = filters.DateFilter(field_name='end__date', label='Max. End Date',
|
end_max = filters.DateFilter(field_name='end__date', label='Max. End Date',
|
||||||
lookup_expr='lte')
|
lookup_expr='lte')
|
||||||
|
end_min = filters.DateFilter(field_name='end__date', label='Min. End Date',
|
||||||
|
lookup_expr='gte')
|
||||||
start = filters.DateFilter(field_name='start__date', label='Start Date')
|
start = filters.DateFilter(field_name='start__date', label='Start Date')
|
||||||
start_min = filters.DateFilter(field_name='start__date', lookup_expr='gte',
|
start_max = filters.DateFilter(field_name='start__date', lookup_expr='lte',
|
||||||
label='Min. Start Date',)
|
|
||||||
start_end = filters.DateFilter(field_name='start__date', lookup_expr='lte',
|
|
||||||
label='Max. End Date')
|
label='Max. End Date')
|
||||||
|
start_min = filters.DateFilter(field_name='start__date', lookup_expr='gte',
|
||||||
|
label='Min. Start Date')
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
abstract = True
|
||||||
|
fields = sorted(ChildFieldFilter.Meta.fields + [
|
||||||
|
'end',
|
||||||
|
'end_max',
|
||||||
|
'end_min',
|
||||||
|
'start',
|
||||||
|
'start_max',
|
||||||
|
'start_min'
|
||||||
|
])
|
||||||
|
|
||||||
|
|
||||||
class DiaperChangeFilter(TimeFieldFilter):
|
class DiaperChangeFilter(TimeFieldFilter):
|
||||||
class Meta:
|
class Meta(TimeFieldFilter.Meta):
|
||||||
model = models.DiaperChange
|
model = models.DiaperChange
|
||||||
fields = ['child', 'wet', 'solid', 'color', 'amount']
|
fields = sorted(TimeFieldFilter.Meta.fields + [
|
||||||
|
'wet',
|
||||||
|
'solid',
|
||||||
|
'color',
|
||||||
|
'amount'
|
||||||
|
])
|
||||||
|
|
||||||
|
|
||||||
class FeedingFilter(StartEndFieldFilter):
|
class FeedingFilter(StartEndFieldFilter):
|
||||||
class Meta:
|
class Meta(StartEndFieldFilter.Meta):
|
||||||
model = models.Feeding
|
model = models.Feeding
|
||||||
fields = ['child', 'type', 'method']
|
fields = sorted(StartEndFieldFilter.Meta.fields + ['type', 'method'])
|
||||||
|
|
||||||
|
|
||||||
class NoteFilter(TimeFieldFilter):
|
class NoteFilter(TimeFieldFilter):
|
||||||
class Meta:
|
class Meta(TimeFieldFilter.Meta):
|
||||||
model = models.Note
|
model = models.Note
|
||||||
fields = ['child']
|
|
||||||
|
|
||||||
|
|
||||||
class SleepFilter(StartEndFieldFilter):
|
class SleepFilter(StartEndFieldFilter):
|
||||||
class Meta:
|
class Meta(StartEndFieldFilter.Meta):
|
||||||
model = models.Sleep
|
model = models.Sleep
|
||||||
fields = ['child']
|
|
||||||
|
|
||||||
|
|
||||||
class TemperatureFilter(TimeFieldFilter):
|
class TemperatureFilter(TimeFieldFilter):
|
||||||
class Meta:
|
class Meta(TimeFieldFilter.Meta):
|
||||||
model = models.Temperature
|
model = models.Temperature
|
||||||
fields = ['child']
|
|
||||||
|
|
||||||
|
|
||||||
class TimerFilter(StartEndFieldFilter):
|
class TimerFilter(StartEndFieldFilter):
|
||||||
class Meta:
|
class Meta(StartEndFieldFilter.Meta):
|
||||||
model = models.Timer
|
model = models.Timer
|
||||||
fields = ['child', 'active', 'user']
|
fields = sorted(StartEndFieldFilter.Meta.fields + ['active', 'user'])
|
||||||
|
|
||||||
|
|
||||||
class TummyTimeFilter(StartEndFieldFilter):
|
class TummyTimeFilter(StartEndFieldFilter):
|
||||||
class Meta:
|
class Meta(StartEndFieldFilter.Meta):
|
||||||
model = models.TummyTime
|
model = models.TummyTime
|
||||||
fields = ['child']
|
|
||||||
|
|
|
@ -11,4 +11,6 @@ class APIMetadata(metadata.SimpleMetadata):
|
||||||
data.pop('description')
|
data.pop('description')
|
||||||
if hasattr(view, 'filterset_fields'):
|
if hasattr(view, 'filterset_fields'):
|
||||||
data.update({'filters': view.filterset_fields})
|
data.update({'filters': view.filterset_fields})
|
||||||
|
elif hasattr(view, 'filterset_class'):
|
||||||
|
data.update({'filters': view.filterset_class.Meta.fields})
|
||||||
return data
|
return data
|
||||||
|
|
Loading…
Reference in New Issue