daklib/textutils.py

   1 #!/usr/bin/env python
   2 # vim:set et ts=4 sw=4:
   3
   4 """Text utility functions
   5
   6 @contact: Debian FTP Master <ftpmaster@debian.org>
   7 @copyright: 2000, 2001, 2002, 2003, 2004, 2005, 2006  James Troup <james@nocrew.org>
   8 @license: GNU General Public License version 2 or later
   9 """
  10
  11 # This program is free software; you can redistribute it and/or modify
  12 # it under the terms of the GNU General Public License as published by
  13 # the Free Software Foundation; either version 2 of the License, or
  14 # (at your option) any later version.
  15
  16 # This program is distributed in the hope that it will be useful,
  17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 # GNU General Public License for more details.
  20
  21 # You should have received a copy of the GNU General Public License
  22 # along with this program; if not, write to the Free Software
  23 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  24
  25 import email.Header
  26
  27 from dak_exceptions import *
  28 from regexes import re_parse_maintainer
  29
  30 ################################################################################
  31
  32 def force_to_utf8(s):
  33     """
  34     Forces a string to UTF-8.  If the string isn't already UTF-8,
  35     it's assumed to be ISO-8859-1.
  36     """
  37     try:
  38         unicode(s, 'utf-8')
  39         return s
  40     except UnicodeError:
  41         latin1_s = unicode(s,'iso8859-1')
  42         return latin1_s.encode('utf-8')
  43
  44 def rfc2047_encode(s):
  45     """
  46     Encodes a (header) string per RFC2047 if necessary.  If the
  47     string is neither ASCII nor UTF-8, it's assumed to be ISO-8859-1.
  48     """
  49     for enc in ['ascii', 'utf-8', 'iso-8859-1']:
  50         try:
  51             h = email.Header.Header(s, enc, 998)
  52             return str(h)
  53         except UnicodeError:
  54             pass
  55
  56     # If we get here, we're boned beyond belief
  57     return ''
  58
  59 ################################################################################
  60
  61 # <Culus> 'The standard sucks, but my tool is supposed to interoperate
  62 #          with it. I know - I'll fix the suckage and make things
  63 #          incompatible!'
  64
  65 def fix_maintainer(maintainer):
  66     """
  67     Parses a Maintainer or Changed-By field and returns:
  68       1. an RFC822 compatible version,
  69       2. an RFC2047 compatible version,
  70       3. the name
  71       4. the email
  72
  73     The name is forced to UTF-8 for both 1. and 3..  If the name field
  74     contains '.' or ',' (as allowed by Debian policy), 1. and 2. are
  75     switched to 'email (name)' format.
  76
  77     """
  78     maintainer = maintainer.strip()
  79     if not maintainer:
  80         return ('', '', '', '')
  81
  82     if maintainer.find("<") == -1:
  83         email = maintainer
  84         name = ""
  85     elif (maintainer[0] == "<" and maintainer[-1:] == ">"):
  86         email = maintainer[1:-1]
  87         name = ""
  88     else:
  89         m = re_parse_maintainer.match(maintainer)
  90         if not m:
  91             raise ParseMaintError, "Doesn't parse as a valid Maintainer field."
  92         name = m.group(1)
  93         email = m.group(2)
  94
  95     # Get an RFC2047 compliant version of the name
  96     rfc2047_name = rfc2047_encode(name)
  97
  98     # Force the name to be UTF-8
  99     name = force_to_utf8(name)
 100
 101     if name.find(',') != -1 or name.find('.') != -1:
 102         rfc822_maint = "%s (%s)" % (email, name)
 103         rfc2047_maint = "%s (%s)" % (email, rfc2047_name)
 104     else:
 105         rfc822_maint = "%s <%s>" % (name, email)
 106         rfc2047_maint = "%s <%s>" % (rfc2047_name, email)
 107
 108     if email.find("@") == -1 and email.find("buildd_") != 0:
 109         raise ParseMaintError, "No @ found in email address part."
 110
 111     return (rfc822_maint, rfc2047_maint, name, email)
 112
 113 ################################################################################