def usage(code=0, errormsg=''):
version(0)
sys.stdout.write("""\
- Usage: %s [-h|--help] [-V|--version] [-cCDP] [-H|--host=hostname] [-f charset] [-d header1[,h2,...]|*[,-h1,...]] [-p header1[,h2,h3,...]:param1[,p2,p3,...]] [-r header1[,h2,...]|*[,-h1,...]] [-R header1[,h2,h3,...]:param1[,p2,p3,...]] [--set-header header:value] [--set-param header:param=value] [-Bbeit mask] [--save-headers|body|message mask] [-O dest_dir] [-o output_file] [input_file [output_file]]
+Usage: %s [-h|--help] [-V|--version] [-cCDP] [-H|--host=hostname] [-f charset] [-d header1[,h2,...]|*[,-h1,...]] [-p header1[,h2,h3,...]:param1[,p2,p3,...]] [-r header1[,h2,...]|*[,-h1,...]] [-R header1[,h2,h3,...]:param1[,p2,p3,...]] [--set-header header:value] [--set-param header:param=value] [-Bbeit mask] [--save-headers|body|message mask] [-O dest_dir] [-o output_file] [input_file [output_file]]
""" % me)
if errormsg:
sys.stderr.write(errormsg + os.linesep)
def output_headers(msg):
unix_from = msg.get_unixfrom()
if unix_from:
- output(unix_from + os.linesep)
+ output(unix_from)
+ output(os.linesep)
for key, value in msg.items():
- output("%s: %s%s" % (key, value, os.linesep))
+ output(key)
+ output(": ")
+ value = value.split(';', 1)
+ output(value[0])
+ if len(value) == 2:
+ output(";")
+ output(_decode_header(value[1], strip=False))
+ output(os.linesep)
output(os.linesep) # End of headers
def recode_if_needed(s, charset):
- if charset and charset.lower() != g.default_encoding:
- s = unicode(s, charset, "replace").encode(g.default_encoding, "replace")
+ if bytes is str: # Python2
+ if isinstance(s, bytes) and \
+ charset and charset.lower() != g.default_encoding:
+ s = s.decode(charset, "replace").\
+ encode(g.default_encoding, "replace")
+ else: # Python3
+ if isinstance(s, bytes):
+ s = s.decode(charset, "replace")
return s
-def _decode_header(s):
+def _decode_header(s, strip=True):
"""Return a decoded string according to RFC 2047.
NOTE: This is almost the same as email.Utils.decode.
"""
rtn = []
for atom, charset in L:
- if charset is None:
- rtn.append(atom)
- else:
- rtn.append(recode_if_needed(atom, charset))
- rtn.append(' ')
- del rtn[-1] # remove the last space
+ atom = recode_if_needed(atom, charset or g.default_encoding)
+ if strip:
+ atom = atom.strip()
+ rtn.append(atom)
# Now that we've decoded everything, we just need to join all the parts
# together into the final string.
- return ''.join(rtn)
+ return ' '.join(rtn)
def decode_header(msg, header):
"Decode mail header (if exists) and put it back, if it was encoded"
caps = mailcap.getcaps()
content_type = msg.get_content_type()
+ if content_type.startswith('text/'):
+ charset = msg.get_content_charset()
+ else:
+ charset = None
filename = tempfile.mktemp()
command = None
return s
outfile = open(filename, 'wb')
+ if charset and isinstance(s, bytes):
+ s = s.decode(charset, "replace")
+ if not isinstance(s, bytes):
+ s = s.encode(g.default_encoding, "replace")
outfile.write(s)
outfile.close()
pipe = os.popen(command, 'r')
- s = pipe.read()
- pipe.close()
+ new_s = pipe.read()
+ if pipe.close() is None: # result=0, Ok
+ s = new_s
os.remove(filename)
set_content_type(msg, "text/plain")
- msg["X-MIME-Autoconverted"] = "from %s to text/plain by %s id %s" % (content_type, g.host_name, command.split()[0])
+ if s is new_s:
+ msg["X-MIME-Autoconverted"] = "from %s to text/plain by %s id %s" % (content_type, g.host_name, command.split()[0])
+ else:
+ msg["X-MIME-Autoconverted"] = "failed conversion from %s to text/plain by %s id %s" % (content_type, g.host_name, command.split()[0])
return s
global output
save_output = output
outfile = open_output_file(fname)
- output = outfile.write
+ def _output_bytes(s):
+ if not isinstance(s, bytes):
+ s = s.encode(g.default_encoding, "replace")
+ outfile.write(s)
+ output = _output_bytes
if save_headers:
output_headers(msg)
if save_body:
g.host_name = socket.gethostname()
g.outfile = outfile
- output = outfile.write
+ if hasattr(outfile, 'buffer'):
+ def output_bytes(s):
+ if not isinstance(s, bytes):
+ s = s.encode(g.default_encoding, "replace")
+ outfile.buffer.write(s)
+ output = output_bytes
+ else:
+ output = outfile.write
import email
msg = email.message_from_file(infile)