not work remark filter

This commit is contained in:
caoqianming 2022-02-22 11:28:22 +08:00
parent 074a53f45f
commit a0d87b6c8b
1 changed files with 17 additions and 5 deletions

View File

@ -5,12 +5,18 @@ from utils.mixins import DynamicFieldsFilterMixin
class ClockRecordFilterSet(filters.FilterSet):
create_time_start = filters.DateFilter(field_name="create_time", lookup_expr='gte')
create_time_end = filters.DateFilter(field_name="create_time", lookup_expr='lte')
year = filters.DateFilter(field_name="create_time_date__year")
month = filters.DateFilter(field_name="create_time_date__month")
year = filters.NumberFilter(method='filter_year')
month = filters.NumberFilter(method='filter_month')
class Meta:
model = ClockRecord
fields = ['create_by', 'create_time_start', 'create_time_end', 'year', 'month']
def filter_year(self, queryset, name, value):
return queryset.filter(create_time_date__year=value)
def filter_month(self, queryset, name, value):
return queryset.filter(create_time_date__month=value)
class EmployeeFilterSet(DynamicFieldsFilterMixin, filters.FilterSet):
class Meta:
@ -19,8 +25,14 @@ class EmployeeFilterSet(DynamicFieldsFilterMixin, filters.FilterSet):
class NotWorkRemarkFilterSet(filters.FilterSet):
year = filters.DateFilter(field_name="not_work_date__year")
month = filters.DateFilter(field_name="not_work_date__month")
year = filters.NumberFilter(method='filter_year')
month = filters.NumberFilter(method='filter_month')
class Meta:
model = NotWorkRemark
fields = ['year', 'month', 'user']
fields = ['year', 'month', 'user']
def filter_year(self, queryset, name, value):
return queryset.filter(not_work_date__year=value)
def filter_month(self, queryset, name, value):
return queryset.filter(not_work_date__month=value)