]> git.decadent.org.uk Git - dak.git/blob - daklib/textutils.py
remove UserExtensions
[dak.git] / 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 ################################################################################