|
|
import datetime
|
|
|
|
|
|
from rest_framework import serializers
|
|
|
from django_filters.rest_framework import FilterSet
|
|
|
import django_filters
|
|
|
from app.models import TP
|
|
|
import logging
|
|
|
logger = logging.getLogger('mylogger')
|
|
|
class SerialMyModel(serializers.ModelSerializer):
|
|
|
class Meta:
|
|
|
model = TP
|
|
|
fields = "__all__"
|
|
|
|
|
|
|
|
|
class SerialFilter(FilterSet):
|
|
|
"""
|
|
|
过滤器，支持模糊查询
|
|
|
record_time，日期时间格式，如 2023-01-01 00:00:00
|
|
|
police_id，支持模糊匹配
|
|
|
event_type，支持模糊匹配
|
|
|
"""
|
|
|
record_time = django_filters.DateTimeFilter(field_name='record_time', lookup_expr='icontains')
|
|
|
police_id = django_filters.CharFilter(field_name='police_id', lookup_expr='icontains')
|
|
|
event_type = django_filters.CharFilter(field_name='event_type', lookup_expr='icontains')
|
|
|
# 记录时间范围查询
|
|
|
start_time = django_filters.DateTimeFilter(field_name='record_time', lookup_expr='gte')
|
|
|
# end_time = django_filters.DateTimeFilter(field_name='record_time', lookup_expr='lte')
|
|
|
end_time = django_filters.DateTimeFilter(field_name='record_time', method='time_range_filter')
|
|
|
# todo bug: start_time 和 end_time同一天的情况 done
|
|
|
violation = django_filters.NumberFilter(field_name='is_violation', method='is_violation_query')
|
|
|
violation_type = django_filters.CharFilter(field_name='ai_analysis', lookup_expr='icontains')
|
|
|
|
|
|
def time_range_filter(self, queryset, name, value):
|
|
|
"""
|
|
|
@params queryset: 为TP.objects.all()返回的合集, 即视图类中的的queryset
|
|
|
@params name: 为field_name
|
|
|
@params value: 为前端通过end_time字段传过来的值
|
|
|
"""
|
|
|
|
|
|
return queryset.filter(record_time__lte=value + datetime.timedelta(days=1))
|
|
|
|
|
|
def is_violation_query(self, queryset, name, value):
|
|
|
if value == 2:
|
|
|
return queryset
|
|
|
elif value == 1:
|
|
|
return queryset.filter(is_violation=True)
|
|
|
elif value == 0:
|
|
|
return queryset.filter(is_violation=False)
|
|
|
else:
|
|
|
return queryset.filter(is_violation=True)
|
|
|
|
|
|
class Meta:
|
|
|
# 指定模型
|
|
|
models = TP
|
|
|
# 指定需要模糊查询的字段
|
|
|
fields = ("record_time", "police_id", "event_type", "ai_analysis",)
|
|
|
|
|
|
|