QuotationToolModel的实现，形成价格明细清单.md

Posted 2020-10-25 dy2903

项目链接：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];