From 6048cfad2518322aa5788fb68356093033f186f0 Mon Sep 17 00:00:00 2001 From: Mathias Rav Date: Sat, 2 May 2015 10:40:21 +0200 Subject: [PATCH] Add -x to delete sane messages from the insane folder --- insane.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/insane.py b/insane.py index a89d14a..fe73134 100644 --- a/insane.py +++ b/insane.py @@ -1,5 +1,6 @@ import os import email +import argparse from io import BytesIO, StringIO from email.generator import Generator @@ -8,6 +9,10 @@ def main(): + parser = argparse.ArgumentParser() + parser.add_argument('-x', '--delete', action='store_true') + args = parser.parse_args() + for filename in sorted(os.listdir('insane')): if filename.endswith('.in'): with open('insane/%s' % filename, 'rb') as fp: @@ -45,6 +50,10 @@ def main(): else: print("%s: Not OK" % (filename,)) + if args.delete and (a == b or a_s == b_s): + os.remove('insane/%s' % filename) + os.remove('insane/%s' % filename.replace('.in', '.out')) + if __name__ == "__main__": main()