[tests] adapt events tests to recent changes
[leap_pycommon.git] / src / leap / common / mail.py
1 # -*- coding: utf-8 -*-
2 # mail.py
3 # Copyright (C) 2013 LEAP
4 #
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 """
18 Utility functions for email.
19 """
20 import email
21 import re
22
23
24 def get_email_charset(content, default="utf-8"):
25     """
26     Mini parser to retrieve the charset of an email.
27
28     :param content: mail contents
29     :type content: unicode or str
30     :param default: optional default value for encoding
31     :type default: str or None
32
33     :returns: the charset as parsed from the contents
34     :rtype: str
35     """
36     if isinstance(content, unicode):
37         content.encode("utf-8", "replace")
38
39     charset = default
40     try:
41         em = email.message_from_string(content)
42         # Miniparser for: Content-Type: <something>; charset=<charset>
43         charset_re = r'''charset=(?P<charset>[\w|\d|-]*)'''
44         charset = re.findall(charset_re, em["Content-Type"])[0]
45         if charset is None or len(charset) == 0:
46             charset = default
47     except Exception:
48         pass
49     return charset