gulag/app/Entities.py

import re

class MailboxException(Exception):
  message = None
  def __init__(self,message):
    self.message = message

class Mailbox:
  email_address = None
  name = None
  imap_server = None
  imap_security = None
  imap_user = None
  imap_pass = None
  imap_mailbox = None
  imap_mailbox_fp = None
  imap_separator = None
  comment = None
  href = None

  def __init__(self,mb_ref):
    if 'email_address' not in mb_ref:
      raise MailboxException("'email_address' is mandatory!")
    self.email_address = mb_ref['email_address']
    if 'name' not in mb_ref:
      raise MailboxException("'name' is mandatory!")
    self.name = mb_ref['name']
    if 'imap_server' not in mb_ref:
      raise MailboxException("'imap_server' is mandatory!")
    self.imap_server = mb_ref['imap_server']
    if 'imap_security' in mb_ref:
      if re.match("^(plain|starttls|tls)$", mb_ref['imap_security']) is not None:
        self.imap_security = mb_ref['imap_security']
      else:
        raise MailboxException('imap_security: {} is invalid! '+ 
          'Valid values: plain,starttls,tls'.format(mb_ref['imap_security'])
        )
    else:
      raise MailboxException("'imap_security' is a mandatory!")
    if 'imap_user' not in mb_ref:
      raise MailboxException("'imap_user' is mandatory!")
    self.imap_user = mb_ref['imap_user']
    if 'imap_pass' not in mb_ref:
      raise MailboxException("'imap_pass' is mandatory!")
    self.imap_pass = mb_ref['imap_pass']
    if 'imap_mailbox' not in mb_ref:
      raise MailboxException("'imap_mailbox' is mandatory!")
    self.imap_mailbox = mb_ref['imap_mailbox']
    if 'imap_mailbox_fp' not in mb_ref:
      raise MailboxException("'imap_mailbox_fp' is mandatory!")
    self.imap_mailbox_fp = mb_ref['imap_mailbox_fp']
    if 'imap_separator' not in mb_ref:
      raise MailboxException("'imap_separator' is mandatory!")
    self.imap_seperator = mb_ref['imap_separator']
    if 'comment' in mb_ref:
      self.comment = mb_ref['comment']
    if 'href' in mb_ref:
      self.href = mb_ref['href']

class QuarMailException(Exception):
  message = None
  def __init__(self,message):
    self.message = message

class QuarMail:
  id = None
  ctime = None
  mx_queue_id = None
  env_from = None
  env_rcpt = None
  hdr_cf = None
  hdr_from = None
  hdr_subject = None
  hdr_msgid = None
  hdr_date = None
  cf_meta = None
  mailbox_id = None
  imap_uid = None
  msg_size = None
  href = None

  def __init__(self,qm_ref):
    if 'id' not in qm_ref:
      raise QuarMailException("'id' is mandatory!")
    self.id = qm_ref['id']
    if 'ctime' not in qm_ref:
      raise QuarMailException("'ctime' is mandatory!")
    self.ctime = qm_ref['ctime']
    if 'mx_queue_id' not in qm_ref:
      raise QuarMailException("'mx_queue_id' is mandatory!")
    self.mx_queue_id = qm_ref['mx_queue_id']
    if 'env_from' not in qm_ref:
      raise QuarMailException("'env_from' is mandatory!")
    self.env_from = qm_ref['env_from']
    if 'env_rcpt' not in qm_ref:
      raise QuarMailException("'env_rcpt' is mandatory!")
    self.env_rcpt = qm_ref['env_rcpt']
    if 'hdr_cf' in qm_ref:
      self.hdr_cf = qm_ref['hdr_cf']
    if 'hdr_from' in qm_ref:
      self.hdr_from = qm_ref['hdr_from']
    if 'hdr_subject' in qm_ref:
      self.hdr_subject = qm_ref['hdr_subject']
    if 'hdr_msgid' in qm_ref:
      self.hdr_msgid = qm_ref['hdr_msgid']
    if 'hdr_date' in qm_ref:
      self.hdr_date = qm_ref['hdr_date']
    if 'cf_meta' in qm_ref:
      self.cf_meta = qm_ref['cf_meta']
    if 'mailbox_id' not in qm_ref:
      raise QuarMailException("'mailbox_id' is mandatory!")
    self.mailbox_id = qm_ref['mailbox_id']
    if 'imap_uid' not in qm_ref:
      raise QuarMailException("'imap_uid' is mandatory!")
    self.imap_uid = qm_ref['imap_uid']
    if 'msg_size' not in qm_ref:
      raise QuarMailException("'msg_size' is mandatory!")
    self.msg_size = qm_ref['msg_size']
    if 'href' in qm_ref:
      self.href = qm_ref['href']


class AttachmentException(Exception):
  message = None
  def __init__(self,message):
    self.message = message

class Attachment:
  id = None
  filename = None
  content_type = None
  comment = None
  href = None

  def __init__(self,at_ref):
    if 'id' not in at_ref:
      raise AttachmentException("'id' is mandatory!")
    self.id = at_ref['id']
    if 'filename' not in at_ref:
      raise AttachmentException("'filename' is mandatory!")
    self.filename = at_ref['filename']
    if 'content_type' not in at_ref:
      raise AttachmentException("'content_type' is mandatory!")
    self.content_type = at_ref['content_type']
    if 'comment' in at_ref:
      self.comment = at_ref['comment']
    if 'href' in at_ref:
      self.href = at_ref['href']