summaryrefslogtreecommitdiff
path: root/src/leap/mail/imap/parser.py
blob: 1ae19c03316594ac418a1ff3d4692d11943e759d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
# -*- coding: utf-8 -*-
# parser.py
# Copyright (C) 2013 LEAP
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""
Mail parser mixins.
"""
import cStringIO
import StringIO
import re

from email.parser import Parser


class MailParser(object):
    """
    Mixin with utility methods to parse raw messages.
    """
    def __init__(self):
        """
        Initializes the mail parser.
        """
        self._parser = Parser()

    def _get_parsed_msg(self, raw):
        """
        Return a parsed Message.

        :param raw: the raw string to parse
        :type raw: basestring, or StringIO object
        """
        msg = self._get_parser_fun(raw)(raw, True)
        return msg

    def _get_parser_fun(self, o):
        """
        Retunn the proper parser function for an object.

        :param o: object
        :type o: object
        :param parser: an instance of email.parser.Parser
        :type parser: email.parser.Parser
        """
        if isinstance(o, (cStringIO.OutputType, StringIO.StringIO)):
            return self._parser.parse
        if isinstance(o, basestring):
            return self._parser.parsestr
        # fallback
        return self._parser.parsestr

    def _stringify(self, o):
        """
        Return a string object.

        :param o: object
        :type o: object
        """
        if isinstance(o, (cStringIO.OutputType, StringIO.StringIO)):
            return o.getvalue()
        else:
            return o


class MBoxParser(object):
    """
    Utility function to parse mailbox names.
    """
    INBOX_NAME = "INBOX"
    INBOX_RE = re.compile(INBOX_NAME, re.IGNORECASE)

    def _parse_mailbox_name(self, name):
        """
        :param name: the name of the mailbox
        :type name: unicode

        :rtype: unicode
        """
        if self.INBOX_RE.match(name):
            # ensure inital INBOX is uppercase
            return self.INBOX_NAME + name[len(self.INBOX_NAME):]
        return name