DRF 分页、排序、过滤
DRF 分页、排序、过滤
1. 自定义频率类
class NewThrottle(BaseThrottle):
# 存放用户访问记录
user_dict = {} # 格式:{用户ip:[访问时间1,访问时间2]}
# 获取当前时间
now_time = time.time
def __init__(self):
self.user_list = None
def allow_request(self, request, view):
# 取出当前访问者的ip地址
user_ip = request.META.get('REMOTE_ADDR')
# 判断当前访问者的ip地址有没有在存放用户访问记录的字典内?
if user_ip not in self.user_dict: # 取反操作,条件成立表示当前用户没有访问过。
# 将用户的ip地址及当前访问时间存入用户访问记录内
self.user_dict[user_ip] = [self.now_time(), ]
# 返回 True 表示用户可以访问
return True
else: # 条件不成立表示用户访问过
# 将用户的访问记录时间取出赋值给self.user_list
self.user_list = self.user_dict.get(user_ip)
"""
使用while循环判断当前ip的时间列表。
表内有值并且当前时间减去表内最后一次访问时间大于限制时间60秒,说明条件成立
pop默认删除最后一次已经超出限制时间的表内的访问时间
这样表内只有限制时间以内的访问时间
"""
while self.user_list and self.now_time() - self.user_list[-1] > 60:
self.user_list.pop()
# 循环结束表内只有60秒以内访问的时间
# 以表内访问时间数据作为限制条件,判断是否一分钟内访问超过限制的次数
if len(self.user_list) < 5: # 条件成立说明当前访问者访问次数暂没有超过限制,
# 将当前时间指定添加到表内
self.user_dict.get(user_ip).insert(0,self.now_time())
# 返回 True 表示用户可以访问
return True
# 条件不成立表示当前访问者访问次数已超出,返回False 不能访问
return False
def wait(self):
# 返回限制访问的剩余时间
return 60 - (self.now_time() - self.user_list[-1])
1.1 频率类源码分析
# SimpleRateThrottle
-源码里执行的频率类的allow_request,读SimpleRateThrottle的allow_request
class SimpleRateThrottle(BaseThrottle):
cache = default_cache
timer = time.time
cache_format = 'throttle_%(scope)s_%(ident)s'
scope = None
THROTTLE_RATES = api_settings.DEFAULT_THROTTLE_RATES
def __init__(self): # 只要类实例化得到对象就会执行,一执行,self.rate就有值了,而且self.num_requests和self.duration
if not getattr(self, 'rate', None): # 去频率类中反射rate属性或方法,发现没有,返回了None,这个if判断就符合,执行下面的代码
self.rate = self.get_rate() #返回了 '3/m'
# self.num_requests=3
# self.duration=60
self.num_requests, self.duration = self.parse_rate(self.rate)
def get_rate(self):
return self.THROTTLE_RATES[self.scope] # 字典取值,配置文件中咱们配置的字典{'ss': '3/m',},根据ss取到了 '3/m'
def parse_rate(self, rate):
if rate is None:
return (None, None)
# rate:字符串'3/m' 根据 / 切分,切成了 ['3','m']
# num=3,period=m
num, period = rate.split('/')
# num_requests=3 数字3
num_requests = int(num)
# period='m' ---->period[0]--->'m'
# {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]]
# duration=60
duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]]
# 3 60
return (num_requests, duration)
def allow_request(self, request, view):
if self.rate is None:
return True
# 咱们自己写的,返回什么就以什么做限制 咱们返回的是ip地址
# self.key=当前访问者的ip地址
self.key = self.get_cache_key(request, view)
if self.key is None:
return True
# self.history 访问者的时间列表,从缓存中拿到,如果拿不到就是空列表,如果之前有 [时间2,时间1]
self.history = self.cache.get(self.key, [])
# 当前时间
self.now = self.timer()
while self.history and self.history[-1] <= self.now - self.duration:
self.history.pop()
if len(self.history) >= self.num_requests:
return self.throttle_failure()
return self.throttle_success()
# 总结:以后要再写频率类,只需要继承SimpleRateThrottle,重写get_cache_key,配置类属性scope,配置文件中配置一下就可以了
2. 分页功能
-
使用场景,查询所有的接口需要使用到分页
-
分页后端写法是固定的,前端展现形式是不一样的
-
drf中的分页有三种方式
- 继承PageNumberPagination
- 继承LimitOffsetPagination
- 继承CursorPagination
-
drf中分页的使用
-- 写一个类,继承drf中提供的三个分页类之一
-- 把它配置在继承自GenericAPIView+ListModelMixin的子视图类上
- 继承PageNumberPagination分页类

views.py

- 继承LimitOffsetPagination分页类

views.py中

- 继承CursorPagination分页类

views.py

-
如果继承的是APIView,需要自己写
pass
3.排序功能
-
查询所有才涉及到排序
-
必须是继承GenericAPIView+ListModelMixin的子视图类上
-
排序类的配置
views.py中

-
继承VPIView的,需要自己从请求地址中取出排序规则,自己排序
分页和排序能一起用,但是是先排序后分页的

3. 过滤功能
-
查询所有才涉及到过滤
-
restful规范中有一条,请求地址中带过滤条件:分页,排序,过滤统称为过滤
-
使用内置过滤类使用步骤
必须是继承GenericAPIView+ListModelMixin的子视图类上


浙公网安备 33010602011771号