read_pdf_plt/parseCreditXml0107.py

#coding=utf-8
from xml.dom import minidom
import base64
import sys
import os
from ini_op import Config;
base_dir = os.path.dirname(os.path.abspath(__file__))
config = Config(base_dir+"/config.ini");
productNumXy = config.get("baseconf", "productNumXy")
productNumJz = config.get("baseconf","productNumJz")
productNumXxw = config.get("baseconf","productNumXxw")
productNumFb = config.get("baseconf", "productNumFb")
productNumKcd = config.get("baseconf", "productNumKcd")

productNumFd = config.get("baseconf", "productNumFd")
productNumKn = config.get("baseconf", "productNumKn")
productNumHst = config.get("baseconf", "productNumHst")
productNumQk = config.get("baseconf", "productNumQk")
productNum500 = config.get("baseconf", "productNum500")
productNumHsz = config.get("baseconf", "productNumHsz")
isPlt = config.get("baseconf", "isPlt");

import xyHttp
import hst_rule_set
import log
import shutil
logger = log.logger
import time
from dbController import DbController
dbController = DbController();
import json
import requests
import timeit
import traceback
import gzip
import io
from xmlParser import XmlParser;
from xmlParserHsz import XmlParserHsz;
from pboc.invokePboc import PBOC
from mailUtil import MailUtil

def gzip_str(string_):
    out = io.BytesIO()

    with gzip.GzipFile(fileobj=out, mode='w') as fo:
        fo.write(string_.encode())

    bytes_obj = out.getvalue()
    return bytes_obj


def gunzip_bytes_obj(bytes_obj):
    in_ = io.BytesIO()
    in_.write(bytes_obj)
    in_.seek(0)
    with gzip.GzipFile(fileobj=in_, mode='rb') as fo:
        gunzipped_bytes_obj = fo.read()

    return gunzipped_bytes_obj.decode()


#解析xml数据
def getBusinessInfo(xmlFile):
    doc = minidom.parse(xmlFile)
    request = doc.documentElement.getElementsByTagName("request")[0]
    responseBody = doc.documentElement.getElementsByTagName("responseBody")[0]
    responseHeader = doc.documentElement.getElementsByTagName("responseHeader")[0]
    result = ""
    xmlData = ""
    isBaihu = "0";

    if len(responseBody.childNodes)==0:
        mailUtil = MailUtil();
        webhook = 'https://oapi.dingtalk.com/robot/send?access_token=64d8b2c7fed4949e9433b807c7c5559939f1517af8f77c1dacb4de19c6910b56'
        if getNodeData(responseHeader,"resultMsg")=="查询成功，无报告":
            isBaihu = "1"
        else:
            isBaihu = "2"#查询失败
            mailUtil.dingtalk("号码:" + xmlFile.split("_")[0] + " 查询失败 ", webhook)
    else:
        responseBodyText = responseBody.childNodes[0].data
        decrpyt_bytes = base64.b64decode(responseBodyText)
        xmlData = gunzip_bytes_obj(decrpyt_bytes)
        # print(xmlData)
        #remove by chenqiwang 1105
        xmlPath = xmlFile+".txt"
        # with open(xmlPath, 'w', encoding='utf-8') as fp:
        #     fp.write(xmlData)
    # 非小赢需要解析
    productNum = getNodeData(request, "productNum")
    if productNum != productNumXy:
        if xmlData !="":
            if productNum == productNumHsz:
                result = parseHsz(xmlData)
            else:
                #跑数据的时候改为parseHsz
                result = parse(xmlData)
    data = {
        "businessNum":getNodeData(request,"businessNum"),
        "coopBusinessNum": getNodeData(request, "coopBusinessNum"),
        "customerNum":getNodeData(request,"customerNum"),
        "certificateNum": getNodeData(request, "certificateNum"),
        "productNum": productNum,
        "creditXml":xmlData,
        "extend": getNodeData(request, "extend"),
        "result":result,
        "isBaihu":isBaihu
    }
    return data
#解析xml报文
def parse(xmlData):
    xmlParse = XmlParser()
    result = ""
    try:
        result = xmlParse.parse(xmlData)
    except:
        info = sys.exc_info()
        logger.error(info[0])
        logger.error(info[1])
        logger.error(traceback.extract_tb(info[2], 1))
    return result
