Changeset 36:f6e3e798fa2a in misc for tictac.py

Timestamp:

09/02/07 00:29:48 (5 years ago)

Author:

Eriol

Branch:

misc

Convert:

svn:e60e002c-0983-44b9-b2ae-8842d539f768/misc@39

Message:

Added interactive mode

File:

• tictac.py (modified) (2 diffs)

tictac.py

@@ -24 +24 @@
 ##
 
+import atexit
 import os.path
+import StringIO
 
 from optparse import OptionParser
-from sys import stdout
+from sys import stdin, stdout
 
 def backread(afile, buffer_size=1024):
@@ -45 +47 @@
         yield ''.join(reversed(data))
 
+def _print(data):
+    for l in backread(data):
+        stdout.write(l)
+    stdout.write('\n')
+
 def main():
-    usage = 'usage: %prog file'
+    interactive = False
+    data = StringIO.StringIO()
+    usage = '''usage: %prog [FILE]
+
+    With no FILE read standard input
+'''
     parser = OptionParser(usage)
     (options, args) = parser.parse_args()
 
-    if len(args) != 1:
-        parser.error('Incorrect number of arguments')
+    if len(args) < 1:
+        interactive = True
     elif not os.path.isfile(args[0]):
         parser.error('No such file')
 
-    fin = open(args[0], 'r')
-    try:
-        for l in backread(fin):
-            stdout.write(l)
-        stdout.write('\n')
-    finally:
-        fin.close()
+    if interactive:
+        while True:
+            line = stdin.readline()
+            if not line: break
+            data.write(line)
+        atexit.register(_print, data)
+    else:
+        fin = open(args[0], 'r')
+        try:
+            _print(fin)
+        finally:
+            fin.close()
 
 if __name__ == '__main__':