manyi/apps/web/api/jn_north/views.py

# coding=utf-8
import json
import logging
import datetime

from mongoengine import DoesNotExist

from django.views.decorators.http import require_POST

from apilib.utils_datetime import to_datetime
from apilib.utils_json import JsonResponse
from apps.web.api.jn_north.constant import RESPONSE_CODE
from apps.web.api.jn_north.utils import get_stations_info,get_station_status,get_get_station_state,get_policy_info
from apps.web.api.utils import AES_CBC_PKCS5padding_encrypt, AES_CBC_PKCS5padding_decrypt, generate_json_token, parse_json_token
from apps.web.device.models import Group, Device
from apps.web.south_intf.shangdong_platform import ShanDongNorther, GroupIdMap

logger = logging.getLogger(__name__)


@require_POST
def queryToken(request):
    """
    通过账号密码获取token
    """
    logger.debug("[queryToken] request body = {}".format(request.body))

    if not request.body:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1001）"})

    try:
        Data = json.loads(request.body).get("Data")
        data = json.loads(AES_CBC_PKCS5padding_decrypt(Data))
    except Exception as e:
        logger.exception(e)
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1002）"})

    OperatorID = data.get("OperatorID")
    OperatorSecret = data.get("OperatorSecret")

    logger.debug("[queryToken] OperatorID = {}, OperatorSecret = {}".format(OperatorID, OperatorSecret))

    if not all((OperatorID, OperatorSecret)):
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1003）"})

    try:
        norther = ShanDongNorther.objects.filter(northOperatorID=OperatorID, northOperatorSecret=OperatorSecret).first()   # type: ShanDongNorther
    except DoesNotExist:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1004）"})
    except Exception as e:
        return JsonResponse({"Ret": RESPONSE_CODE.SYS_ERROR, "Msg": u"系统错误"})

    expire = 60 * 60 * 24 * 7
    result = {
        "OperatorID": norther.agentOperatorID,
        "SuccStat": 0,
        "AccessToken": generate_json_token(data=norther.get_token_data(), expire=expire),
        "TokenAvailableTime": expire,
        "FailReason": None
    }
    logger.debug("[queryToken] return result = {}".format(result))

    # 拉取的时候加密 显式指明加密秘钥为 pull
    resultData = AES_CBC_PKCS5padding_encrypt(
        json.dumps(result),
        dataSecret=norther.pullDataSecret,
        dataSecretIV=norther.pullDataSecretIV
    )
    sig = norther.get_sig(resultData)

    return JsonResponse({
        "Ret": RESPONSE_CODE.SUCCESS,
        "Msg": u"请求成功",
        "Data": resultData,
        "Sig": sig
    })

