#!/usr/bin/python
# -*- coding: utf-8 -*-
# @todo: CTGReportAll 充值分析报告所有汇总

import CommFunc
import ConfigParser

import logging
import mylog
import json

# ==================== 配置 ====================
cfg = ConfigParser.ConfigParser()
cfg.read("../../InterfaceConfig.php")
ServerPath = cfg.get("ServerInfo", "ServerPath")
# 需要处理的流向名及顺序列表
DRNameList = ["CTGOK"]
# ==================== 配置 ====================


def getCTGReportInfo(argvDict):
    logging.info("getCTGReportInfo %s" % argvDict)
    startDate = argvDict.get("startDate", "")
    endDate = argvDict.get("endDate", "")
    payOrderTypeGroup = argvDict.get("payOrderTypeGroup", "")

    CTGReportOrderTypeList = eval(CommFunc.getSPConfig(
        argvDict, "Config", "CTGReportOrderType", "[1]"))

    reportInfo = {} # {key:{orderInfo:{k:v, ...}, ...}, ...}
    # 查询中心备份的
    if CommFunc.isQueryCenterbak(argvDict):
        queryCenterBak(startDate, endDate, argvDict, CTGReportOrderTypeList)
        return

    needQueryCenterbak = CommFunc.loopMainServerDR(cfg, startDate, endDate, argvDict, checkDrFileNeedParseFunc,
                                                   parseLineFunc, reportInfo, CTGReportOrderTypeList, payOrderTypeGroup, drNameList=DRNameList)

    # logging.info("1 reportInfo %s" % reportInfo)
    logging.info("needQueryCenterbak %s" % needQueryCenterbak)
    if needQueryCenterbak:
        retBak = CommFunc.queryBackupCenterDR(cfg, argvDict)
        if retBak == None:
            return
        reportInfoBak = retBak[0]
        # logging.info("reportInfoBak %s" % reportInfoBak)
        for key, orderInfoDictBak in reportInfoBak.items():
            if key not in reportInfo:
                reportInfo[key] = orderInfoDictBak
                continue
            orderInfoDict = reportInfo[key]
            for orderInfo, bakData in orderInfoDictBak.items():
                orderInfo = str(orderInfo)
                if orderInfo not in orderInfoDict:
                    orderInfoDict[orderInfo] = bakData
                else:
                    retData = orderInfoDict[orderInfo]
                    for k, v in bakData.items():
                        if k == "orderCoin":
                            continue;
                        retData[k] = retData.get(k, 0) + v
    # logging.info("2 reportInfo %s" % reportInfo)
    ret = {"OK": 1, "reportInfo": reportInfo}
    print json.dumps(ret, ensure_ascii=False, default=lambda obj: obj.__dict__)
    return


def queryCenterBak(startDate, endDate, argvDict, CTGReportOrderTypeList):
    # 查询中心服务器备份充值信息
    reportInfoBak = {}
    payOrderTypeGroup = argvDict.get("payOrderTypeGroup", "")
    if not CommFunc.loopCenterbakRarDR(cfg, startDate, endDate, argvDict, checkDrFileNeedParseFunc,
                                       parseLineFunc, reportInfoBak, CTGReportOrderTypeList, payOrderTypeGroup, drNameList=DRNameList):
        return
    return CommFunc.queryBackupCenterOK([reportInfoBak])


def checkDrFileNeedParseFunc(drFileName, *parseArgs, **kv):
    ''' 检查流向是否需要处理
    @param drFileName: 流向文件名  xxx_日期.txt
    @param *parseArgs: 自定义参数
    @return: isNeed, checkNeedParseRetInfo
    '''
    # 因为指定了 drNameList , 所以默认返回True
    return True, None


def parseLineFunc(drName, dateStr, checkNeedParseRetInfo, line, *parseArgs, **kv):
    ''' 解析流向行内容
    @param drName: 流向名
    @param dateStr: 对应日期字符串
    @param checkNeedParseRetInfo: checkDrFileNeedParseFunc 返回的内容
    @param line: 本行内容
    @param *parseArgs: 自定义参数
    '''

    reportInfo, CTGReportOrderTypeList, payOrderTypeGroup = parseArgs

    drDict = eval(line)

    payOrderType = drDict.get("payOrderType", 1)
    if payOrderType not in CTGReportOrderTypeList:
        return

    if "orderInfo" not in drDict:
        return

    orderInfo = drDict["orderInfo"]
    orderCoin = drDict.get("orderCoin", 0)
    key = str(payOrderType) if payOrderTypeGroup else ""
    if key not in reportInfo:
        reportInfo[key] = {}
    orderInfoDict = reportInfo[key]
    if orderInfo not in orderInfoDict:
        orderInfoDict[orderInfo] = {}
    retData = orderInfoDict[orderInfo]
    retData["orderCoin"] = orderCoin
    retData["payCount"] = retData.get("payCount", 0) + 1
    payOrderTypeKey = "payOrderType%s" % payOrderType
    retData[payOrderTypeKey] = retData.get(payOrderTypeKey, 0) + 1
    return

# =================================================================================================


def main():
    CommFunc.setdefaultencoding()
    argvDict = CommFunc.parse_args()
    mylog.InitMyLog(argvDict.get("eventType", ""))
    CommFunc.gettextInstall(argvDict.get("lang", ""))
    getCTGReportInfo(argvDict)
    return


if __name__ == "__main__":
    try:
        main()
    except:
        CommFunc.printExceptionError()