#解析xml报文 汇算帐
def parseHsz(xmlData):
    xmlParseHsz = XmlParserHsz()
    result = ""
    try:
        result = xmlParseHsz.parse(xmlData)
    except:
        info = sys.exc_info()
        logger.error(info[0])
        logger.error(info[1])
        logger.error(traceback.extract_tb(info[2], 1))
    return result
def getNodeData(request,key):
    data = request.getElementsByTagName(key)[0].childNodes[0].data
    return data;
# if __name__ == '__main__':
    # businessInfo = getBusinessInfo('./test.xml')
    # print(businessInfo)
    # xml = ""
    # data = base64.b64encode(xml.encode("UTF-8"))
    # print(str(data))
#调用http
def process(businessInfo,basePath,xml_path):
    productNum = businessInfo["productNum"]
    if productNumXy.find(productNum) >= 0:
        try:
            result = xyHttp.call_credit(businessInfo)
            # result = json.loads(result);
            # logger.info(result)
            #上传审批结果
            jsonPath = basePath + businessInfo["certificateNum"] + ".txt"
            logger.info(jsonPath)
            with open(jsonPath, 'w', encoding='utf-8') as fp:
                fp.write(result)
            uploadJsonFile(businessInfo,jsonPath)
            #移动xml文件
            try:
                dayStrPath = basePath+productNum+"/"+time.strftime('%Y%m%d', time.localtime(time.time()))+"/";
                descXmlPath = dayStrPath +"xml/"+ os.path.basename(xml_path)
                descJsonPath = dayStrPath+"txt/" + os.path.basename(jsonPath)
                if not os.path.exists(basePath+productNum):
                    os.mkdir(basePath+productNum)
                if not os.path.exists(dayStrPath):
                    os.mkdir(dayStrPath)
                if not os.path.exists(dayStrPath+"xml/"):
                    os.mkdir(dayStrPath + "xml/")
                if not os.path.exists(dayStrPath+"txt/"):
                    os.mkdir(dayStrPath+"txt/")
                logger.info("移动文件 from " + xml_path + " to " + descXmlPath)
                if isPlt == "1":
                    move(xml_path, descXmlPath)
                    shutil.move(jsonPath, descJsonPath)
            except:
                logger.error(businessInfo["certificateNum"]+"#创建文件夹或者移动文件失败，不影响业务")
                logger.error(traceback.print_exc())
                logger.error(traceback.format_exc())
        except:
            logger.error(traceback.print_exc())
            logger.error(traceback.format_exc())
            mailUtil = MailUtil();
            webhook = 'https://oapi.dingtalk.com/robot/send?access_token=64d8b2c7fed4949e9433b807c7c5559939f1517af8f77c1dacb4de19c6910b56'
            mailUtil.dingtalk("号码:" + businessInfo["certificateNum"] + " 调用xy服务失败 ", webhook)
    elif productNumQk.find(productNum) >= 0 or productNumHsz.find(productNum) >= 0:#洽客
        txtPath = xml_path.replace(".xml", ".txt")
        with open(txtPath, 'w', encoding='utf-8') as fp:
            fp.write(businessInfo["result"])
        if productNumQk.find(productNum) >= 0:
            result = xyHttp.callQk(businessInfo)
        elif productNumHsz.find(productNum) >= 0:
            result = xyHttp.callHsz(businessInfo)
        # result = json.loads(result);
        logger.info(result)
        #上传审批结果
        jsonPath = basePath + businessInfo["certificateNum"] + ".txt"
        logger.info(jsonPath)
        with open(jsonPath, 'w', encoding='utf-8') as fp:
            fp.write(result)
        uploadJsonFile(businessInfo,jsonPath)
        #移动xml文件
        dayStrPath = basePath+productNum+"/"+time.strftime('%Y%m%d', time.localtime(time.time()))+"/";
        descXmlPath = dayStrPath +"xml/"+ os.path.basename(xml_path)
        descJsonPath = dayStrPath+"txt/" + os.path.basename(jsonPath)
        descTxtPath = dayStrPath + "txt/" + os.path.basename(txtPath)
        if not os.path.exists(basePath+productNum):
            os.mkdir(basePath+productNum)
        if not os.path.exists(dayStrPath):
            os.mkdir(dayStrPath)
        if not os.path.exists(dayStrPath+"xml/"):
            os.mkdir(dayStrPath + "xml/")
        if not os.path.exists(dayStrPath+"txt/"):
            os.mkdir(dayStrPath+"txt/")
        logger.info("移动文件 from " + xml_path + " to " + descXmlPath)
        if isPlt == "1":
            move(xml_path, descXmlPath)
            shutil.move(jsonPath, descJsonPath)
            shutil.move(txtPath, descTxtPath)
    else:
        #走jar包逻辑
        # print(businessInfo["result"])
        txtPath = xml_path.replace(".xml",".txt")
        with open(txtPath, 'w', encoding='utf-8') as fp:
            fp.write(businessInfo["result"])
        if productNum == productNumJz:#桔子
            invokePboc(businessInfo, txtPath)
            #移动xml文件
            descXmlPath = basePath + "execed_new/" + os.path.basename(xml_path)
            if isPlt == "1":
                move(xml_path, descXmlPath)
        elif productNum == productNumXxw:#新希望
            invokeXxw(businessInfo, txtPath);
            jsonPath = basePath + businessInfo["certificateNum"] + ".txt"    
            #移动xml文件
            descXmlPath = basePath + "execed_new/" + os.path.basename(xml_path)
            if not os.path.exists(basePath + "execed_new/"):
                os.mkdir(basePath + "execed_new/")
            logger.info("移动文件 from " + xml_path + " to " + descXmlPath)
            if isPlt == "1":
                move(xml_path, descXmlPath)
        elif productNum == productNumFb or productNum== productNumKcd or productNum500.find(productNum)>=0:#没有java包 #快车道和风暴 上传500个字段
            logger.info(businessInfo["productNum"])
            uploadJsonFile(businessInfo,txtPath)
            # 移动xml文件
            descXmlPath = basePath + "execed_new/" + os.path.basename(xml_path)
            if not os.path.exists(basePath + "execed_new/"):
                os.mkdir(basePath + "execed_new/")
            logger.info("移动文件 from " + xml_path + " to " + descXmlPath)
            if isPlt == "1":
                move(xml_path, descXmlPath)
        elif productNum == productNumFd:#分蛋，上传解析后的xml
            uploadJsonFile(businessInfo, xml_path)#xml_path+".txt" #暂时没有业务取消生产xml文件
            # 移动xml文件
            descXmlPath = basePath + "execed_new/" + os.path.basename(xml_path)
            if not os.path.exists(basePath + "execed_new/"):
                os.mkdir(basePath + "execed_new/")
            logger.info("移动文件 from " + xml_path + " to " + descXmlPath)
            if isPlt == "1":
                move(xml_path, descXmlPath)
        elif productNum == productNumKn:#快牛本地jar包
            if businessInfo["result"]!="":
                result = xyHttp.callLocal(businessInfo)
                uploadAudit(result,businessInfo["businessNum"],productNum)
            else:
                result = {"approveResult":"0","rule":"白户"}
                uploadAudit(result, businessInfo["businessNum"], productNum)
            # 移动xml文件
            descXmlPath = basePath + "execed_new/" + os.path.basename(xml_path)
            if not os.path.exists(basePath + "execed_new/"):
                os.mkdir(basePath + "execed_new/")
            logger.info("移动文件 from " + xml_path + " to " + descXmlPath)
            if isPlt == "1":
                move(xml_path, descXmlPath)
        elif productNum == productNumHst:#海盛通
            if businessInfo["result"]!="":
                result = hst_rule_set.callLocal(businessInfo)
                uploadAudit(result,businessInfo["businessNum"],productNum)
            else:
                result = {"approveResult":"0","rule":"白户"}
                uploadAudit(result, businessInfo["businessNum"], productNum)
            # 移动xml文件
            descXmlPath = basePath + "execed_new/" + os.path.basename(xml_path)
            if not os.path.exists(basePath + "execed_new/"):
                os.mkdir(basePath + "execed_new/")
            logger.info("移动文件 from " + xml_path + " to " + descXmlPath)
            if isPlt == "1":
                move(xml_path, descXmlPath)
        else:
            i=0
            #本地jar
        if isPlt == "1":
            if not os.path.exists(basePath + "execed_txt/"):
                os.mkdir(basePath + "execed_txt/")
            descTxtPath = basePath + "execed_txt/" + os.path.basename(txtPath)
            shutil.move(txtPath, descTxtPath)