def queryStationsInfo(request):
    """
    查询充电站的信息
    """
    token = request.META.get('HTTP_AUTHORIZATION', "").replace("Bearer", "").strip()
    logger.info('[queryStationsInfo] , token = {}'.format(token))

    # 验证身份
    tokenData = parse_json_token(token)
    if not tokenData:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_TOKEN, "Msg": u"请求参数错误（1001）"})

    # 获取这个平台下面的所有的northers记录
    northers = ShanDongNorther.get_norther(**tokenData)
    if not northers:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_PARAM, "Msg": u"请求参数错误（1002）"})

    # 准备token所获取的参数信息
    norther = northers.first()
    pullDataSecret = norther.pullDataSecret
    pullDataSecretIV = norther.pullDataSecretIV

    # 这个平台下的所有对接过的经销商
    dealerIds = [_.dealerId for _ in northers]

    # 验证参数
    logger.debug("[queryStationsInfo] request body = {}".format(request.body))

    if not request.body:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1003）"})

    try:
        Data = json.loads(request.body).get("Data")
        data = json.loads(AES_CBC_PKCS5padding_decrypt(
            Data,
            dataSecret=pullDataSecret,
            dataSecretIV=pullDataSecretIV
        ))
    except Exception as e:
        logger.exception(e)
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1004）"})

    # 分页以及查询参数
    pageNo = int(data.get('PageNo', 1))
    pageSize = int(data.get('PageSize', 10))
    lastQueryTime = data.get("LastQueryTime")

    # 查找出所有符合条件的地质组的信息
    query = Group.objects.filter(ownerId__in=dealerIds)
    if lastQueryTime:
        dateTime = to_datetime(lastQueryTime)
        query = query.filter(dateTimeAdded__gte=dateTime)

    # 过滤出有设备的地址组
    StationInfos = list()
    groupIds = [str(item.id) for item in query.only("id") if Device.objects.filter(groupId = str(item.id)).count()]
    groupQuery = Group.objects.filter(id__in=groupIds).only("id").paginate(pageNo, pageSize)
    for item in groupQuery:
        # 获取站点信息
        groupId = str(item.id)

        # 我们的GroupID是24位，省平台的是20位，尝试添加映射
        GroupIdMap.add(groupId)
        StationInfos.append(get_stations_info(Group.get_group(item.id), norther))

    result = {
        "PageNo": pageNo,
        "ItemSize": pageSize,
        "PageCount": groupQuery.count(),
        "StationInfos": StationInfos
    }
    logger.debug("[queryStationsInfo] return result = {}".format(result))

    resultData = AES_CBC_PKCS5padding_encrypt(
        json.dumps(result),
        dataSecret=pullDataSecret,
        dataSecretIV=pullDataSecretIV
    )
    sig = norther.get_sig(resultData)
    logger.debug("groupId = {}".format(groupIds[0]))
    return JsonResponse({
        "Ret": 0,
        "Msg": u"请求成功",
        "Data": resultData,
        "Sig": sig
    })

def queryStationStats(request):
    """
    取每个充电站在某个周期内的统计信息
    :param request:
    :return:
    """
    token = request.META.get('HTTP_AUTHORIZATION', "").replace("Bearer", "").strip()
    logger.info('[queryStationsInfo] , token = {}'.format(token))

    # 验证身份
    tokenData = parse_json_token(token)
    if not tokenData:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_TOKEN, "Msg": u"请求参数错误（1001）"})
    northers = ShanDongNorther.get_norther(**tokenData)
    if not northers:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_PARAM, "Msg": u"请求参数错误（1002）"})

    # 准备token所获取的参数信息
    norther = northers.first()
    pullDataSecret = norther.pullDataSecret
    pullDataSecretIV = norther.pullDataSecretIV

    # 验证参数
    logger.debug("[queryStationsInfo] request body = {}".format(request.body))

    if not request.body:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1003）"})

    try:
        Data = json.loads(request.body).get("Data")
        data = json.loads(AES_CBC_PKCS5padding_decrypt(
            Data,
            dataSecret=pullDataSecret,
            dataSecretIV=pullDataSecretIV
        ))
    except Exception as e:
        logger.exception(e)
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1004）"})

    stationID = data.get("StationID")
    startTime = data.get("StartTime")
    endTime = data.get("EndTime")

    startTimeObj = datetime.datetime.strptime(startTime, "%Y-%m-%d")
    endTimeObj = datetime.datetime.strptime(endTime, "%Y-%m-%d")

    res = get_get_station_state(stationID,startTimeObj, endTimeObj)

    res.update({
        "StationID": stationID,
        "StartTime": startTime,
        "EndTime": endTime
    })

    result = {"StationStats": res}
    resultData = AES_CBC_PKCS5padding_encrypt(
        json.dumps(result),
        dataSecret=pullDataSecret,
        dataSecretIV=pullDataSecretIV
    )
    sig = norther.get_sig(resultData)
    logger.debug("[queryStationStats] return result = {}".format(result))

    return JsonResponse({
        "Ret": 0,
        "Msg": u"请求成功",
        "Data": resultData,
        "Sig": sig
    })

