mirror of https://github.com/snachodog/mybuddy.git
109 lines
3.0 KiB
Python
109 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
from django_filters import rest_framework as filters
|
|
|
|
from core import models
|
|
|
|
|
|
class CharInFilter(filters.BaseInFilter, filters.CharFilter):
|
|
pass
|
|
|
|
|
|
class ChildFieldFilter(filters.FilterSet):
|
|
class Meta:
|
|
abstract = True
|
|
fields = ["child"]
|
|
|
|
|
|
class TagsFieldFilter(filters.FilterSet):
|
|
tags = CharInFilter(
|
|
field_name="tags__name",
|
|
label="tag",
|
|
help_text="A list of tag names, comma separated",
|
|
)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class TimeFieldFilter(ChildFieldFilter):
|
|
date = filters.DateFilter(field_name="time__date", label="Date")
|
|
date_max = filters.DateFilter(
|
|
field_name="time__date", label="Max. Date", 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(ChildFieldFilter):
|
|
end = filters.DateFilter(field_name="end__date", label="End Date")
|
|
end_max = filters.DateFilter(
|
|
field_name="end__date", label="Max. End Date", 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_max = filters.DateFilter(
|
|
field_name="start__date", lookup_expr="lte", 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 BreastpumpFilter(TimeFieldFilter):
|
|
class Meta(TimeFieldFilter.Meta):
|
|
model = models.Breastpump
|
|
|
|
|
|
class DiaperChangeFilter(TimeFieldFilter, TagsFieldFilter):
|
|
class Meta(TimeFieldFilter.Meta):
|
|
model = models.DiaperChange
|
|
fields = sorted(
|
|
TimeFieldFilter.Meta.fields + ["wet", "solid", "color", "amount"]
|
|
)
|
|
|
|
|
|
class FeedingFilter(StartEndFieldFilter, TagsFieldFilter):
|
|
class Meta(StartEndFieldFilter.Meta):
|
|
model = models.Feeding
|
|
fields = sorted(StartEndFieldFilter.Meta.fields + ["type", "method"])
|
|
|
|
|
|
class NoteFilter(TimeFieldFilter, TagsFieldFilter):
|
|
class Meta(TimeFieldFilter.Meta):
|
|
model = models.Note
|
|
|
|
|
|
class SleepFilter(StartEndFieldFilter, TagsFieldFilter):
|
|
class Meta(StartEndFieldFilter.Meta):
|
|
model = models.Sleep
|
|
|
|
|
|
class TemperatureFilter(TimeFieldFilter, TagsFieldFilter):
|
|
class Meta(TimeFieldFilter.Meta):
|
|
model = models.Temperature
|
|
|
|
|
|
class TimerFilter(StartEndFieldFilter):
|
|
class Meta(StartEndFieldFilter.Meta):
|
|
model = models.Timer
|
|
fields = sorted(StartEndFieldFilter.Meta.fields + ["active", "user"])
|
|
|
|
|
|
class TummyTimeFilter(StartEndFieldFilter, TagsFieldFilter):
|
|
class Meta(StartEndFieldFilter.Meta):
|
|
model = models.TummyTime
|