#调用jar包
def move(xml_path, descXmlPath):
    try:
        shutil.move(xml_path, descXmlPath)
        if not os.path.exists(basePath + "execed_xml/"):
            os.mkdir(basePath + "execed_xml/")
        shutil.move(xml_path+".txt", descXmlPath.replace("execed_new","execed_xml")+".txt")
        # os.remove(xml_path+".txt")#删除生成的xml.txt文件
    except:
        logger.info("移动文件 from " + xml_path + " to " + descXmlPath +" error")
def invokePboc(businessInfo,txtPath):
    # ===================================
    try:
        businessNum = businessInfo["businessNum"]
        coopBusinessNum = businessInfo["coopBusinessNum"]
        pboc = PBOC()
        jarTxt = pboc.calc(txtPath,coopBusinessNum);
        result = json.loads(jarTxt)
        logger.info(result)
        if result.get("errcode")== None:
            uploadAudit(result,businessNum,businessInfo["productNum"])
        else:
            if businessInfo["isBaihu"]=="1":
                result = {"approveResult":"1"}
                uploadAudit(result, businessNum, businessInfo["productNum"])
            logger.error(result["errmsg"])
    except:
        logger.error("certificateNum=" + businessInfo["coopBusinessNum"])
        info = sys.exc_info()
        logger.error(info[0])
        logger.error(info[1])
        # logging.log(logging.ERROR, info[2])
        logger.error(traceback.extract_tb(info[2], 1))
