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