Fix the branch protect/unprotect message

Signed-off-by: Catalin Marinas <catalin.marinas@gmail.com>

diff --git a/stgit/commands/branch.py b/stgit/commands/branch.py
index 56a5fe608634d8af075658f4ffebabbc2698dbaf..c731a6987951b23bf8cfd5e3984784668e16c3ea 100644 (file)
--- a/stgit/commands/branch.py
+++ b/stgit/commands/branch.py
@@ -161,8 +161,11 @@ def func(parser, options, args):
         if not os.path.isfile(base):
             raise CmdException, 'Branch "%s" is not controlled by StGit' % branch
 
         if not os.path.isfile(base):
             raise CmdException, 'Branch "%s" is not controlled by StGit' % branch
 

-        print 'Protecting branch "%s"...' % branch
+        print 'Protecting branch "%s"...' % branch,
+        sys.stdout.flush()

         stack.Series(branch).protect()
         stack.Series(branch).protect()

+        print 'done'
+

         return
 
     elif options.rename:
         return
 
     elif options.rename:


@@ -185,8 +188,11 @@ def func(parser, options, args):
         if not os.path.isfile(base):
             raise CmdException, 'Branch "%s" is not controlled by StGit' % branch
 
         if not os.path.isfile(base):
             raise CmdException, 'Branch "%s" is not controlled by StGit' % branch
 

-        print 'Unprotecting branch "%s"...' % branch
+        print 'Unprotecting branch "%s"...' % branch,
+        sys.stdout.flush()

         stack.Series(branch).unprotect()
         stack.Series(branch).unprotect()

+        print 'done'
+

         return
 
     elif len(args) == 1:
         return
 
     elif len(args) == 1: