应用于多个文件或折叠的Python代码

Question

我对Python相当陌生，我已经创建了一些简单的Python代码（如下）来打印（或返回）一个.eml文件的主体、主题和电子邮件地址。这个很好用。但是，如何将此代码应用于.eml文件的文件夹？在

import email

fp = open('C:\emlfiles\TEST_00000.eml')

msg = email.message_from_file(fp)

subject = msg.get('subject')
subject = str(subject)
fromAdd = msg.get('from')
fromAdd = str(fromAdd)
payload = msg.get_payload()
if type(payload) == type(list()) :
    payload = payload[0] 
if type(payload) != type('') :
    payload = str(payload)

print subhec + fromAdd + payload

Answer 1

使用os.listdir将所有内容放入for循环：

import email

infolder = "C:\\emlfiles\\"
for infile in os.listdir(infolder):
    fp = open(os.path.join(infolder, infile))
    msg = email.message_from_file(fp)
    # etc

也许更好的方法是将代码包装在一个函数中，然后调用它：

^{pr2}$

Answer 2

看一下^{}，它做了你想要的。在