快捷導航

python目錄操作之python遍歷文件夾后將結(jié)果存儲為xml

更新時間：2014年01月27日 10:26:15 作者：

需求是獲取服務(wù)器某個目錄下的某些類型的文件，考慮到服務(wù)器即有Linux、又有Windows，所以寫了一個Python小程序來完成這項工作,大家參考使用吧

Linux服務(wù)器有CentOS、Fedora等，都預先安裝了Python，版本從2.4到2.5不等，而Windows類型的服務(wù)器也多數(shù)安裝了Python，因此只要在本機寫好一個腳本，上傳到對應(yīng)機器，在運行時修改參數(shù)即可。

Python操作文件和文件夾使用的是os庫，下面的代碼中主要用到了幾個函數(shù)：

os.listdir:列出目錄下的文件和文件夾
os.path.join:拼接得到一個文件/文件夾的全路徑
os.path.isfile:判斷是否是文件
os.path.splitext：從名稱中取出一個子部分

下面是目錄操作的代碼

復制代碼代碼如下:

def search(folder, filter, allfile):
    folders = os.listdir(folder)
    for name in folders:
        curname = os.path.join(folder, name)
        isfile = os.path.isfile(curname)
        if isfile:
            ext = os.path.splitext(curname)[1]
            count = filter.count(ext)
            if count>0:
                cur = myfile()
                cur.name = curname
                allfile.append(cur)
        else:
            search(curname, filter, allfile)
    return allfile

在返回文件的各種信息時，使用自定義類allfile來保存文件的信息，在程序中只用到了文件的全路徑，如果需要同時記錄文件的大小、時間、類型等信息，可以仿照代碼進行擴充。

復制代碼代碼如下:

class myfile:
    def __init__(self):
        self.name = ""

得到存儲文件信息的數(shù)組后，還可以將其另存成xml格式，下面是代碼，在使用時，需要從Document中導入xml.dom.minidom

下面是保存為xml的代碼

復制代碼代碼如下:

def generate(allfile, xml):
doc = Document()

root = doc.createElement("root")
doc.appendChild(root)

    for myfile in allfile:
        file = doc.createElement("file")
        root.appendChild(file)

        name = doc.createElement("name")
        file.appendChild(name)
        namevalue = doc.createTextNode(myfile.name)
        name.appendChild(namevalue)

    print doc.toprettyxml(indent="")
    f = open(xml, 'a+')
    f.write(doc.toprettyxml(indent=""))
    f.close()

執(zhí)行的代碼如下

復制代碼代碼如下:

if __name__ == '__main__':
    folder = "/usr/local/apache/htdocs"
    filter = [".html",".htm",".php"]
    allfile = []
    allfile = search(folder, filter, allfile)
    len = len(allfile)
    print "found: " + str(len) + " files"

xml = "folder.xml"
generate(allfile, xml)

在Linux命令行狀態(tài)下，執(zhí)行Python filesearch.py，便可以生成名為folder.xml的文件。

如果要在Windows中運行該程序，需要把folder變量改成Windows下的格式，例如c:\\apache2\htdocs，然后執(zhí)行c:\python25\python.exe filesearch.py（這里假設(shè)python的安裝目錄是c:\python25）

您可能感興趣的文章: