flyapps/fir_ser/api/views/notify.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# project: fir_ser
# filename: notify
# author: liuyu
# data: 2022/3/25


import logging

from rest_framework.response import Response
from rest_framework.views import APIView

from api.models import NotifyReceiver, ThirdWeChatUserInfo, NotifyConfig, UserInfo
from api.utils.auth.util import AuthInfo
from api.utils.modelutils import PageNumber
from api.utils.response import BaseResponse
from api.utils.serializer import NotifyReceiverSerializer, NotifyConfigSerializer
from api.views.login import check_common_info
from common.base.baseutils import get_choices_dict, get_choices_name_from_key, is_valid_email, is_valid_phone
from common.core.auth import ExpiringTokenAuthentication
from common.core.sysconfig import Config
from common.utils.caches import login_auth_failed
from common.utils.sendmsg import is_valid_sender_code

logger = logging.getLogger(__name__)


class NotifyConfigView(APIView):
    authentication_classes = [ExpiringTokenAuthentication, ]

    def get(self, request):
        res = BaseResponse()
        config_pk = request.query_params.get('pk')

        obj_lists = NotifyConfig.objects.filter(user_id=request.user).order_by('message_type').order_by("-create_time")
        if config_pk:
            obj_lists = obj_lists.filter(pk=config_pk).first()
            info = NotifyConfigSerializer(obj_lists)
            res.data = info.data
            return Response(res.dict)

        info = NotifyConfigSerializer(obj_lists.all(), many=True).data
        res.count = obj_lists.count()
        message_type_choices = get_choices_dict(NotifyConfig.message_type_choices)
        for message_info in message_type_choices:
            message_info['data'] = []
            for notify_config_info in info:
                if notify_config_info['message_type'] == message_info['id']:
                    message_info['data'].append(notify_config_info)
            if len(message_info['data']) == 0:
                data_info = NotifyConfigSerializer(NotifyConfig.objects.filter(pk=-1).first()).data
                data_info['config_name'] = message_info['name']
                data_info['message_type'] = message_info['id']
                data_info['m_type'] = message_info['id']
                data_info['id'] = -1
                message_info['data'].append(data_info)
        res.message_type_choices = message_type_choices
        return Response(res.dict)

    def post(self, request):
        res = BaseResponse()
        data = request.data
        act = data.get('act', '')
        if act and act != 'get':
            threshold = data.get('threshold', {})
            if threshold:
                notify_available_downloads = threshold.get('notify_available_downloads')
                notify_available_signs = threshold.get('notify_available_signs')

                if notify_available_downloads is not None:
                    request.user.notify_available_downloads = notify_available_downloads
                if notify_available_signs is not None:
                    request.user.notify_available_signs = notify_available_signs
                if notify_available_downloads is not None or notify_available_signs is not None:
                    request.user.save(update_fields=['notify_available_downloads', 'notify_available_signs'])

        res.data = UserInfo.objects.filter(pk=request.user.pk).values('notify_available_downloads',
                                                                      'notify_available_signs').first()
        return Response(res.dict)

    def put(self, request):
        res = BaseResponse()
        data = request.data
        config_pk = data.get('config')
        receiver_ids = data.get('receiver')
        enable_email = data.get('enable_email')
        enable_weixin = data.get('enable_weixin')
        m_type = data.get('m_type')
        if config_pk and config_pk == -1 and m_type is not None:
            notify_config_obj = NotifyConfig.objects.filter(user_id=request.user, message_type=int(m_type)).filter()
            if not notify_config_obj:
                notify_config_obj = NotifyConfig.objects.create(user_id=request.user, message_type=int(m_type),
                                                                config_name=get_choices_name_from_key(
                                                                    NotifyConfig.message_type_choices, int(m_type)))
        else:
            notify_config_obj = NotifyConfig.objects.filter(user_id=request.user, pk=config_pk).first()
        if notify_config_obj:
            if isinstance(receiver_ids, list):
                notify_config_obj.sender.set(
                    NotifyReceiver.objects.filter(user_id=request.user, pk__in=receiver_ids).all())
            if enable_weixin is not None:
                notify_config_obj.enable_weixin = enable_weixin
            if enable_email is not None:
                notify_config_obj.enable_email = enable_email
            if enable_weixin is not None or enable_email is not None:
                notify_config_obj.save(update_fields=['enable_email', 'enable_weixin'])

        return Response(res.dict)


