Python寫xml文件

　　在上一篇 使用selenium找出外賣點餐次數最多的10個顧客 中，抓取信息后只輸出了點餐次數TOPN的顧客，其實可以把信息保存到本地，便於做更多的統計。

　　為了便於后續的讀取處理，這里就將信息保存在xml文件中，想到得到的文件如下：

<?xml version="1.0" encoding="utf-8"?>
<orderlist>
    <order>
        <customer>姓名1</customer>
        <phone>電話1</phone>
        <address>地址1</address>
        <count>點餐次數1</count>
    </order>
    <order>
        <customer>姓名2</customer>
        <phone>電話2</phone>
        <address>地址2</address>
        <count>點餐次數2</count>
    </order>
</orderlist>

　　Python寫xml比較簡單，直接使用minidom即可。

　　思路也比較簡單，因為dom文檔就是一棵樹，每一個標簽都是一個節點，包括文本內容也是節點，因此從根節點開始，把節點一級一級的插入即可。

　　在上一篇中已經將抓取的信息存入了字典orderDict中，這里我們直接在類Order中添加一個方法，將orderDict里面的每一項信息插入DOM樹，然后再寫入本地xml文件。

　　由於比較簡單，直接給出代碼：

　　

    from xml.dom.minidom import Document
    # 將self.orderDict中的信息寫入本地xml文件，參數filename是xml文件名
    def writeInfoToXml(self, filename):
        # 創建dom文檔
        doc = Document()

        # 創建根節點
        orderlist = doc.createElement('orderlist')
        # 根節點插入dom樹
        doc.appendChild(orderlist)

        # 依次將orderDict中的每一組元素提取出來，創建對應節點並插入dom樹
        for (k, v) in self.orderDict.iteritems():
            # 分離出姓名，電話，地址，點餐次數
            (name, tel, addr, cnt) = (v[0], k, v[1], v[2])

            # 每一組信息先創建節點<order>，然后插入到父節點<orderlist>下
            order = doc.createElement('order')
            orderlist.appendChild(order)

            # 將姓名插入<order>中
            # 創建節點<customer>
            customer = doc.createElement('customer')
            # 創建<customer>下的文本節點
            customer_text = doc.createTextNode(name)
            # 將文本節點插入到<customer>下
            customer.appendChild(customer_text)
            # 將<customer>插入到父節點<order>下
            order.appendChild(customer)

            # 將電話插入<order>中，處理同上
            phone = doc.createElement('phone')
            phone_text = doc.createTextNode(tel)
            phone.appendChild(phone_text)
            order.appendChild(phone)

            # 將地址插入<order>中，處理同上
            address = doc.createElement('address')
            address_text = doc.createTextNode(addr)
            address.appendChild(address_text)
            order.appendChild(address)

            # 將點餐次數插入<order>中，處理同上
            count = doc.createElement('count')
            count_text = doc.createTextNode(str(cnt))
            count.appendChild(count_text)
            order.appendChild(count)

        # 將dom對象寫入本地xml文件
        with open(filename, 'w') as f:
            f.write(doc.toprettyxml(indent='\t', encoding='utf-8'))

        return

　　在先調用getAllOrders方法之后，再調用writeInfoToXml即可將所有信息寫入xml文件中。