Advertisement
Guest User

Untitled

a guest
May 27th, 2017
48
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.16 KB | None | 0 0
  1. import email
  2. import sys
  3.  
  4. class AdMailParser(object):
  5.  
  6.     def __init__(self, raw_mail):
  7.         self.pics = []
  8.         self.text_message = ''
  9.         self.raw_mail = raw_mail
  10.  
  11.  
  12.     def parse(self):
  13.         mime_mail = email.message_from_string(self.raw_mail)
  14.         if mime_mail.is_multipart():
  15.             self.__process_multipart(mime_mail.get_payload())
  16.         else:
  17.             self.__process_payload(mime_mail)
  18.  
  19.  
  20.     def __process_payload(self, message):
  21.         if message.get_content_type() == 'text/plain':
  22.             self.text_message += message.get_payload(decode=True)
  23.         elif message.get_content_type() == 'application/octet-stream':
  24.             self.pics.append(message.get_payload(decode=True))    
  25.    
  26.     def __process_multipart(self, parts):
  27.         for part in parts:
  28.             if part.is_multipart():
  29.                 self.__process_multipart(part.get_payload())
  30.             else:
  31.                 self.__process_payload(part)
  32.  
  33.  
  34. if __name__ == '__main__':
  35.     input_lines = sys.stdin.readlines()
  36.     raw_mail = ''.join(input_lines)
  37.     p = AdMailParser(raw_mail)
  38.     p.parse()
  39.     #print p.text_message
  40.     #print p.pics[0]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement