Django过滤数据遇到的坑

首先创建form表单

<div>
   <form action="" method="get">
        <div class="row">
              循环list_filters中需要过滤的字段
             {% for column in admin_class.list_filters %}
                 <div class="col-lg-2">                
                      <span class="col-lg-1">{{ column }}</span>

                      #<select>与<option>在标签中完成，filter_conditions用来检索完成后option显示对应的值
                      {% render_filter_column column admin_class filter_conditions %}
                 </div>
             {% endfor %}
             <button type="submit" class="btn btn-success">检索</button>
          </div>
     </form>
</div>

在视图中调用该方法

def table_filter(request,admin_class):
    '''进行条件过滤并返回过滤后的数据'''
    filter_conditions = {}
    # 遍历 url 参数的字典
    for k,v in request.GET.items():
        # print(k,v)
        # 把需要过滤的选项放入字典，未选择过滤的字段 v = None
        if v:
            filter_conditions[k] = v
    # 视图函数中接受收的 model_objs 是过滤后的列表,把filter_conditions传入前端，再从前端传入简单标签
    return admin_class.model.objects.filter(**filter_conditions), filter_conditions

简单标签

外键与非外键的choices不同

• 非外键 ：field_obj.choices
• 外键 ：field_obj.get_choices()

使用filter_conditions[column] 会报 KeyError: 'source' ，必须filter_conditions.get(column),因为没点检索也会传flter_conditions参数到标签中，这时flter_conditions为空，使用get()不会报错

• selected 是过滤后保留字段在option中的属性

'''过滤数据'''
@register.simple_tag
def render_filter_column(column,admin_class,filter_conditions):

    # <option value=''>---</option>是防止option默认显示字段的第一个choice
    select_ele = "<select class='form-control' name=%s><option value=''>---</option>"%column
    # 得到字段对象  
    field_obj = admin_class.model._meta.get_field(column)
    
    # 外键字段不能直接 choices
    if field_obj.choices:

        # print(field_obj.choices)   #((0, '转介绍'), (1, 'QQ群'), (2, '官网'), (3, '百度推广')....)
        for column_content in field_obj.choices:
            selected = ''
            # 注意！ column_content[0] 是int
            if filter_conditions.get(column) == str(column_content[0]):
                selected = 'selected'
            select_ele += "<option value=%s %s>%s</option>"%(column_content[0],selected,column_content[1])

    if type(field_obj).__name__ == 'ForeignKey':
        # print(field_obj.get_choices())   #[('', '---------'), (1, '销售1号'), (2, '销售2号'), (3, '销售3号')]
        for column_content in field_obj.get_choices()[1:]:    # 对'-----'切片
            selected = ''
            if filter_conditions.get(column) == str(column_content[0]):
                selected = 'selected'
            select_ele += "<option value=%s %s>%s</option>"%(column_content[0],selected,column_content[1])
    return mark_safe(select_ele+"</select>")

未完待续