From 3b8e1604e8dd4c871e5e8a22d6c738c7f2d39f3a Mon Sep 17 00:00:00 2001
From: Guido van Rossum <guido@python.org>
Date: Thu, 10 Aug 1995 19:26:37 +0000
Subject: [PATCH] upgdaded the test program

---
 Lib/base64.py | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/Lib/base64.py b/Lib/base64.py
index 2f9bdc1340e..55f01ced092 100755
--- a/Lib/base64.py
+++ b/Lib/base64.py
@@ -109,11 +109,25 @@ def decodestring(s):
 # Small test program, reads stdin, writes stdout.
 # no arg: encode, any arg: decode.
 def test():
-	import sys
-	if sys.argv[1:]:
-		decode(sys.stdin, sys.stdout)
+	import sys, getopt
+	try:
+		opts, args = getopt.getopt(sys.argv[1:], 'deut')
+	except getopt.error, msg:
+		print msg
+		print """usage: basd64 [-d] [-e] [-u] [-t] [file|-]
+		-d, -u: decode
+		-e: encode (default)
+		-t: decode string 'Aladdin:open sesame'"""
+	func = encode
+	for o, a in opts:
+		if o == '-e': func = encode
+		if o == '-d': func = decode
+		if o == '-u': func = decode
+		if o == '-t': test1(); return
+	if args and args[0] != '-':
+		func(open(args[0]), sys.stdout)
 	else:
-		encode(sys.stdin, sys.stdout)
+		func(sys.stdin, sys.stdout)
 
 def test1():
 	s0 = "Aladdin:open sesame"