# -*- coding: utf-8 -*-

from email.mime.text import MIMEText

from email.header import Header

from email.Utils import formatdate

class Message:

    def __init__(self,content, typ="Message"):

        self.content=content

        self.typ = typ

    def getContent(self):

        return self.content

    def __eq__(self,other):

        return self.content == other.content

    def __ne__(self,other):

        return not self.__eq__(other)

class SMS(Message):

    def __init__(self, cont, from_ = None):

        self.from_ = from_

class Fax(Message):

    def __init__(self,header,cont,attachments=[]):

        self.header=header

        self.attachments=attachments

    def getAttachment(self,i):

        return self.attachments[i]

    def __eq__(self,other):

        if not  Message.__eq__(self,other):

            return False

        if self.header != other.header:

            return False

        if len(self.attachments) != len(other.attachments):

            return False

        for a in self.attachments:

            if a not in other.attachments:

                return False

        return True

class Mail(Message):

    def __init__(self, subject, body, frm):

        con = MIMEText(body.encode("utf-8"), _charset='utf-8')

        sub = Header(subject.encode('utf-8'), 'utf-8')

        con['Subject'] = sub

        con['Date'] = formatdate(localtime=True)

        self.subject = subject

        self.body = body

        self.frm=frm

        Message.__init__(self, con, typ='mail')

    def as_string(self):

        return self.content.as_string()

    def getFrom(self):

        return self.frm

    def __eq__(self,other):

        if self.as_string() != other.as_string():

            return False

        if self.frm != other.frm:

            return False

        return True

    def __repr__(self):

        return "<Mail(%s,%s,%s)>"%(self.subject,self.body,self.frm)