View.py
from ast import literal_eval import snowflake.client # 雪花算法存id from rest_framework.response import Response from rest_framework.views import APIView from opportunity_market.checkresult import CheckResult from opportunity_market.models import BusOpportunityFollow, BusOpportunitySetting from opportunity_market.my_pagination import MyPagination from opportunity_market.serializers import OpportunitySearchSerializer, BusOpportunityFollowSerializer, \ BusOpportunitySettingSerializer from read_db_app.models import GboOdsBusOpportunity class OpportunitySearch(APIView): def get(self, request): # 只接收get请求 keyword = request.GET.get('keyword') country = request.GET.get('country_code') post_time_type = request.GET.get('post_time_type') quantity_type = request.GET.get('quantity_type') page = request.GET.get('page', 1) user_id = request.GET.get('user_id') if keyword.strip() == '': return Response(CheckResult.check('关键词为空, 请求失败')) # 发布时间筛选 if post_time_type == '1': post_datetime = (datetime.datetime.now() - datetime.timedelta(hours=12)) post_time = time.mktime(post_datetime.timetuple()) elif post_time_type == '2': post_datetime = (datetime.datetime.now() - datetime.timedelta(days=1)) post_time = time.mktime(post_datetime.timetuple()) elif post_time_type == '3': post_datetime = (datetime.datetime.now() - datetime.timedelta(days=3)) post_time = time.mktime(post_datetime.timetuple()) elif post_time_type == '4': post_datetime = (datetime.datetime.now() - datetime.timedelta(weeks=1)) post_time = time.mktime(post_datetime.timetuple()) else: post_time = 0 if quantity_type == '1': quantity_min = 1 quantity_max = 100 elif quantity_type == '2': quantity_min = 100 quantity_max = 1000 elif quantity_type == '3': quantity_min = 1000 quantity_max = float("inf") # 无穷大 if country == '': opp_data = GboOdsBusOpportunity.objects.filter(bo_subject__icontains=keyword, post_time__gte=post_time, quantity__gte=quantity_min, quantity__lte=quantity_max) # __icontains: 忽略大小写模糊查询‘%s%’, __gte:大于等于， __lte：小于等于 else: opp_data = GboOdsBusOpportunity.objects.filter(country_en_name__icontains=country, bo_subject__icontains=keyword, post_time__gte=post_time, quantity__gte=quantity_min, quantity__lte=quantity_max) mypage = MyPagination() page_roles = mypage.paginate_queryset(queryset=opp_data, request=request, view=self) # 设置返回列表每页数据条数 serializer_data = OpportunitySearchSerializer(instance=page_roles, many=True).data # 序列化每页的取到的数据 bo_id_list = [li.get('bo_id') for li in serializer_data] res = BusOpportunityFollow.objects.filter(bo_id__in=bo_id_list, user_id=user_id).only('bo_id') # .only():经取回‘bo_id’字段的值 bus_ser = BusOpportunityFollowSerializer(res, many=True).data for data in serializer_data: if data.get('bo_id') in [bus_bo_id.get('bo_id') for bus_bo_id in bus_ser]: # 给序列化后，返回前端前的列表中的每条数据添加一个key:value值 data.update({'follow_type': '1'}) else: data.update({'follow_type': '2'}) result = { 'errcode': 0, 'errmsg': '请求成功', 'total': opp_data.count(), 'page': page, 'data': serializer_data } return Response(result) # Rsponse 返回到前端 class buyerRecommendSetting(APIView): """商机推荐设置""" def post(self, request): user_id = request.POST.get('user_id') setting_value = request.POST.get('settting_value') try: s_values = literal_eval(setting_value) except: return Response(CheckResult.check('格式错误')) else: for value in s_values[0].values(): if value.strip() == '': return Response(CheckResult.check('存在推荐关键词为空,请求失败')) res = BusOpportunitySetting.objects.filter(user_id=user_id) if not res: setting_id = snowflake.client.get_guid() # 雪花算法存id create_time = datetime.datetime.now() BusOpportunitySetting.objects.create(setting_id=setting_id, user_id=user_id, settting_value=setting_value, create_time=create_time) # 创建一条数据到数据库 else: res.update(settting_value=setting_value) # 更新一条数据库指定字段数据 result = { 'errcode': 0, 'errmsg': '提交成功' } return Response(result) class OpportunityFollowSave(APIView): def post(self, request): user_id = request.POST.get('user_id') bo_id = request.POST.get('bo_id') follow_type = request.POST.get('follow_type') res = BusOpportunityFollow.objects.filter(user_id=user_id, bo_id=bo_id) # 过滤筛选 if not res: if follow_type == '1': follow_id = snowflake.client.get_guid() create_time = datetime.datetime.now() BusOpportunityFollow.objects.create(follow_id=follow_id, user_id=user_id, bo_id=bo_id, create_time=create_time) # 创建一条数据到数据库 # （注：.create创建后，可以不用res.save(),因为创建相当于已经保存数据到数据库了，再save()可能就是保存了两次进数据库） elif follow_type == '2': BusOpportunityFollow.objects.filter(user_id=user_id, bo_id=bo_id).delete() # 数据库删除指定的一条数据 else: return Response(CheckResult.check('关注状态传入错误')) return Response({'errcode': 0, 'errmsg': '操作成功', }) else: return Response(CheckResult.check('已关注'))
重写pagination设置页数：
my_pagination.py
from rest_framework import pagination class MyPagination(pagination.PageNumberPagination): page_size = 10 # 每页数目 page_query_param = 'page' page_size_query_param = 10 # 前端最多能涉及的每页数量
关于雪花算法模块：
# 安装：pip install pysnowflake # 启动：snowflake_start_server # snowflake_start_server --address=127.0.0.1 --port=8910 --dc=1 --worker=1 # 引入 import snowflake.client # 应用 guid = snowflake.client.get_guid()