mirror of
https://github.com/himool/HimoolERP.git
synced 2026-01-17 14:48:41 +08:00
24 lines
825 B
Python
24 lines
825 B
Python
from django_filters import rest_framework as filters
|
|
from .models import PaymentRecord, PurchaseOrder
|
|
import pendulum
|
|
|
|
|
|
class PurchaseOrderFilter(filters.FilterSet):
|
|
start_date = filters.CharFilter(field_name='date', lookup_expr='gte')
|
|
end_date = filters.CharFilter(method='end_date_filter', label='end_date')
|
|
|
|
class Meta:
|
|
model = PurchaseOrder
|
|
fields = ['start_date', 'end_date', 'warehouse', 'supplier', 'is_done', 'is_return']
|
|
|
|
def end_date_filter(self, queryset, name, value):
|
|
date = pendulum.parse(value).add(days=1)
|
|
return queryset.filter(date__lte=date)
|
|
|
|
|
|
class PurchasePaymentRecordFilter(filters.FilterSet):
|
|
is_return = filters.BooleanFilter(field_name='purchase_order__is_return')
|
|
|
|
class Meta:
|
|
model = PaymentRecord
|
|
fields = ['is_return']
|