/* * @Descripttion: 导入通用代码 * @Author: vian * @Date: 2020-09-09 10:45:54 */ const INTERFACE_EXPORT_BASE = (() => { // xml字符实体 const XMLEntity = { ' ': 'escape{space}', ' ': 'escape{simpleSpace}', ' ': 'escape{tab}', ' ': 'escape{simpleTab}', ' ': 'escape{return}', ' ': 'escape{simpleReturn}', '�A;': 'escape{newLine}', ' ': 'escape{simpleNewLine}', '<': 'escape{less}', '>': 'escape{greater}', '&': 'escape{and}', '"': 'escape{quot}', ''': 'escape{apos}' }; // 避免字符实体进行转义。原文本中含有xml字符实体,转换为其他字符。 function escapeXMLEntity(str) { for (const [key, value] of Object.entries(XMLEntity)) { str = str.replace(new RegExp(key, 'g'), value); } return str; } // 将文本还原为字符实体 function restoreXMLEntity(str) { for (const [key, value] of Object.entries(XMLEntity)) { str = str.replace(new RegExp(value, 'g'), key); } return str; } /* * 根据字段数组获得所要字段的值 eg: 要获取标段下的单项工程: ['标段', '单项工程']; * @param {Object}source 源数据 * {Array}fields 字段数组 * @return {String} * @example getValue(source, ['标段', '_文件类型']) * */ function getValue(source, fields) { let cur = source; for (const field of fields) { if (!cur[field]) { return ''; } cur = cur[field]; } return cur || ''; } // 获取数据类型 function _plainType(v) { return Object.prototype.toString.call(v).slice(8, -1); } /* * 获取某字段的值,强制返回数组,防止一些错误。如果期待返回数组,可以用此方法。 * @param {Object}source 数据源 * {Array}fields 取的字段 * @return {Array} * @example arrayValue(source, ['标段', '单项工程']) * */ function arrayValue(source, fields) { let target = getValue(source, fields); if (_plainType(target) === 'Object') { target = [target]; } else if (_plainType(target) !== 'Array') { target = [] } return target; } // 获取费用 function getFee(fees, fields) { if (!Array.isArray(fees) || !fees.length) { return '0'; } const feeData = fees.find(fee => fee.fieldName === fields[0]); return feeData && feeData[fields[1]] || '0'; } // 合并价格 function mergeFees(feesA, feesB) { if (!feesA) { return feesB; } if (!feesB) { return []; } feesB.forEach(feeB => { const sameKindFee = feesA.find(feeA => feeA.fieldName === feeB.fieldName); if (sameKindFee) { Object.assign(sameKindFee, feeB); } else { feesA.push(feeB); } }); return feesA; } // 将A对象的一部分属性赋值到B对象上 function assignAttr(target, source, attrs, isExcepted = false) { if (!source || !target) { return; } const sourceAttrs = attrs ? isExcepted ? Object.keys(source).filter(attr => !attrs.includes(attr)) : attrs : Object.keys(source); sourceAttrs.forEach(attr => { // 如果是价格,不能简单地覆盖,要合并两个对象的价格 target[attr] = attr === 'fees' ? mergeFees(target[attr], source[attr]) : source[attr]; }); } // 获取固定ID function getFlag(data) { return data.flags && data.flags[0] && data.flags[0].flag || 0; } // 获取布尔型的数据 function getBool(v) { return v === 'true' ? true : false; } // 设置成树结构数据 function setTreeData(data, parent, next) { const defalutID = -1; data.ID = uuid.v1(); data.ParentID = parent && parent.ID || defalutID; data.NextSiblingID = next && next.ID || defalutID; } // 递归设置树结构数据,并返回设置好的数据,递归items数组 function mergeDataRecur(parent, items) { const rst = []; for (let i = 0; i < items.length; i++) { const cur = items[i]; const next = items[i + 1]; setTreeData(cur, parent, next); rst.push(cur); if (cur.items && cur.items.length) { rst.push(...mergeDataRecur(cur, cur.items)); } } return rst; } /* * 递归获取相关数据,eg:获取组织措施清单下的所有子清单,该子清单们可能由分类及公式措施项各种组合组成。(参考调用处) * @param {Object}src(数据源) {Array}fields(二维数组,数组里的成员由需要取的字段组成) * eg: ['组织措施分类'], ['公式计算措施项'],该层数据可能由组织措施分类 或 公式计算措施项组成 (同层不可同时存在) * {Function} 获得源数据后,需要提取的数据方法 * @return {Array} * */ function getItemsRecur(src, fields, extractFuc) { let itemsSrc = []; let curField = ['']; for (const field of fields) { itemsSrc = arrayValue(src, field); if (itemsSrc.length) { curField = field; break; } } return itemsSrc.map(itemSrc => { const obj = extractFuc(itemSrc, curField); obj.items = getItemsRecur(itemSrc, fields, extractFuc); return obj; }); } // 递归获取相关数据,与上方的方法不同的点在:同层可能出现不同节点,上方方法暂时不取消(防止bug) // fields内字段的顺序即决定了提取数据类型的顺序,如fields = [['gruop'], ['item']],则提取的数据同层中group数据在item数据之前 function extractItemsRecur(src, fields, extractFuc) { const rst = []; for (const field of fields) { const itemsSrc = arrayValue(src, field); if (itemsSrc.length) { const items = itemsSrc.map(itemSrc => { const obj = extractFuc(itemSrc, field); obj.items = extractItemsRecur(itemSrc, fields, extractFuc); return obj; }); rst.push(...items); } } return rst; } const UTIL = Object.freeze({ escapeXMLEntity, restoreXMLEntity, getValue, arrayValue, getFee, mergeFees, assignAttr, setTreeData, mergeDataRecur, getFlag, getBool, getItemsRecur, extractItemsRecur, }); /* * 读取文件转换为utf-8编码的字符串 * @param {Blob}file * @return {Promise} * */ function readAsTextSync(file) { return new Promise((resolve, reject) => { const fr = new FileReader(); fr.readAsText(file); // 默认utf-8,如果出现乱码,得看导入文件是什么编码 fr.onload = function () { resolve(this.result); }; fr.onerror = function () { reject('读取文件失败,请重试。'); } }); } /** * * @param {Function} entryFunc - 各导入接口提取导入数据方法 * @param {File} file - 导入的文件 * @param {String} areaKey - 地区标识,如:'安徽@马鞍山' * @param {String} feeRateStandard - 导入接口地区对应的费率标准名称 * @param {Boolean} escape - 是否需要避免xml中的实体字符转换 * @return {Promise} */ async function extractImportData(entryFunc, file, areaKey, feeRateStandard, escape = false) { const valuationID = compilationData.ration_valuation[0].id; const templateData = await ajaxPost('/pm/api/getImportTemplateData', { user_id: userID, valuationID, feeRateStandard }); if (!templateData) { throw '无法获取有效模板数据。'; } console.log(templateData); // 将二进制文件转换成字符串 let xmlStr = await readAsTextSync(file); if (escape) { // x2js的str to json的实现方式基于DOMParser,DOMParser会自动将一些实体字符进行转换,比如 “< to <”。如果不想进行自动转换,需要进行处理。 xmlStr = escapeXMLEntity(xmlStr); } // 将xml格式良好的字符串转换成对象 const x2js = new X2JS(); let xmlObj = x2js.xml_str2json(xmlStr); xmlObj = JSON.parse(restoreXMLEntity(JSON.stringify(xmlObj))); console.log(xmlObj); if (!xmlObj) { throw '无有效数据。'; } return await entryFunc(areaKey, xmlObj, templateData); } return { UTIL, extractImportData, } })();