昨天有個需求，要獲取服務器某個目錄下的某些類型的文件，考慮到服務器即有Linux、又有Windows，所以寫了一個Python小程序來完成這項工作。
Linux服務器有CentOS、Fedora等，都預先安裝了Python，版本從2.4到2.5不等，而Windows類型的服務器也多數安裝了Python，因此只要在本機寫好一個腳本，上傳到對應機器，在運行時修改參數即可。
Python操作文件和文件夾使用的是os庫，下面的代碼中主要用到了幾個函數：
- 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 =
""
? 得到存儲文件信息的數組后，還可以將其另存成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()
執行的代碼如下
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命令行狀態下，執行Python filesearch.py，便可以生成名為 folder.xml 的文件。
如果要在Windows中運行該程序，需要把folder變量改成Windows下的格式，例如c:\\apache2\htdocs，然后執行c:\python25\python.exe filesearch.py（這里假設python的安裝目錄是c:\python25）
所有程序可以在這里下載: 源程序
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