class NotifyReceiverView(APIView):
    authentication_classes = [ExpiringTokenAuthentication, ]

    def get(self, request):
        res = BaseResponse()
        config_pk = request.query_params.get('config')
        message_type = request.query_params.get('message_type')

        obj_lists = NotifyReceiver.objects.filter(user_id=request.user)

        config_obj_lists = NotifyConfig.objects.filter(user_id=request.user)
        if config_pk and config_pk != '-1':
            res.senders = config_obj_lists.values('sender').filter(pk=config_pk).all()
            config_obj = config_obj_lists.filter(pk=config_pk).first()
            if config_obj:
                res.config = {'config_name': config_obj.config_name,
                              'id': config_obj.pk, 'm_type': config_obj.message_type,
                              'message_type': config_obj.get_message_type_display()}
        if config_pk and config_pk == '-1' and message_type is not None:
            res.senders = config_obj_lists.values('sender').filter(pk=config_pk).all()
            config_name = get_choices_name_from_key(NotifyConfig.message_type_choices, int(message_type))
            res.config = {'config_name': config_name, 'id': -1, 'message_type': config_name,
                          'm_type': message_type}

        page_obj = PageNumber()
        obj_info_serializer = page_obj.paginate_queryset(queryset=obj_lists.order_by("-create_time"),
                                                         request=request,
                                                         view=self)
        info = NotifyReceiverSerializer(obj_info_serializer, many=True, )
        res.data = info.data
        res.count = obj_lists.count()
        return Response(res.dict)

    def post(self, request):
        res = BaseResponse()
        data = request.data
        receiver_name = data.get('receiver_name')
        description = data.get('description')
        wxopenid = data.get('wxopenid')
        email = data.get('email')
        if receiver_name:
            data_info = {
                'receiver_name': receiver_name,
                'description': description,
            }
            wx_obj = None
            if wxopenid:
                wx_obj = ThirdWeChatUserInfo.objects.filter(user_id=request.user, openid=wxopenid).first()
                if wx_obj:
                    data_info['weixin'] = wx_obj
            if email:
                if is_valid_email(email):
                    act = 'email'
                elif is_valid_phone(email):
                    act = 'sms'
                else:
                    res.code = 1001
                    res.msg = "邮箱或手机校验有误"
                    return Response(res.dict)

                is_valid, target = is_valid_sender_code(act, data.get("auth_token", None),
                                                        data.get("seicode", None))
                if is_valid and str(target) == str(email):
                    if login_auth_failed("get", email):
                        data_info['email'] = email
            try:
                if not data_info.get('email') and not data_info.get('weixin'):
                    raise Exception('至少存在一个联系方式')
                NotifyReceiver.objects.create(user_id=request.user, **data_info)
                if wx_obj:
                    wx_obj.enable_notify = True
                    wx_obj.save(update_fields=['enable_notify'])
            except Exception as e:
                logger.error(f"{request.user} notify receiver add failed . data:{data} Exception:{e}")
                res.code = 1001
                res.msg = '数据有误或者已经存在该接收人信息'
        return Response(res.dict)

    def delete(self, request):
        res = BaseResponse()
        pk = request.query_params.get('id')
        if pk:
            NotifyReceiver.objects.filter(user_id=request.user, pk=pk).delete()
        return Response(res.dict)


class NotifyInfoView(APIView):
    authentication_classes = [ExpiringTokenAuthentication, ]

    def get(self, request):
        response = BaseResponse()
        response.data = {}
        allow_f = Config.NOTIFY.get("enable")
        if allow_f:
            auth_obj = AuthInfo(Config.NOTIFY.get("captcha"), Config.NOTIFY.get("geetest"))
            response.data['auth_rules'] = auth_obj.make_rules_info()
            response.data['notify_type'] = Config.NOTIFY.get("notify_type")
        response.data['enable'] = allow_f
        return Response(response.dict)

    def put(self, request):
        response = BaseResponse()
        user_id = request.data.get('user_id', None)
        if user_id:
            auth_obj = AuthInfo(Config.NOTIFY.get("captcha"), Config.NOTIFY.get("geetest"))
            response.data = auth_obj.make_geetest_info(user_id, request.META.get('REMOTE_ADDR'))
        else:
            response.code = 1002
            response.msg = '参数错误'
        return Response(response.dict)

    def post(self, request):
        res = BaseResponse()
        res.data = {}
        target = request.data.get("target", None)
        p_data = Config.NOTIFY
        auth_obj = AuthInfo(p_data.get("captcha"), p_data.get("geetest"))
        is_valid, msg = auth_obj.valid(request.data)
        if not is_valid:
            res.code = 1001
            res.msg = msg
            return Response(res.dict)

        if is_valid_email(target):
            act = 'email'
        elif is_valid_phone(target):
            act = 'sms'
        else:
            res.code = 1001
            res.msg = "邮箱或手机校验有误"
            return Response(res.dict)

        res = check_common_info(target, act, 'notify')
        return Response(res.dict)