zcdsj
/
hberp
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
hb_server/apps/hrm/filters.py
View File

@ -5,12 +5,18 @@ from utils.mixins import DynamicFieldsFilterMixin
class ClockRecordFilterSet(filters.FilterSet): class ClockRecordFilterSet(filters.FilterSet):
create_time_start = filters.DateFilter(field_name="create_time", lookup_expr='gte') create_time_start = filters.DateFilter(field_name="create_time", lookup_expr='gte')
create_time_end = filters.DateFilter(field_name="create_time", lookup_expr='lte') create_time_end = filters.DateFilter(field_name="create_time", lookup_expr='lte')
year = filters.DateFilter(field_name="create_time_date__year") year = filters.NumberFilter(method='filter_year')
month = filters.DateFilter(field_name="create_time_date__month") month = filters.NumberFilter(method='filter_month')
class Meta: class Meta:
model = ClockRecord model = ClockRecord
fields = ['create_by', 'create_time_start', 'create_time_end', 'year', 'month'] 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 EmployeeFilterSet(DynamicFieldsFilterMixin, filters.FilterSet):
class Meta: class Meta:
@ -19,8 +25,14 @@ class EmployeeFilterSet(DynamicFieldsFilterMixin, filters.FilterSet):
class NotWorkRemarkFilterSet(filters.FilterSet): class NotWorkRemarkFilterSet(filters.FilterSet):
year = filters.DateFilter(field_name="not_work_date__year") year = filters.NumberFilter(method='filter_year')
month = filters.DateFilter(field_name="not_work_date__month") month = filters.NumberFilter(method='filter_month')
class Meta: class Meta:
model = NotWorkRemark 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)