From: Oleg Broytman <phd@phdru.name>
Date: Thu, 30 Jan 2014 18:15:06 +0000 (+0400)
Subject: Add an optional output file argument
X-Git-Tag: v2.3.2~25
X-Git-Url: https://git.phdru.name/?a=commitdiff_plain;h=dafc33620aac47c26b3997a10aa4bcf72059373a;p=mimedecode.git

Add an optional output file argument
---

diff --git a/mimedecode.docbook b/mimedecode.docbook
index 2b51ec0..cc8ee2b 100644
--- a/mimedecode.docbook
+++ b/mimedecode.docbook
@@ -38,7 +38,9 @@
       <arg choice="opt">
          <option>-beit mask</option>
       </arg>
-      <arg choice="opt">filename</arg>
+      <arg choice="opt">input_file
+        <arg choice="opt">output_file</arg>
+      </arg>
    </cmdsynopsis>
 </refsynopsisdiv>
 
diff --git a/mimedecode.py b/mimedecode.py
index 96c3ea5..131d551 100755
--- a/mimedecode.py
+++ b/mimedecode.py
@@ -30,21 +30,21 @@ Broytman mimedecode.py version %s, %s
 def usage(code=0):
     version(0)
     sys.stdout.write("""\
-Usage: %s [-h|--help] [-V|--version] [-cCDP] [-f charset] [-d header] [-p header:param] [-beit mask] [filename]
+Usage: %s [-h|--help] [-V|--version] [-cCDP] [-f charset] [-d header] [-p header:param] [-beit mask] [input_file [output_file]]
 """ % me)
     sys.exit(code)
 
 
-def output(s, outfile=sys.stdout):
-    outfile.write(s)
+def output(s):
+    gopts.outfile.write(s)
 
-def output_headers(msg, outfile=sys.stdout):
+def output_headers(msg):
     unix_from = msg.get_unixfrom()
     if unix_from:
         output(unix_from + '\n')
     for key, value in msg.items():
-        output("%s: %s\n" % (key, value), outfile)
-    output("\n", outfile) # End of headers
+        output("%s: %s\n" % (key, value))
+    output("\n") # End of headers
 
 
 def recode(s, charset):
@@ -302,7 +302,7 @@ class GlobalOptions:
 gopts = GlobalOptions
 
 
-def init():
+def get_opt():
     from getopt import getopt, GetoptError
 
     try:
@@ -349,14 +349,28 @@ def init():
 
 
 if __name__ == "__main__":
-    arguments = init()
+    arguments = get_opt()
 
     la = len(arguments)
-    if la >= 2:
-        usage(1)
-    if (la == 0) or (arguments[0] == '-'):
+    if la == 0:
         infile = sys.stdin
+    elif la in (1, 2):
+        if (arguments[0] == '-'):
+            infile = sys.stdin
+        else:
+            infile = open(arguments[0], 'r')
+        if la == 1:
+            outfile = sys.stdout
+        elif la == 2:
+            if (arguments[1] == '-'):
+                outfile = sys.stdout
+            else:
+                outfile = open(arguments[1], 'w')
     else:
-        infile = open(arguments[0], 'r')
+        usage(1)
 
+    gopts.outfile = outfile
     decode_file(infile)
+
+    infile.close()
+    outfile.close()