28. 构建xml文档

拼搏现实的明天。 2023-06-27 03:44 12阅读 0赞

有时，我们需要将其他格式数据转换为xml，例如把书籍信息csv文件转换成xml：

books.csv：
        书名, 作者, 出版社, 价格
        精通scrapy网络爬虫, 刘硕, 清华大学出版社, 46.00
        ...

books.xml：
    <Data>
        <Book>
            <书名>精通scrapy网络爬虫</书名>
            <作者>刘硕</作者>
            <出版社>清华大学出版社</出版社>
            <价格>46.00</价格>
        </Book>
        ...
    <Data>

要求：将csv文件转换成xml。

解决方案：

使用标准库中的`xml.etree.ElementTree`类，构建ElementTree，然后调用`write()`方法写入文件。

--------------------

*  对于`xml.tree.ElementTree`类：

`ElementTree`提供了两个对象：`ElementTree`和`Element`

`ElementTree`：将整个xml转化为树，对整个xml文档进行操作（读取，写入，查找等）一般在ElementTree层面进行。

`Element`：代表树上单个节点，对单个xml元素及其子元素进行操作，则是在Element层面进行。

*  构建xml：

>>> import xml.etree.ElementTree as ET
    
    >>> data = ET.Element('Data')
    
    >>> book = ET.Element('Book')
    
    >>> autor = ET.Element('作者')
    
    >>> book.set('x', 'abc')                #设置属性，与get对应
    
    >>> autor.text = '刘硕'
    
    >>> data.append(book)               #为data设置子节点
    
    >>> book.append(autor)              #为book设置子节点
    
    >>> ET.dump(data)
    <Data><Book x="abc"><作者>刘硕</作者></Book></Data>
    
    >>> et = ET.ElementTree(data)
    
    >>> et
    <xml.etree.ElementTree.ElementTree object at 0x7fe39ac381d0>
    
    >>> et.write('test.xml', encoding='utf8')               #写入文件

# cat test.xml 
    <?xml version='1.0' encoding='utf8'?>
    <Data><Book x="abc"><作者>刘硕</作者></Book></Data>

设置子节点还可以使用`ET.SubElement()`方法，例如`book = ET.Element('Book')` `data.append(book)`等价于`book = ET.SubElement(data, 'Book')`

--------------------

*  方案示例：

books.csv：

书名, 作者, 出版社, 价格
    精通scrapy网络爬虫, 刘硕, 清华大学出版社, 46.00
    算法导论, Charles E.Leiserson, 人民邮电出版社, 85.00
    Python灰帽子, Justin Seitz, 电子工业出版社, 39.00
    精通scrapy网络爬虫1, 刘硕, 清华大学出版社, 46.00
    算法导论1, Charles E.Leiserson, 人民邮电出版社, 85.00
    Python灰帽子1, Justin Seitz, 电子工业出版社, 39.00
    精通scrapy网络爬虫2, 刘硕, 清华大学出版社, 46.00
    算法导论2, Charles E.Leiserson, 人民邮电出版社, 85.00
    Python灰帽子2, Justin Seitz, 电子工业出版社, 39.00
    精通scrapy网络爬虫3, 刘硕, 清华大学出版社, 46.00
    算法导论3, Charles E.Leiserson, 人民邮电出版社, 85.00
    Python灰帽子3, Justin Seitz, 电子工业出版社, 39.00

import csv
    from xml.etree.ElementTree import ElementTree, Element, SubElement
    
    def csv_to_xml(csv_path, xml_path):
        with open(csv_path) as f:
            reader = csv.reader(f)
            headers = next(reader)
    
            root = Element('Data')
            root.text = '\n\t'
            root.tail = '\n'
    
            for row in reader:
                book = SubElement(root, 'Book')
                book.text = '\n\t\t'
                book.tail = '\n\t'
    
                for tag, text in zip(headers, row):
                    e = SubElement(book, tag)
                    e.text = text
                    e.tail = '\n\t\t'
                e.tail = '\n\t'
            book.tail = '\n'
    
            ElementTree(root).write(xml_path, encoding='utf8')
    
    csv_to_xml('books.csv', 'books.xml')

books.xml：

<?xml version='1.0' encoding='utf8'?>
    <Data>
    	<Book>
    		<书名>精通scrapy网络爬虫</书名>
    		<作者>刘硕</作者>
    		<出版社>清华大学出版社</出版社>
    		<价格>46.00</价格>
    	</Book>
    	<Book>
    		<书名>算法导论</书名>
    		<作者>Charles E.Leiserson</作者>
    		<出版社>人民邮电出版社</出版社>
    		<价格>85.00</价格>
    	</Book>
    	<Book>
    		<书名>Python灰帽子</书名>
    		<作者>Justin Seitz</作者>
    		<出版社>电子工业出版社</出版社>
    		<价格>39.00</价格>
    	</Book>
    	<Book>
    		<书名>精通scrapy网络爬虫1</书名>
    		<作者>刘硕</作者>
    		<出版社>清华大学出版社</出版社>
    		<价格>46.00</价格>
    	</Book>
    	<Book>
    		<书名>算法导论1</书名>
    		<作者>Charles E.Leiserson</作者>
    		<出版社>人民邮电出版社</出版社>
    		<价格>85.00</价格>
    	</Book>
    	<Book>
    		<书名>Python灰帽子1</书名>
    		<作者>Justin Seitz</作者>
    		<出版社>电子工业出版社</出版社>
    		<价格>39.00</价格>
    	</Book>
    	<Book>
    		<书名>精通scrapy网络爬虫2</书名>
    		<作者>刘硕</作者>
    		<出版社>清华大学出版社</出版社>
    		<价格>46.00</价格>
    	</Book>
    	<Book>
    		<书名>算法导论2</书名>
    		<作者>Charles E.Leiserson</作者>
    		<出版社>人民邮电出版社</出版社>
    		<价格>85.00</价格>
    	</Book>
    	<Book>
    		<书名>Python灰帽子2</书名>
    		<作者>Justin Seitz</作者>
    		<出版社>电子工业出版社</出版社>
    		<价格>39.00</价格>
    	</Book>
    	<Book>
    		<书名>精通scrapy网络爬虫3</书名>
    		<作者>刘硕</作者>
    		<出版社>清华大学出版社</出版社>
    		<价格>46.00</价格>
    	</Book>
    	<Book>
    		<书名>算法导论3</书名>
    		<作者>Charles E.Leiserson</作者>
    		<出版社>人民邮电出版社</出版社>
    		<价格>85.00</价格>
    	</Book>
    	<Book>
    		<书名>Python灰帽子3</书名>
    		<作者>Justin Seitz</作者>
    		<出版社>电子工业出版社</出版社>
    		<价格>39.00</价格>
    	</Book>
    </Data>

--------------------