Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import email
- import sys
- class AdMailParser(object):
- def __init__(self, raw_mail):
- self.pics = []
- self.text_message = ''
- self.raw_mail = raw_mail
- def parse(self):
- mime_mail = email.message_from_string(self.raw_mail)
- if mime_mail.is_multipart():
- self.__process_multipart(mime_mail.get_payload())
- else:
- self.__process_payload(mime_mail)
- def __process_payload(self, message):
- if message.get_content_type() == 'text/plain':
- self.text_message += message.get_payload(decode=True)
- elif message.get_content_type() == 'application/octet-stream':
- self.pics.append(message.get_payload(decode=True))
- def __process_multipart(self, parts):
- for part in parts:
- if part.is_multipart():
- self.__process_multipart(part.get_payload())
- else:
- self.__process_payload(part)
- if __name__ == '__main__':
- input_lines = sys.stdin.readlines()
- raw_mail = ''.join(input_lines)
- p = AdMailParser(raw_mail)
- p.parse()
- #print p.text_message
- #print p.pics[0]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement