#coding=utf-8
import django_filters
from apps.base import clean_datetime_range
from .models import *


class BranchFilter(django_filters.FilterSet):
    name = django_filters.CharFilter(field_name='name',lookup_expr='icontains')
    tel = django_filters.CharFilter(field_name='tel', lookup_expr='icontains')
    enabled = django_filters.CharFilter(field_name='enabled')

    class Meta:
        model = Branch
        fields = (
            'name', 'tel', 'enabled'
        )

    def __init__(self, data=None, *args, **kwargs):
        data = clean_datetime_range(data, 'create_time')
        super(BranchFilter, self).__init__(data, *args, **kwargs)


class MemberFilter(django_filters.FilterSet):
    nickname = django_filters.CharFilter(field_name='nickname', lookup_expr='icontains')
    tel = django_filters.CharFilter(field_name='tel', lookup_expr='icontains')

    class Meta:
        model = Member
        fields = '__all__'


class MemberCouponFilter(django_filters.FilterSet):
    write_off = django_filters.CharFilter(field_name='write_off')
    coupon_name = django_filters.CharFilter(field_name='coupon__name', lookup_expr='icontains')
    activity_name = django_filters.CharFilter(field_name='activity__title', lookup_expr='icontains')

    class Meta:
        model = MemberCoupon
        fields = '__all__'


class OrderFilter(django_filters.FilterSet):
    activity_name = django_filters.CharFilter(field_name='activity__title', lookup_expr='icontains')
    branch_name = django_filters.CharFilter(field_name='branch__name', lookup_expr='icontains')
    activity = django_filters.CharFilter(field_name='activity')
    delete = django_filters.CharFilter(field_name='delete')

    class Meta:
        model = Order
        fields = '__all__'



class CouponFilter(django_filters.FilterSet):
    name = django_filters.CharFilter(field_name='name',lookup_expr='icontains')
    activity = django_filters.CharFilter(field_name='activity')
    enabled = django_filters.CharFilter(field_name='enabled')

    class Meta:
        model = Coupon
        fields = (
            'name', 'enabled'
        )

    def __init__(self, data=None, *args, **kwargs):
        data = clean_datetime_range(data, 'create_time')
        super(CouponFilter, self).__init__(data, *args, **kwargs)


class ActivityFilter(django_filters.FilterSet):
    title = django_filters.CharFilter(field_name='title',lookup_expr='icontains')
    branch_name = django_filters.CharFilter(field_name='branch_name', lookup_expr='icontains')
    enabled = django_filters.CharFilter(field_name='enabled')
    check_status = django_filters.CharFilter(field_name='check_status')

    class Meta:
        model = Activity
        fields = (
            'title', 'branch_name', 'check_status', 'enabled'
        )

    def __init__(self, data=None, *args, **kwargs):
        data = clean_datetime_range(data, 'create_time')
        super(ActivityFilter, self).__init__(data, *args, **kwargs)