#调用xxwjar包
def invokeXxw(businessInfo,txtPath):
    try:
        businessNum = businessInfo["businessNum"]
        coopBusinessNum = businessInfo["coopBusinessNum"]
        customerNum = businessInfo["customerNum"]
        certificateNum =  businessInfo["certificateNum"]
        pboc = PBOC()
        jarTxt = pboc.calcXxw(coopBusinessNum,customerNum,txtPath);
        logger.info("jarTxt:"+jarTxt)
        result = json.loads(jarTxt)
        jsonPath = basePath+certificateNum+".txt"
        # logger.info(jsonPath)
        with open(jsonPath, 'w', encoding='utf-8') as fp:
            fp.write(jarTxt)
        uploadJsonFile(businessInfo,jsonPath)
        if not os.path.exists(basePath + "execed_txt/"):
            os.mkdir(basePath + "execed_txt/")
        descJsonPath = basePath + "execed_txt/" + os.path.basename(jsonPath)
        shutil.move(jsonPath, descJsonPath)
    except:
        logger.error("certificateNum=" + businessInfo["coopBusinessNum"])
        info = sys.exc_info()
        logger.error(info[0])
        logger.error(info[1])
        logger.error(traceback.extract_tb(info[2], 1))
#上传审批结果
def uploadAudit(result,businessNum,productNum):
    approvalType = result["approveResult"]
    if productNum == productNumJz:
        if approvalType=="1":
            approvalOpinion = "征信通过"
            approvalType = "4"
        else:
            approvalOpinion = "征信拒绝"
            approvalType = "3"
    elif productNum == productNumKn or productNum == productNumHst:
        if approvalType=="1":
            approvalOpinion = "征信通过"
            approvalType = "4"
        else:#快牛 需要添加拒绝原因
            approvalOpinion = "征信拒绝" + "#" + result["rule"]
            approvalType = "3"
    taskKey = config.get("baseconf","taskKey")
    appoveApiUrl = config.get("baseconf","appoveApiUrl")
    key = config.get("baseconf", "AESKey")
    data = {"header":{
            "ticket": "2938123198320412343",
            "timestamp": int(int(round(time.time() * 1000+60*1000))),
            "nonce": config.get("baseconf", "nonce")
        },
       "body":{"approvalType": approvalType, "businessNum": businessNum,"taskKey":taskKey,"approvalOpinion":approvalOpinion}}
    access_token = dbController.getToken();
    if access_token == "":#获取token失败重新获取
        access_token = dbController.getToken();
    appoveApiUrl = appoveApiUrl+"?access_token="+access_token
    headers = {"Content-Type": "application/json"}
    jsonStr = json.dumps(data);
    jsonStr = jsonStr.replace('"',"\\\"")#必须替换才行
    logger.info(jsonStr)
    pboc = PBOC();
    encryData = pboc.encrypt(jsonStr,key)
    encryData = encryData[0:len(encryData)-2]
    logger.info(encryData)
    response = requests.post(appoveApiUrl, data=encryData,headers=headers)
    text = response.text
    pboc = PBOC();
    resultText = pboc.decrypt(text, config.get("baseconf", "AESKey"))
    logger.info(businessNum + "#" + "uploadAudit upload_result:" + resultText)
