ASJServer/Object/UnicomObject.py

# -*- encoding: utf-8 -*-
"""
@File    : UnicomObject.py
@Time    : 2022/6/17 11:03
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""
import base64
import json
import logging

import requests
from Crypto.Cipher import AES

from Ansjer.config import unicomAppUrl, unicomAppId, unicomAppSecret, unicomTenantId, \
    unicomEncodeKey, unicomIvKey, unicomUserName, unicomPassword, unicomPushKey
from Object.RedisObject import RedisObject
from Object.utils import SM3Util
from Object.utils.SymmetricCryptoUtil import AESencrypt

"""
联通4Gapi
具体参数查看接口文档
https://www.showdoc.com.cn/unicomJYHapi/8158648460007467
"""

logger = logging.getLogger('info')


class UnicomObjeect:
    def __init__(self):
        self.appUrl = unicomAppUrl
        self.appId = unicomAppId
        self.appSecret = unicomAppSecret
        self.tenantId = unicomTenantId
        self.encodeKey = unicomEncodeKey
        self.ivKey = unicomIvKey
        self.username = unicomUserName
        self.password = unicomPassword
        self.pushKey = unicomPushKey
        self.headers = {'Tenant': self.tenantId, 'content-type': 'application/x-www-form-urlencoded'}

    # pip install snowland-smx
    def createSign(self, reverse=False, **sign_params):
        """
        调用接口(API)时需要对请求参数进行签名（sign）验证，
        算法：
        根据参数名称将你的所有请求参数按照字母先后顺序排序: key = value & key = value，对除签名外的所有请求参数按
        key
        做的升序排列。
        如：将
        foo = 1, bar = 2, baz = 3
        排序为
        bar = 2, baz = 3, foo = 1
        参数名和参数值链接后，得到拼装字符串（注意：参数名与参数值左右两边不能包含空格）
        bar = 2 & baz = 3 & foo = 1
        将
        pushkey拼接到参数字符尾部进行
        SM3
        加密，再转化成大写，格式是
        (SM3(key1=value1 & key2=value2 &...& key= pushKey)).upcase
        @param reverse:
        @param sign_params:
        @return:
        """
        dict_2 = dict(sorted(sign_params.items(), key=lambda item: item[0], reverse=reverse))
        data_list = []
        for item in dict_2.items():
            if item[0] and item[1]:
                data_list.append("{}={}".format(item[0], item[1]))
        val = '&'.join(data_list)
        push_key = '&key={}'.format(self.pushKey)
        val = val + push_key
        return SM3Util.Hash_sm3(val).upper()

    def get_login_authorization(self):
        """
        获取登录授权
        注意登录认证Authorization和登录后的请求认证不一样
        算法：appId+":"+appSecret base64转码生成 前面加Basic
        @return: "Basic " + base64_data
        """
        voucher = self.appId + ':' + self.appSecret
        base64_data = str(base64.b64encode(voucher.encode("utf-8")), "utf-8")
        return "Basic " + base64_data

    def get_encode_password(self):
        """
        获取对称加密
        @return: encrypt_pwd
        """
        aes = AESencrypt(self.encodeKey.encode('utf-8'), AES.MODE_CBC, self.ivKey.encode('utf-8'),
                         paddingMode="ZeroPadding",
                         characterSet='utf-8')
        encrypt_pwd = aes.encryptFromString(self.password)
        return encrypt_pwd

    def generate_token(self):
        """
        生成令牌
        @return: token
        """
        url = self.appUrl + '/auc/oauth/token'
        pwd = self.get_encode_password()
        body = {'username': self.username, 'password': pwd, 'grant_type': 'password', 'scope': 'server'}
        headers = self.headers
        headers['Authorization'] = self.get_login_authorization()
        response_data = requests.post(url, data=body, headers=headers)
        response_data = json.loads(response_data.text)
        return response_data['access_token']

    def refresh_token(self, refresh_token):
        """
        刷新令牌
        @param refresh_token:
        @return:
        """
        url = self.appUrl + '/auc/oauth/token?grant_type=refresh_token'
        body = {'refresh_token': refresh_token, 'grant_type': 'refresh_token'}
        headers = self.headers
        headers['Authorization'] = self.get_login_authorization()
        response_data = requests.post(url, data=body, headers=headers)
        return response_data.text

    def business_unify_headers(self):
        """
        业务统一headers
        在请求头中增加key为Authorization，value为"Bearer " + token
        @return: headers
        """
        token = self.generate_token()
        headers = self.headers
        headers['Authorization'] = 'Bearer ' + token
        return headers

    # 业务api  注意 get与post 请求数据类型不同 post使用：application/json
    def verify_device(self, **re_params):
        """
        验证设备
        @param re_params:
        @return:
        """
        url = self.appUrl + '/platform/api/device/verify-device'
        return requests.get(url, params=re_params, headers=self.business_unify_headers())

    def query_device_status(self, **re_params):
        """
        查询设备状态
        @param re_params:
        @return:
        """
        url = self.appUrl + '/platform/api/device/device-status'
        return requests.get(url, params=re_params, headers=self.business_unify_headers())

    def update_device_state(self, **re_data):
        """
        修改设备状态
        @param re_data:
        @return:
        """
        url = self.appUrl + '/platform/api/device/device-state'
        headers = self.business_unify_headers()
        headers['content-type'] = 'application/json'
        return requests.post(url, data=json.dumps(re_data), headers=headers)

    def query_device_usage_history(self, **re_params):
        """
        查询设备用量历史
        @param re_params:
        @return:
        """
        url = self.appUrl + '/platform/api/usage/device-usage-history'
        return requests.get(url, params=re_params, headers=self.business_unify_headers())

    def query_current_renew_list_usage_details(self, **re_params):
        """
        查询设备当前队列用量详情
        @param re_params:
        @return:
        """
        url = self.appUrl + '/platform/api/usage/current-renewlist-usage-details'
        return requests.get(url, params=re_params, headers=self.business_unify_headers())

    def get_device_batch_detail(self, **re_data):
        """
        查询设备当前队列用量详情
        @return:
        """
        url = self.appUrl + '/platform/api/device/batch-detail'
        headers = self.business_unify_headers()
        headers['content-type'] = 'application/json'
        return requests.post(url, data=json.dumps(re_data), headers=headers)

    def query_package_list(self, **re_params):
        """
        查询套餐列表
        @return:
        """
        url = self.appUrl + '/platform/api/package/list'
        return requests.get(url, params=re_params, headers=self.business_unify_headers())

    def query_renewal_list(self, **re_params):
        """
        续费套餐列表
        @param re_params:
        @return:
        """
        url = self.appUrl + '/platform/api/package/list'
        return requests.get(url, params=re_params, headers=self.business_unify_headers())

    def async_buy_package(self, **re_data):
        """
        查询设备当前队列用量详情
        @return:
        """
        url = self.appUrl + '/platform/api/package/async-buy-package'
        headers = self.business_unify_headers()
        headers['content-type'] = 'application/json'
        return requests.post(url, data=json.dumps(re_data), headers=headers)

    @staticmethod
    def get_text_dict(result):
        """
        响应结果转字典
        @param result:
        @return:
        """
        if result.status_code == 200:
            return json.loads(result.text)
        return None

    @staticmethod
    def get_flow_total_usage(key, expire=0, **usage_data):
        """
        设备当前队列用量详情（实现缓存）
        @param key: 缓存key
        @param expire: 失效时间
        @param usage_data: 查询参数
        @return: 返回结果dict
        """
        redis = RedisObject()
        flow_usage_details = redis.get_data(key)
        if flow_usage_details:
            flow_usage_details = json.loads(flow_usage_details)
        else:
            flow_usage_details = UnicomObjeect().query_current_renew_list_usage_details(**usage_data)
            flow_usage_details = UnicomObjeect().get_text_dict(flow_usage_details)
            redis.set_data(key=key, val=json.dumps(flow_usage_details), expire=expire)
        return flow_usage_details

    @staticmethod
    def get_flow_usage_total(iccid):
        """
        获取实时当前套餐队列总已用流量
        @param iccid: 联通id
        @return: flow_total_usage 当前套餐总已使用流量
        """
        flow_key = 'ASJ:UNICOM:FLOW:{}'
        usage_data = {'iccid': iccid, 'realTime': True}
        expire_time = 60 * 10 + 60
        flow_usage_details = UnicomObjeect().get_flow_total_usage(flow_key.format(iccid), expire_time, **usage_data)
        # 当月实际总使用流量
        flow_total_usage = 0
        if flow_usage_details and flow_usage_details['success']:
            flow_total_usage = flow_usage_details['data']['flowTotalUsage']
        return flow_total_usage

    @staticmethod
    def change_device_to_activate(iccid):
        """
        根据iccid判断是否激活，未激活则修改为激活状态
        @param iccid:
        @return:
        """
        if iccid:
            re_data = {'iccid': iccid}
            result = UnicomObjeect().query_device_status(**re_data)
            res_dict = UnicomObjeect().get_text_dict(result)
            # 状态不等于1（激活）时进行激活 1:激活；2:停用
            if res_dict['data']['status'] != 1:
                re_data = {"iccid": iccid, "status": 1}
                UnicomObjeect().update_device_state(**re_data)
            return True
        return None

    @staticmethod
    def change_device_to_disable(iccid):
        """
        修改设备为停用,并查看是否修改成功
        @param iccid:
        @return:
        """
        if iccid:
            re_data = {"iccid": iccid, "status": 2}
            response = UnicomObjeect().update_device_state(**re_data)
            logger.info('停用iccid响应结果:{}'.format(response.text))
            # 查询是否停用成功
            re_data.pop('status')
            result = UnicomObjeect().query_device_status(**re_data)
            res_dict = UnicomObjeect().get_text_dict(result)
            logger.info('查询iccid状态：{}'.format(res_dict))
            if res_dict['data']['status'] != 3:
                re_data['status'] = 2
                response = UnicomObjeect().update_device_state(**re_data)
                logger.info('再次停卡：{}'.format(response.text))
            return True
        return None