X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=mimedecode.py;h=1e3bf74b3964717de33883c8b160c19db805f664;hb=63f035c2a1e63b84155a41adbaadfc842615a1d6;hp=3a8cbbe533bbd29f4e7589b40ee44d8b635fa8bd;hpb=44f274a71a4b30c8d6f54b7ea374c8d7f8e95eed;p=mimedecode.git diff --git a/mimedecode.py b/mimedecode.py index 3a8cbbe..1e3bf74 100755 --- a/mimedecode.py +++ b/mimedecode.py @@ -18,7 +18,7 @@ Broytman mimedecode.py version %s, %s 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] [-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 + '\n') @@ -266,6 +266,33 @@ def totext(msg, instring): output_headers(msg) output(s) + return s + + +def _save_message(msg, outstring, save_headers=False, save_body=False): + for header, param in ( + ("Content-Disposition", "filename"), + ("Content-Type", "name"), + ): + fname = msg.get_param(param, header=header) + if fname: + fname = '-' + fname + break + else: + fname = '' + g.save_counter += 1 + fname = str(g.save_counter) + fname + + global output + save_output = output + outfile = open(os.path.join(g.destination_dir, fname), 'w') + output = outfile.write + if save_headers: + output_headers(msg) + if save_body: + output(outstring) + outfile.close() + output = save_output def decode_part(msg): @@ -285,7 +312,10 @@ def decode_part(msg): left_binary = False for content_type in masks: - if content_type in g.binary_mask: + if content_type in g.totext_mask or \ + content_type in g.decoded_binary_mask: + break + elif content_type in g.binary_mask: left_binary = True break @@ -299,23 +329,30 @@ def decode_part(msg): for content_type in masks: if content_type in g.totext_mask: - totext(msg, outstring) - return + outstring = totext(msg, outstring) + break elif content_type in g.binary_mask or \ content_type in g.decoded_binary_mask: output_headers(msg) output(outstring) - return + break elif content_type in g.ignore_mask: output_headers(msg) - output("\nMessage body of type `%s' skipped.\n" % content_type) - return + output("\nMessage body of type %s skipped.\n" % ctype) + break elif content_type in g.error_mask: - raise ValueError, "content type `%s' prohibited" % content_type - - # Neither content type nor masks were listed - decode by default - totext(msg, outstring) + raise ValueError, "content type %s prohibited" % ctype + else: + # Neither content type nor masks were listed - decode by default + outstring = totext(msg, outstring) + for content_type in masks: + if content_type in g.save_headers_mask: + _save_message(msg, outstring, save_headers=True, save_body=False) + elif content_type in g.save_body_mask: + _save_message(msg, outstring, save_headers=False, save_body=True) + elif content_type in g.save_message_mask: + _save_message(msg, outstring, save_headers=True, save_body=True) def decode_multipart(msg): "Decode multipart" @@ -392,8 +429,14 @@ class GlobalOptions: ignore_mask = [] # Ignore (skip, do not decode and do not include into output) error_mask = [] # Raise error if encounter one of these + save_counter = 0 + save_headers_mask = [] + save_body_mask = [] + save_message_mask = [] + input_filename = None output_filename = None + destination_dir = os.curdir g = GlobalOptions @@ -403,8 +446,10 @@ def get_opts(): try: options, arguments = getopt(sys.argv[1:], - 'hVcCDPH:f:d:p:r:R:b:B:e:i:t:o:', - ['help', 'version', 'host=', 'set-header=', 'set-param=']) + 'hVcCDPH:f:d:p:r:R:b:B:e:i:t:O:o:', + ['help', 'version', 'host=', + 'save-headers=', 'save-body=', 'save-message=', + 'set-header=', 'set-param=']) except GetoptError: usage(1) @@ -454,6 +499,14 @@ def get_opts(): g.ignore_mask.append(value) elif option == '-e': g.error_mask.append(value) + elif option == '--save-headers': + g.save_headers_mask.append(value) + elif option == '--save-body': + g.save_body_mask.append(value) + elif option == '--save-message': + g.save_message_mask.append(value) + elif option == '-O': + g.destination_dir = value elif option == '-o': g.output_filename = value else: @@ -470,7 +523,7 @@ if __name__ == "__main__": g.input_filename = '-' infile = sys.stdin if g.output_filename: - outfile = open(g.output_filename, 'w') + outfile = open(os.path.join(g.destination_dir, g.output_filename), 'w') else: g.output_filename = '-' outfile = sys.stdout @@ -483,7 +536,7 @@ if __name__ == "__main__": infile = open(arguments[0], 'r') if la == 1: if g.output_filename: - outfile = open(g.output_filename, 'w') + outfile = open(os.path.join(g.destination_dir, g.output_filename), 'w') else: g.output_filename = '-' outfile = sys.stdout @@ -495,7 +548,7 @@ if __name__ == "__main__": outfile = sys.stdout else: g.output_filename = arguments[1] - outfile = open(arguments[1], 'w') + outfile = open(os.path.join(g.destination_dir, g.output_filename), 'w') else: usage(1, 'Too many arguments')