#上传解析json
def uploadJsonFile(businessInfo,json_path):
    # ===================================
    try:
        fileName = os.path.basename(json_path)
        #上传文件逻辑
        logger.info(json_path+"#"+"准备上传文件")
        uploadApiUrl = config.get("baseconf", "uploadApiUrl");
        token = dbController.getToken();
        if token == "":#如果获取失败，重新获取
            token = dbController.getToken();
        uploadApiUrl = uploadApiUrl + "?access_token=" + token
        files = {'file': open(json_path, 'rb')}
        logger.info(fileName+"#"+"businessNum:"+businessInfo["businessNum"])
        data = {'docType': "23", 'businessNum': businessInfo["businessNum"]}
        response = requests.post(uploadApiUrl, files=files, data=data,timeout=15)
        text = response.text
        logger.info("上传结果:"+text)
        pboc = PBOC();
        resultText = pboc.decrypt(text,config.get("baseconf", "AESKey"))
        logger.info(fileName+"#"+"uploadJsonFile:" + resultText)
    except:
        logger.error("certificateNum="+businessInfo["coopBusinessNum"])
        info = sys.exc_info()
        logger.error(info[0])
        logger.error(info[1])
        # logging.log(logging.ERROR, info[2])
        logger.error(traceback.extract_tb(info[2], 1))
        mailUtil = MailUtil();
        webhook = 'https://oapi.dingtalk.com/robot/send?access_token=64d8b2c7fed4949e9433b807c7c5559939f1517af8f77c1dacb4de19c6910b56'
        mailUtil.dingtalk("号码:" + businessInfo["certificateNum"] + " 上传文件失败 ", webhook)
if __name__ == '__main__':
    file_name = ""
    # basePath = "D:/mydocument/myproject/git/parse/"
    basePath = "D:/mydocument/myprojects/creditreport/parse/"

    file_name = "黄振武_420115199306180091_230034699446131714.xml"

    file_name = "周平强_360122197901070610_255387334772537344.xml"
    file_name = "lwl_450803199507215834_254603000272664585.xml"
    # file_name="dqy_320821199205184730_255559091387119623.xml"
    file_name='zl_342425199204087911_255220233202055177.xml'
    file_name="sxg_330719197909242017_257564808331673603.xml"
    file_name="pxy_522401199510147022_258623031256039426.xml"
    file_name = "任赞_130185199601221319_257538384191307779.xml"
    # file_name="hrh_430722198805068479_255677390955493378.xml"
    # file_name="ll_310104196604260518_257161903451684870.xml"
    file_name = "1_210303197510273014_269873665141647369-202012091702117200.xml"
    file_name="段王爷_340621198906093610_271693595784776712.xml"
    file_name="刘溜_110101198309193765_272050857560771588.xml"
    xml_path = basePath + file_name
    start = timeit.default_timer();
    if len(sys.argv) > 1:
        basePath = sys.argv[1]
        xml_path = basePath + sys.argv[2]
        file_name = sys.argv[2]
    
    logger.info(xml_path+" 解析开始")
    businessInfo = getBusinessInfo(xml_path)
    # logger.info(businessInfo)
    if businessInfo["isBaihu"]!="2":#白户不上传1211
        process(businessInfo,basePath,xml_path)
    s = timeit.default_timer() - start;
    logger.info(str(s) + " 秒")
    logger.info(xml_path+" 解析完成")