python批量提取eml附件

從批量eml文件中提取附件，使用方式如下

 

代碼如下

import email
import os
import sys

#獲取eml附件信息
def Get_Annex_Message(FilePath, Annex_Path):
    global sum
    try:
        fp = open(FilePath, 'rb')        #打開任意格式文件，通過email庫來判斷是否為eml文件
        msg = email.message_from_binary_file(fp)
        for part in msg.walk():            #循環信件中的每一個mime的數據塊
            if part.get_content_maintype() == 'multipart':
                continue
            Annex_name = part.get_filename()
            if Annex_name:        #如果附件存在名字
                Annex_name = str(sum) + '.' + Annex_name
                fp = open(os.path.join(Annex_Path, Annex_name), 'wb')
                fp.write(part.get_payload(decode=True))
                sum += 1
                
    except Exception as e:
        print(e)
        return
        
#遞歸文件夾下所有文件
def List_Filepath(Eml_Path, Annex_Path):
    for parent,dirnames,filenames in os.walk(Eml_Path): #遍歷文件夾
        for dirname in dirnames:        #對文件夾進行遞歸
            List_Filepath(dirname, Annex_Path)
        for filename in filenames:        #r對文件進行判斷
            FilePath = os.path.join(parent,filename)
            Get_Annex_Message(FilePath, Annex_Path)
        
#創建目的文件夾
def Create_Dir(Annex_Path):
    if os.path.exists(Annex_Path):
        print("dir exists, Annex file will create in %s" % Annex_Path)
    else:
        os.mkdir(Annex_Path)
        
#主函數
def main():
    global sum 
    sum = int(1)
    Eml_Path = sys.argv[1]        #第一個參數為eml所在文件夾
    Annex_Path = sys.argv[2]    #第二個參數為eml附件輸出的路徑
    Create_Dir(Annex_Path)        #創建保存附加的文件夾
    List_Filepath(Eml_Path, Annex_Path)        

if __name__ == "__main__":
    main()