项目链接:https://gitee.com/xyjtysk/quotationTools
在【QuotationTool】Model的实现(一),获得Excel路径以及Excel输出格式里面我们已经获得了Excel的路径,已经规定好了输出和输出有哪些列,下面就可以开始正式转换了。
预处理
由Controller进行调度
首先自然是读取Excel,我们在Controller里面调用XlrdTool中的getAssociativeArray
lists = XlrdTool().getAssociativeArray(inputPath, sheetName, inputParam.keys())
我们知道Controller其实是数据的中转站,所以其他Model处理以后的lists都要发到Controller中。
接下来就是调度rehandleModelClass.py进行预处理了
rehandleInstance = M("rehandle");
rehandleInstance.assign(lists);
lists = rehandleInstance.doRehandel(diffList);
那么我们来看一下rehandleModelClass.py是怎么实现的。
rehandleModelClass
这个Model主要是对读入的数组进行预处理,主要是
# 删除含有#的行
self.removeRows();
# 加行
self.addRow();
# 加colorTag
self.addColorTag();
# 加列
self.addColumns(diffList);
因为读入的Excel可能不规范,比如没有总计行或者小计行等,所以我们需要把这些行加上。
然后加上ColorTag
最后按照输出的keys把不存在的列加上。
removeRows
功能:删除不需要的行
我们在分析需求的时候就说过,官方的报价清单里面冗余太多,需要删除
那怎么删除呢?当然是遍历数组,对符合条件的删除呗。
这就有一个问题,删除以后iterator就改变了,所以最后的结果会乱七八糟
有什么办法可以解决吗?可以参考Python的list循环遍历中,删除数据的正确方法
def removeRows(self):
# 逆序遍历,否者一边删除一边iterator就改变了
for aList in self.lists[::-1]:
try:
if set([‘BOM‘,‘typeID‘,‘ID‘]) < set(aList.keys()) and str (aList[‘BOM‘]).find("#") != -1 and aList[‘ID‘] == "" and aList[‘typeID‘] == "":
self.lists.remove(aList);
info("删除了含有#的行");
elif ‘description‘ in aList.keys() and str(aList[‘description‘]).find(‘Factory integrated‘) != -1:
self.lists.remove(aList);
info("删除了含有Factory integrated行");
# 单独删除NHCT导出模板中的含截止日期行
elif ‘unitsNetPrice‘ in aList.keys() and str(aList[‘unitsNetPrice‘]).find(u‘截止日期‘) != -1:
self.lists.remove(aList);
info("删除了含有截止日期的行");
elif ‘ID‘ in aList.keys() and str(aList[‘ID‘]).find(u‘价格明细清单‘) != -1:
self.lists.remove(aList);
info("删除了含有价格明细清单的行");
# 删除空行,取出所有的values,通过map全部变为str类型,然后转换为list,最后串接在一起。
elif len("".join(list(map(str,aList.values())))) == 0 :
self.lists.remove(aList);
info("删除空行");
else:
continue;
except Exception as data:
error("删除空行时,超出表格范围"+str(data));
加行
因为输入的Excel可能不含有小计行等,我们需要再进行一次遍历,把该加上行的地方加上行。
def getRow (self , key , value):
# 先全部填上空白
row = {};
for k in self.lists[0].keys():
row[k] = "";
row[key] = value;
return row;
# **************加上小计行、总计行**************
def addRow(self):
# 遍历lists,插入小计行、总计行
try:
aDiff = [i for i in [‘BOM‘,‘typeID‘,‘description‘] if i in self.lists[0].keys()];
colTag = aDiff[0];
for i in range(len(self.lists) - 1 , 1 , -1):
list = self.lists[i];
if list[‘ID‘] != "" and self.lists[i-1][colTag] != ‘小计‘:
self.lists.insert(i,self.getRow(colTag,‘小计‘));
info (‘在第‘+str(i)+‘行增加了小计行‘)
if self.lists[-1][colTag] != ‘总计‘:
self.lists.append(self.getRow(colTag,‘总计‘));
info (‘在最后一行增加了总计行‘)
if self.lists[-2][colTag] != ‘小计‘:
self.lists.insert(len(self.lists)-1 , self.getRow(colTag,‘小计‘));
info (‘在倒数第二行增加了小计行‘)
except Exception as data:
error(data);
error ("addRow函数中")
加列
我们把要加的列放到inputVariable.py中的diff数组中
比如
diff = {
‘totalNum‘:‘0‘,
"unit":"个",
"billType":"增值税",
"taxRate":"17%"
};
再动态的从inputVariable.py里面读取diff数组
# **************加列 **************
def addColumns (self , diffList):
var = __import__("libs.inputVariable");
inputvar = getattr(var , "inputVariable");
diff = getattr(inputvar , "diff");
for arr in self.lists:
if arr[‘colorTag‘] == "general":
for d in diffList:
# 查找到相应的字段则直接复制,没查找到的则为空
arr[d] = diff.get(d) if diff.get(d) != None else "";
else:
for d in diffList:
arr[d] = "";
这样就可以灵活的扩展输出的列了。
加颜色标签
遍历数组加上colorTag,用于区别不同的行的角色,主要有
- header:标题
- site:设备的标题
- subtotal:小计
- total:总计
- general:其他
# **************加颜色标签**************
def addColorTag (self) :
try:
aDiff = [i for i in [‘BOM‘,‘typeID‘,‘description‘] if i in self.lists[0].keys()];
colTag = aDiff[0];
for aList in self.lists:
if aList[colTag] == "小计":
aList[‘colorTag‘] = "subtotal";
elif aList[colTag] == "总计":
aList[‘colorTag‘] = "total";
elif aList[‘ID‘] != "":
aList[‘colorTag‘] = ‘site‘;
else:
aList[‘colorTag‘] = "general";
self.lists[0][‘colorTag‘] = "header"
except Exception as data:
error(data)
error("缺少字段");
添加公式
预处理完了就把相应的公式添加上就可以了,对应formulaModelClass.py
处理数量列
从NHCT导出来的文档有个特点,每套设备的配置的第一行一定是主机,也就是说它的数量代表着有多少套设备
这样其他行只要除以设备数就可以得到单套设备的配置了
如何区分site
那么就有个问题了,怎么区分不同的设备呢?
我们可以使用
- self.aSite:数组,存放site行的序号
- self.aSubtotal:数组,存放小计行的序号
- self.aTotal :存放标题的序号
这样就知道每套设备从那里开始呢
那怎么获得这些数组呢?
遍历一下即可。
def getSubtotalIndex (self):
self.aSite = [];
self.aSubtotal = [];
self.aTotal = 0;
# 遍历数组,根据colorTag来进行判断
for i , arr in enumerate (self.lists):
if arr[‘colorTag‘] == ‘site‘:
self.aSite.append(i);
elif arr[‘colorTag‘] == ‘subtotal‘:
self.aSubtotal.append(i);
elif arr[‘colorTag‘] == ‘total‘:
self.aTotal = i;
else:
continue;
self.aHeader = 0;
添加“单套数量”列
关键代码如下:
# 从aSite数组里面取出site所在行的行号
for i , s in enumerate(self.aSite):
# 如果site标题所在行的quantity为空,同时在‘BOM‘那一列没有BTO的字样时
if self.lists[s][‘quantity‘] == "" and self.lists[s][tag].find("BTO") == -1:
# 获得到了套数
Qty = int (self.lists[s+1][‘quantity‘]);
# 配置开始行均为主机,所以他的quantity实际就是套数
self.lists[s][‘quantity‘] = Qty
# 将剩下的都除以套数
for j in range(s + 1 , self.aSubtotal[i]):
self.lists[j][‘quantity‘]= int(self.lists[j][‘quantity‘])/Qty;
首先从Site序号数组中获得site在那里,它的下一行即为主机
取出主机的数量,即为设备实际的套数
剩下的行都除以套数即可得到单套配置
添加总数量列
总数量列需要添加公式
关键代码如下:
for i , s in enumerate(self.aSite):
# siteInitial代表表格中显示的site起始行(表格是从1开始)
siteInitial = str(s + 1);
for j in range(s + 1 , self.aSubtotal[i]-1 + 1):
self.lists[j][‘totalQuantity‘] = ‘=$‘ + self.dCol[‘quantity‘] + "$" + siteInitial + "*" + self.dCol[‘quantity‘] + str (j + 1);
i表示site在aSite数组里面的序号,s表示每个site的序号
需要注意的是Excel是从1开始的,而数组一般是从0开始的,所以在Excel里面site的序号 = s + 1
最后一行我们可以详细的说一下:
‘=$‘ + self.dCol[‘quantity‘] + "$" + siteInitial + "*" + self.dCol[‘quantity‘] + str (j + 1);
- 在看self.dCol[‘quantity‘]表示什么意思之前,我们可以看一下assign函数里面有这样一段
colOrdinal = [‘A‘, ‘B‘, ‘C‘, ‘D‘, ‘E‘, ‘F‘, ‘G‘, ‘H‘, ‘I‘, ‘J‘, ‘K‘, ‘L‘, ‘M‘,
‘N‘, ‘O‘, ‘P‘, ‘Q‘, ‘R‘, ‘S‘, ‘T‘, ‘U‘, ‘V‘, ‘W‘, ‘X‘, ‘Y‘, ‘Z‘];
# 先组合成为dict
self.dCol = dict(zip(self.outputKeys, colOrdinal));
colOrdinal其实就是A~Z,它与outputKeys一起组成一个dict,这样的好处在于,我们可以通过self.dCol[‘quantity‘]
获得数量列所在的下标。在上图中就是"E"
- 那么"j"从那里来?
for j in range(s + 1 , self.aSubtotal[i]-1 + 1):
也就是说j表示每套设备的配置细节行。
self.aSubtotal[i]-1表示小计行前一行,然后+1就可以得到在Excel里面的行号
所以这段表示对每套的配置进行遍历,加上这个总数量行的公式即可。
总结一下,主要过程是
- 对设备site数组进行遍历,可以得到每套设备的起始行号,
- 然后对每套设备的详细配置项进行遍历,在每一行加上总数量的公式
- 注意Excel的行号与python 的数组的行号不同。
把这一小节理解了,后面其他的函数基本上都是沿着这个思路来写的
比如说重构单价列
# # **************重构单价列**************
def rehandleUnitPrice(self):
try :
for i , s in enumerate(self.aSite):
siteInitial = str(s + 1 );
for j in range(s + 1 , self.aSubtotal[i] - 1+ 1 ):
self.lists[j][‘unitsNetPrice‘] = ‘=‘ + self.dCol[‘unitsNetListPrice‘] + str(j+1) + "*" + self.dCol[‘discount‘] + str(j + 1 );
except Exception as data:
error(‘缺少price字段‘ + str(data));
添加总价列
def addTotalPrice (self):
try :
for i , s in enumerate(self.aSite):
siteInitial = str(s + 1 );
for j in range(s + 1 , self.aSubtotal[i] - 1 + 1):
self.lists[j][‘totalPrice‘] = ‘=‘ + self.dCol[‘unitsNetPrice‘] + str(j+1) + "*" + self.dCol[‘totalQuantity‘] + str(j+1);
except Exception as data:
error(‘缺少price字段‘ + str(data));
重构折扣列
rehandleDisc主要目的是方便我们统一修改折扣。
如下图所示
在每个site里面加一个折扣,它等于总计栏里面的折扣。
而配置细项里面的折扣又等于对应site里面的折扣。
这样只需要修改总计栏里面的折扣,就可以把全局的折扣改变了。
然后再修改每套设备里面的折扣就可以了。
缺点就是没有办法针对某些单板、模块进行折扣的修改。
具体代码如下:
# **************重构折扣列#######################
def rehandleDisc(self):
# 若输出含有折扣
if ‘discount‘ in self.outputKeys:
# 在总计行上填上100%
self.lists[-1][‘discount‘] = 1;
for i , s in enumerate(self.aSite):
# 所有的site上的off与总计行的off相等
self.lists[s][‘discount‘] = ‘=‘ + self.dCol[‘discount‘] + str(self.aTotal + 1);
# 详细配置的disc列与site行的相等
siteInitial = str(s + 1 );
for j in range(s + 1 , self.aSubtotal[i] - 1+ 1 ):
self.lists[j][‘discount‘] = ‘=‘ + self.dCol[‘discount‘] + siteInitial;
添加小计和总计行的公式
小计行公式
小计行要做的主要有三件事:
- 添加上“小计”字样,有些输出的表格里面可能不含有BOM或者description,我们要做一下判断
- 添加单套设备的小计,用SUMPRODUCT来实现,本质上就是数量行与价格行一一相乘并相加
- 添加设备总单价公式,直接使用SUM就可以了。
核心代码为:
# 看typeID或者description谁在输入的列中
aDiff = [i for i in [‘typeID‘, ‘description‘] if i in self.outputKeys];
tag = aDiff[0];
# 在小计行的typeID或者description位处加上配置主机的型号
for i,sub in enumerate(self.aSubtotal):
siteInitial = self.aSite[i] + 1;
siteEnd = sub - 1;
self.lists[sub][tag] = ‘‘;
if ‘typeID‘ in self.outputKeys and ‘BOM‘ not in self.outputKeys:
self.lists[sub][‘typeID‘] = ‘小计‘;
self.lists[sub][‘totalPrice‘] = ‘=SUM(‘ + self.dCol[‘totalPrice‘] + str(siteInitial + 1) + ":" + self.dCol[‘totalPrice‘] + str(siteEnd + 1) + ")";
# 单套总价格
if getParser(‘inOutmode‘,‘outputMode‘) in ["internal",‘HPE‘]:
self.lists[sub][‘unitsNetPrice‘] = ‘=SUMPRODUCT(‘ + self.dCol[‘unitsNetPrice‘] + str(siteInitial + 1) + ":" + self.dCol[‘unitsNetPrice‘] + str(siteEnd + 1) + "," + self.dCol[‘quantity‘] + str(siteInitial + 1 ) + ":" + self.dCol[‘quantity‘] + str(siteEnd + 1) + ")";
添加总计
总计的公式等于所有的当前列加起来,除以二。这是因为每套设备的价格明细之和与小计相等,如果把所有的行加起来,说明算了两次,除以二即可。
self.lists[-1][‘totalPrice‘] = ‘=SUM(‘ + self.dCol[‘totalPrice‘] + ‘2:‘ + self.dCol[‘totalPrice‘] + str(self.aTotal) + ‘)/2‘;
controller进行调度
最后我们来看一下controller是如何调度上述的代码的
# —————————————————————————参数准备—————————————————————————
# 分别获取输入和输出文件的名称
var = __import__("libs.inputVariable")
inputvar = getattr(var,"inputVariable")
inputFile = M("file").getProjectName();
outputFile = M("outputfile").getOutputFile(inputFile);
info("打开的文件是" + inputFile);
# 获得输入和输出的keys
[inputParam , outputParam] = M("parameter").getParameter(inputFile);
# 以quotationTools的根目录作为基准
basepath = os.path.dirname(os.path.dirname(os.path.dirname(__file__)));
inputPath = os.path.join(basepath,getParser(‘path‘,‘inputfilePath‘),inputFile);
outputPath = os.path.join(basepath,getParser(‘path‘,‘outputfilePath‘),outputFile);
# 主sheetName
sheetName = ‘价格明细清单‘;
# 添加公式
iFormula = M("formula");
iFormula.assign(lists, list(outputParam.keys()));
lists = iFormula.addFormula();
# 替换首行为想让他输出的模式
for k in outputParam.keys():
lists[0][k] = outputParam[k];
原文地址:https://www.cnblogs.com/dy2903/p/8466877.html