Treat filename '-' as stdin/stdout in edit

Signed-off-by: David Kågedal <davidk@lysator.liu.se>
Signed-off-by: Karl Hasselström <kha@treskal.com>

diff --git a/stgit/commands/edit.py b/stgit/commands/edit.py
index ec2c4aec9dfe75304fa370db5e3c92f4196fb4d0..4d1475f3b37f9b755a5c1dd8614db6dc504ec704 100644 (file)
--- a/stgit/commands/edit.py
+++ b/stgit/commands/edit.py
@@ -96,7 +96,10 @@ def __update_patch(pname, fname, options):
     bottom = patch.get_bottom()
     top = patch.get_top()
 
     bottom = patch.get_bottom()
     top = patch.get_top()
 

-    f = open(fname)
+    if fname == '-':
+        f = sys.stdin
+    else:
+        f = open(fname)

     message, author_name, author_email, author_date, diff = parse_patch(f)
     f.close()
 
     message, author_name, author_email, author_date, diff = parse_patch(f)
     f.close()
 


@@ -171,9 +174,12 @@ def __generate_file(pname, fname, options):
     text = tmpl % tmpl_dict
 
     # write the file to be edited
     text = tmpl % tmpl_dict
 
     # write the file to be edited

-    f = open(fname, 'w+')
-    f.write(text)
-    f.close()
+    if fname == '-':
+        sys.stdout.write(text)
+    else:
+        f = open(fname, 'w+')
+        f.write(text)
+        f.close()

 
 def __edit_update_patch(pname, options):
     """Edit the given patch interactively.
 
 def __edit_update_patch(pname, options):
     """Edit the given patch interactively.