def queryStationStatus(request):
    token = request.META.get('HTTP_AUTHORIZATION', "").replace("Bearer", "").strip()
    logger.info('[queryStationsInfo] , token = {}'.format(token))

    # 验证身份
    tokenData = parse_json_token(token)
    if not tokenData:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_TOKEN, "Msg": u"请求参数错误（1001）"})
    northers = ShanDongNorther.get_norther(**tokenData)
    if not northers:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_PARAM, "Msg": u"请求参数错误（1002）"})

    # 准备token所获取的参数信息
    norther = northers.first()
    pullDataSecret = norther.pullDataSecret
    pullDataSecretIV = norther.pullDataSecretIV

    # 验证参数
    logger.debug("[queryStationsInfo] request body = {}".format(request.body))

    if not request.body:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1003）"})

    try:
        Data = json.loads(request.body).get("Data")
        data = json.loads(AES_CBC_PKCS5padding_decrypt(
            Data,
            dataSecret=pullDataSecret,
            dataSecretIV=pullDataSecretIV
        ))
    except Exception as e:
        logger.exception(e)
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1004）"})

    stationIDs = data.get("StationIDs")
    if not isinstance(stationIDs, list):
        return JsonResponse({"Ret": 4004, "Msg": u"系统错误"})

    StationStatusInfos = list()
    for stationID in stationIDs:
        groupId = GroupIdMap.get_groupId(stationID)

        StationStatusInfos.append(
            {
                "StationID": stationID,
                "ConnectorStatusInfos": get_station_status(groupId)
            }
        )

    result = {
        "StationStatusInfos": StationStatusInfos
    }
    resultData = AES_CBC_PKCS5padding_encrypt(
        json.dumps(result),
        dataSecret=pullDataSecret,
        dataSecretIV=pullDataSecretIV
    )
    sig = norther.get_sig(resultData)
    logger.debug("[queryStationStatus] return result = {}".format(result))

    return JsonResponse({
        "Ret": 0,
        "Msg": u"请求成功",
        "Data": resultData,
        "Sig": sig
    })

def queryEquipBusinessPolicy(request):
    """
    用于查询运营商的充电设备接口计费模型信息
    :param request:
    :return:
    """
    token = request.META.get('HTTP_AUTHORIZATION', "").replace("Bearer", "").strip()
    logger.info('[queryStationsInfo] , token = {}'.format(token))

    # 验证身份
    tokenData = parse_json_token(token)
    if not tokenData:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_TOKEN, "Msg": u"请求参数错误（1001）"})
    northers = ShanDongNorther.get_norther(**tokenData)
    if not northers:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_PARAM, "Msg": u"请求参数错误（1002）"})

    # 准备token所获取的参数信息
    norther = northers.first()
    pullDataSecret = norther.pullDataSecret
    pullDataSecretIV = norther.pullDataSecretIV

    # 验证参数
    logger.debug("[queryStationsInfo] request body = {}".format(request.body))

    if not request.body:
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1003）"})

    try:
        Data = json.loads(request.body).get("Data")
        data = json.loads(AES_CBC_PKCS5padding_decrypt(
            Data,
            dataSecret=pullDataSecret,
            dataSecretIV=pullDataSecretIV
        ))
    except Exception as e:
        logger.exception(e)
        return JsonResponse({"Ret": RESPONSE_CODE.ERROR_POST, "Msg": u"请求参数错误（1004）"})

    equipBizSeq = data.get("EquipBizSeq")
    connectorID = data.get("ConnectorID")

    policyInfo = get_policy_info(connectorID)

    result = {
        "EquipBizSeq": equipBizSeq,
        "ConnectorID": connectorID,
        "SuccStat": 0 if policyInfo else 1,
        "FailReason": 0 if policyInfo else 1,
        "SumPeriod": 1,
        "PolicyInfos": [policyInfo] if policyInfo else "-"
    }
    resultData = AES_CBC_PKCS5padding_encrypt(
        json.dumps(result),
        dataSecret=pullDataSecret,
        dataSecretIV=pullDataSecretIV
    )
    sig = norther.get_sig(resultData)

    logger.debug("[queryEquipBusinessPolicy] return result = {}".format(result))

    return JsonResponse({
        "Ret": 0,
        "Msg": u"请求成功",
        "Data": resultData,
        'Sig':sig
    })