ASJServer/Controller/UnicomCombo/WXTechController.py

# -*- encoding: utf-8 -*-
"""
@File    : WXTechController.py
@Time    : 2023/5/6 16:40
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""
import time

from django.http import QueryDict
from django.views import View

from Ansjer.config import LOGGER
from Controller.UnicomCombo.UnicomComboController import UnicomComboView
from Model.models import UnicomDeviceInfo, Order_Model, UnicomComboExperienceHistory, UnicomCombo, ExchangeCode, \
    LogModel
from Object.Enums.WXOperatorEnum import WXOperatorEnum
from Object.Enums.WXStartTypeEnum import WXStartTypeEnum
from Object.ResponseObject import ResponseObject
from Object.WXTechObject import WXTechObject
from Service.CommonService import CommonService

# 五兴科技电信套餐编码 1G-15天流量包
COMBO_ID = 'DX-DX-FDX-JCB-1G-15天-4DX-ASJ-YTC-AT-2022123020221230144615'


class WXTechController(View):
    class SmartSocketView(View):
        def get(self, request, *args, **kwargs):
            request.encoding = 'utf-8'
            operation = kwargs.get('operation')
            return self.validation(request.GET, request, operation)

        def post(self, request, *args, **kwargs):
            request.encoding = 'utf-8'
            operation = kwargs.get('operation')
            return self.validation(request.POST, request, operation)

        def delete(self, request, *args, **kwargs):
            request.encoding = 'utf-8'
            operation = kwargs.get('operation')
            delete = QueryDict(request.body)
            if not delete:
                delete = request.GET
            return self.validation(delete, request, operation)

        def put(self, request, *args, **kwargs):
            request.encoding = 'utf-8'
            operation = kwargs.get('operation')
            put = QueryDict(request.body)
            return self.validation(put, request, operation)

        def validation(self, request_dict, request, operation):
            if operation == 'deleteCardPackage':
                return self.delete_card_package(request_dict, ResponseObject('cn'))
            elif operation == 'packageRefund':
                return self.wx_package_refund(request_dict, ResponseObject('cn'))
            token_code, user_id, response = CommonService \
                .verify_token_get_user_id(request_dict, request)
            if token_code != 0:
                return response.json(token_code)
            if operation == 'getCardsInfo':
                return self.get_cards_info(request_dict, response)
            elif operation == 'getExperiencePackage':
                return self.get_experience_package(request_dict, response)
            elif operation == 'createOrder':
                return self.create_order_package(request_dict, response)
            elif operation == 'exchangePackage':
                return self.wx_exchange_package(request_dict, response, request, user_id)

        @classmethod
        def wx_package_refund(cls, request_dict, response):
            """
            五兴套餐退订
            """
            try:
                serial_number = request_dict.get('serialNumber', None)
                if not serial_number:
                    return response(444)
                device_qs = UnicomDeviceInfo.objects.filter(serial_no=serial_number).values('iccid')
                if not device_qs.exists():
                    return response(173)
                uid = CommonService.query_uid_with_serial(serial_number)
                order_info_qs = Order_Model.objects.filter(UID=uid, order_type=3).values('trade_no') \
                    .order_by('-payTime')
                if not order_info_qs.exists():
                    return response(173)
                iccid = device_qs[0]['iccid']
                trade_no = order_info_qs[0]['trade_no']
                data = {'iccid': iccid, 'operator': 3, 'orderNumber': trade_no}
                wx_tech = WXTechObject()
                res = wx_tech.package_refund(**data)
                if res['code'] == '0':
                    return response.json(0)
                return response.json(10059)
            except Exception as e:
                LOGGER.info('*****WXTechController.get_cards_info:errLine:{}, errMsg:{}'
                            .format(e.__traceback__.tb_lineno, repr(e)))
                return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

        @classmethod
        def get_cards_info(cls, request_dict, response):
            """
            五兴单卡获取信息
            """
            try:
                LOGGER.info('*****WXTechController.get_cards_info:params:{}'.format(request_dict))
                iccid = request_dict.get('iccid', None)
                operator = int(request_dict.get('operator', 3))
                if not iccid:
                    return response.json(444)
                data = {'iccid': iccid, 'operator': operator}
                wx_tech = WXTechObject()
                return response.json(0, wx_tech.get_cards_info(**data))
            except Exception as e:
                LOGGER.info('*****WXTechController.get_cards_info:errLine:{}, errMsg:{}'
                            .format(e.__traceback__.tb_lineno, repr(e)))
                return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

        @classmethod
        def get_experience_package(cls, request_dict, response):
            """
            获取体验套餐
            """
            try:
                LOGGER.info('*****WXTechController.get_experience_package:params{}'.format(request_dict))
                serial_no = request_dict.get('serialNo', None)
                if not serial_no:
                    return response.json(444)
                # 查询是否注册iccid
                unicom_device_qs = UnicomDeviceInfo.objects.filter(serial_no=serial_no) \
                    .values('serial_no', 'user_id', 'iccid')
                if not unicom_device_qs.exists():
                    return response.json(173)
                unicom_device_qs = unicom_device_qs.first()
                iccid = unicom_device_qs['iccid']
                # 查看是否体验过免费套餐
                experience_history_qs = UnicomComboExperienceHistory.objects.filter(iccid=iccid)
                if experience_history_qs.exists():
                    return response.json(10062)
                combo_info_qs = UnicomCombo.objects.filter(combo_type=1, status=0, is_del=False) \
                    .values('combo_name')
                if not combo_info_qs.exists():
                    return response.json(173)
                result = {'comboName': combo_info_qs[0]['combo_name']}
                return response.json(0, result)
            except Exception as e:
                LOGGER.info('*****WXTechController.get_experience_package:errLine:{}, errMsg:{}'
                            .format(e.__traceback__.tb_lineno, repr(e)))
                return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

        @classmethod
        def create_order_package(cls, request_dict, response):
            """
            五兴电信领取1G15天流量体验包
            """
            try:
                LOGGER.info('*****WXTechController.create_order_package:params:{}'.format(request_dict))
                serial_no = request_dict.get('serialNo', None)
                operator = request_dict.get('operator', None)
                if not all([operator, serial_no]):
                    return response.json(444)
                unicom_device_qs = UnicomDeviceInfo.objects.filter(serial_no=serial_no) \
                    .values('serial_no', 'user_id', 'iccid')
                if not unicom_device_qs.exists():
                    return response.json(173)
                unicom_device_qs = unicom_device_qs.first()
                user_id = unicom_device_qs['user_id']
                iccid = unicom_device_qs['iccid']
                # 查看是否体验过免费套餐
                experience_history_qs = UnicomComboExperienceHistory.objects.filter(iccid=iccid)
                if experience_history_qs.exists():
                    return response.json(10062)
                data = {'iccid': iccid, 'operator': WXOperatorEnum(int(operator)).value,
                        'startType': str(WXStartTypeEnum.EFFECTIVE_IMMEDIATELY.value), 'packageCode': COMBO_ID}
                wx_tech = WXTechObject()
                # 请求五兴API创建套餐接口
                res = wx_tech.create_order_package(**data)
                LOGGER.info('*****五兴创建体验套餐信息:{}'.format(res))
                if res['code'] == '0':
                    trade_no = res['data']['orderNumber']
                    experience_history_vo = {'iccid': iccid, 'experience_type': 0, 'do_time': int(time.time())}
                    UnicomComboExperienceHistory.objects.create(**experience_history_vo)
                    order_res = cls.created_order(serial_no, user_id, trade_no, pay_type=10)
                    LOGGER.info('*****系统创建体验订单:{}'.format(order_res))
                    return response.json(0)
                return response.json(10063)
            except Exception as e:
                LOGGER.info('*****WXTechController.create_order_package:errLine:{}, errMsg:{}'
                            .format(e.__traceback__.tb_lineno, repr(e)))
                return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

        @classmethod
        def created_order(cls, serial_no, user_id, trade_no, combo_id=None, pay_type=10):
            """
            创建系统订单
            """
            combo_info_qs = UnicomCombo.objects.filter(is_del=False)
            if combo_id:
                combo_info_qs = combo_info_qs.filter(id=int(combo_id))
            else:
                combo_info_qs = combo_info_qs.filter(combo_type=1, status=0)
            combo_info_qs = combo_info_qs.values('id', 'combo_name', 'price', 'virtual_price', 'remark') \
                .order_by('sort')
            if not combo_info_qs.exists():
                return False
            combo_info_vo = combo_info_qs[0]
            n_time = int(time.time())
            # 根据序列号获取UID
            uid = CommonService.query_uid_with_serial(serial_no)
            order_id = CommonService.createOrderID()
            # 生成订单必须添加该字段
            rank_id, ai_rank_id = UnicomComboView().get_cloud_or_ai_combo()
            order_dict = {'orderID': order_id, 'UID': uid, 'rank_id': rank_id, 'ai_rank_id': ai_rank_id,
                          'userID_id': user_id, 'desc': combo_info_vo['combo_name'], 'payType': pay_type,
                          'payTime': n_time, 'price': combo_info_vo['price'], 'addTime': n_time,
                          'updTime': n_time, 'status': 1, 'currency': 'CNY',
                          'unify_combo_id': str(combo_info_vo['id']), 'order_type': 3,
                          'store_meal_name': combo_info_vo['combo_name'],
                          'trade_no': trade_no}
            Order_Model.objects.create(**order_dict)
            return order_id

        @classmethod
        def delete_card_package(cls, request_dict, response):
            """
            PC工具清空卡片的所有套餐订购记录
            """
            try:
                LOGGER.info('*****WXTechController.delete_card_package:params:{}'.format(request_dict))
                sign = request_dict.get('sign', None)
                serial_no = request_dict.get('serialNo', None)
                time_stamp = request_dict.get('timeStamp', None)
                if not CommonService.check_time_stamp_token(sign, time_stamp):
                    return response.json(13)
                    # 查询是否注册iccid
                unicom_device_qs = UnicomDeviceInfo.objects.filter(serial_no=serial_no) \
                    .values('serial_no', 'user_id', 'iccid')
                if not unicom_device_qs.exists():
                    return response.json(173)
                iccid = unicom_device_qs[0]['iccid']
                data = {'iccids': iccid, 'operator': WXOperatorEnum.TELECOM.value}
                wx_tech = WXTechObject()
                res = wx_tech.delete_card_package(**data)
                if res['code'] == '0':
                    UnicomComboExperienceHistory.objects.filter(iccid=iccid).delete()
                    return response.json(0)
                return response.json(177)
            except Exception as e:
                LOGGER.info('*****WXTechController.delete_card_package:errLine:{}, errMsg:{}'
                            .format(e.__traceback__.tb_lineno, repr(e)))
                return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

        @classmethod
        def wx_exchange_package(cls, request_dict, response, request, user_id):
            """
            使用兑换码订购五兴五年流量套餐包
            """
            ip = CommonService.get_ip_address(request)
            try:
                code = request_dict.get('code')
                serial_number = request_dict.get('serialNumber')
                LOGGER.info('*****WXTechController.wx_exchange_package:params,{}'.format(request_dict))
                if not all([code, serial_number]):
                    return response.json(444)  # 参数缺失
                # 校验兑换码格式是否正确
                if not (code.isalnum() and len(code) == 10):
                    return response.json(10040)  # 兑换码格式错误
                device_info_qs = UnicomDeviceInfo.objects.filter(serial_no=serial_number) \
                    .values('iccid', 'card_type')
                if not device_info_qs.exists():
                    return response.json(173)  # 设备信息不存在
                #  查询兑换码信息
                exchange_code_qs = ExchangeCode.objects.filter(code=code, status=False) \
                    .values('package_id', 'expire_time', 'package_type')
                if not exchange_code_qs.exists():
                    return response.json(10040)  # 兑换码无效
                combo_id = exchange_code_qs[0]['package_id']
                # 获取五兴套餐套餐编码
                package_info_qs = UnicomCombo.objects.filter(id=combo_id, is_del=False) \
                    .values('package_id')
                if not package_info_qs.exists():
                    return response.json(173)
                package_code = package_info_qs[0]['package_id']
                iccid = device_info_qs[0]['iccid']
                # 兑换码套餐类型
                card_type = device_info_qs[0]['card_type']
                if card_type == 0:  # 联通无限流量兑换码
                    result = cls.exchange_unicom_package(serial_number, iccid,
                                                         user_id, combo_id)
                    if result:
                        exchange_code_qs.update(status=True, updated_time=int(time.time()))
                        cls.save_log(ip, 200, '兑换成功{},{}'.format(serial_number, code))
                        return response.json(0)
                elif card_type == 1:  # 五兴电信无限流量兑换码
                    # 五兴订购流量包请求参数
                    data = {'iccid': iccid, 'operator': WXOperatorEnum.TELECOM.value,
                            'startType': str(WXStartTypeEnum.EFFECTIVE_IMMEDIATELY.value), 'packageCode': package_code}
                    wx_tech = WXTechObject()
                    # 请求五兴API订购套餐接口
                    res = wx_tech.create_order_package(**data)
                    LOGGER.info('*****五兴订购年卡套餐结果:{}'.format(res))
                    if res['code'] == '0':
                        trade_no = res['data']['orderNumber']
                        cls.created_order(serial_number, user_id, trade_no, combo_id, 11)
                        exchange_code_qs.update(status=True, updated_time=int(time.time()))
                        cls.save_log(ip, 200, '兑换成功{},{}'.format(serial_number, code))
                        return response.json(0)
                cls.save_log(ip, 10040, '兑换失败{},{}'.format(serial_number, code))
                return response.json(10040)
            except Exception as e:
                LOGGER.info('*****WXTechController.wx_exchange_package:errLine:{}, errMsg:{}'
                            .format(e.__traceback__.tb_lineno, repr(e)))
                cls.save_log(ip, 500, '兑换内部错误')
                return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

        @classmethod
        def exchange_unicom_package(cls, serial_number, iccid, user_id, combo_id):
            try:
                # 通过兑换码创建订单记录
                order_id = cls.created_order(serial_number, user_id, 'ansjer', combo_id, 11)
                # 根据订单信息创建流量套餐包
                UnicomComboView.create_combo_order_info(order_id, 0, iccid, combo_id)
                return True
            except Exception as e:
                LOGGER.info('*****WXTechController.exchange_unicom_package:errLine:{}, errMsg:{}'
                            .format(e.__traceback__.tb_lineno, repr(e)))
                return False

        @staticmethod
        def save_log(ip, code, operation):
            """
            保存操作日志
            @param ip: ip地址
            @param code: 状态码
            @param operation: 操作说明
            @return: 保存结果
            """
            try:
                log = {
                    'ip': ip,
                    'user_id': 1,
                    'status': code,
                    'time': int(time.time()),
                    'operation': operation,
                    'url': 'unicom/open/wxtech/exchangePackage',
                }
                LogModel.objects.create(**log)
                return True
            except Exception as e:
                LOGGER.info('*****WXTechController.save_log:errLine:{}, errMsg:{}'
                            .format(e.__traceback__.tb_lineno, repr(e)))
                return False