python 3.x 學習筆記8 (os模塊及xml修改)

1.os模塊操作

os.getcwd()：                                       # 查看當前所在路徑。

os.listdir(path):                                       # 列舉目錄下的所有文件，返回的是列表類型。

os.path.abspath(path):                          # 返回path的絕對路徑。

os.path.join(path1,path2,...):                   # 將path進行組合，若其中有絕對路徑，則之前的path將被刪除。

os.path.dirname(path):                        #   返回path中的文件夾部分，結果不包含'\'

os.path.basename(path):                     # 返回path中的文件名。

os.path.getmtime(path):                      #  文件或文件夾的最後修改時間，從新紀元到訪問時的秒數。

os.path.getatime(path):                      #  文件或文件夾的最後訪問時間，從新紀元到訪問時的秒數。

os.path.getctime(path):                       #  文件或文件夾的創建時間，從新紀元到訪問時的秒數。

os.path.getsize(path):                         #  文件或文件夾的大小，若是文件夾返回0

os.path.exists(path):                           #  文件或文件夾是否存在，返回True 或 False。

2.xml的使用

 xml建立

from xml.etree import ElementTree as ET
def build_sitemap():
    urlset = ET.Element("urlset")       # ET.Element創建一個根節點，標簽為urlset
    url = ET.SubElement(urlset,"url")   # ET.SubElement在根節點urlset下建立子節點
    loc = ET.SubElement(url,"loc",attrib={"name":"百度"})         #attrib為創建屬性
    loc.text = "http://www/baidu.com"                             #loc.test 為寫入內容
    time = ET.SubElement(url,"time")
    time.text = "2018-1-30"
    change = ET.SubElement(url,"change")
    change.text = "daily"
    priority = ET.SubElement(url,"priority")
    priority.text = "1.0"
    tree = ET.ElementTree(urlset)
    tree.write("set.xml",'utf-8')         #寫入時加上‘utf-8’可以轉譯中文，不會有亂碼
if __name__ == '__main__':
    build_sitemap()

 生成的xml

<urlset>
    <url>
        <loc name="百度">http://www/baidu.com</loc>
        <time>2018-1-30</time>
        <change>daily</change>
        <priority>1.0</priority>
    </url>
</urlset>

 下麵是要修改的文件

<?xml version="1.0"?>
<data>
    <country name="Liechtenstein">
        <rank>1</rank>
        <year>2008</year>
        <gdppc>141100</gdppc>
        <neighbor name="Austria" direction="E"/>
        <neighbor name="Switzerland" direction="W"/>
    </country>
    <country name="Singapore">
        <rank>4</rank>
        <year>2011</year>
        <gdppc>59900</gdppc>
        <neighbor name="Malaysia" direction="N"/>
    </country>
    <country name="Panama">
        <rank>68</rank>
        <year>2011</year>
        <gdppc>13600</gdppc>
        <neighbor name="Costa Rica" direction="W"/>
        <neighbor name="Colombia" direction="E"/>
    </country>
</data>

修改程式

import xml.etree.ElementTree as ET

tree = ET.parse('xmltest.xml')
root = tree.getroot()
#修改
for node in root.iter('year'):
    new_year = int(node.text) + 1
    node.text = str(new_year)
    node.set('updated_by','hsj')

tree.write('xmltest2.xml')


#刪除

for country in root.findall('country'):
    rank = int(country.find('rank').text)
    if rank > 50:
        root.remove(country)

tree.write('xmltest3.xml')