提取不同的内容类型MHT文件分成多个MHT文件

Question

我写的MHT脚本解析MHT文件，并从母公司中提取部分信息，并将其写入到一个单独的MHT文件

我写了下面的这在file_location和搜索特定的content_id打开MHT文件，并将其写入到一个新的MHT文件

def extract_content(self, file_location, content_id,extension): 
    first_part = file_location.split(extension)[0] 
    #checking if file exists 
    new_file = first_part + "-" + content_id.split('.')[0] + extension 

    while os.path.exists(new_file): 
     os.remove(new_file) 

    with open(file_location, 'rb') as mime_file, open(new_file, 'w') as output: 
     ***#Extracting the message from the mht file*** 
     message = message_from_file(mime_file) 
     t = mimetypes.guess_type(file_location)[0] 

     #Walking through the message 
     for i, part in enumerate(message.walk()): 

      #Check the content_id if the one we are looking for 
      if part['Content-ID'] == '<' + content_id + '>': 
       ***witing the contents*** 
       output.write(part.as_string(unixfrom=False)) 

显然我不能够在IE中应用/ PDF和应用的情况下，打开输出部分功能/ octet-stream。

我怎么写这些内容类型如应用/ PDF和 应用程序/八位字节流中的MHT文件，以便我能够查看IE图片或PDF？

感谢

Answer 1

试试这个：

... 
if m['Content-type'].startswith('text/'): 
        m["Content-Transfer-Encoding"] = "quoted-printable" 

       else: 
        m["Content-Transfer-Encoding"] = "base64" 

       m.set_payload(part.get_payload())       
       ****Writing to output**** 
       info = part.as_string(unixfrom=False) 
       info = info.replace('application/octet-stream', 'text/plain') 
       output.write(info) 
... 

告诉我，如果它的工作原理。