Index: utils/set_tree_status.py |
diff --git a/utils/set_tree_status.py b/utils/set_tree_status.py |
index a189503bf17c2ec8568bb477dda21fb07b7d5ff7..d140eb7ff258bd92739d39914bf06cba19bca977 100755 |
--- a/utils/set_tree_status.py |
+++ b/utils/set_tree_status.py |
@@ -7,9 +7,10 @@ |
# password protected authorization. |
# |
# Example usage: |
-# ./set_tree_status.py "a quoted space separated message." |
+# ./set_tree_status.py [options] "a quoted space separated message." |
import getpass |
+import optparse |
import os |
import sys |
import urllib |
@@ -30,12 +31,13 @@ def get_pwd(): |
return getpass.getpass() |
-def post_status(message): |
- status = get_status() |
- if 'tree is closed' in status.lower(): |
- print >> sys.stderr, 'Tree is already closed for some other reason.' |
- print >> sys.stderr, status |
- return -1 |
+def post_status(force, message): |
+ if not force: |
+ status = get_status() |
+ if 'tree is closed' in status.lower(): |
+ print >> sys.stderr, 'Tree is already closed for some other reason.' |
+ print >> sys.stderr, status |
+ return -1 |
data = { |
'message': message, |
'username': getpass.getuser(), |
@@ -46,8 +48,12 @@ def post_status(message): |
if __name__ == '__main__': |
- if len(sys.argv) != 2: |
- print('Usage: set_tree_status.py "message"') |
- sys.exit(1) |
- |
- sys.exit(post_status(sys.argv[1])) |
+ parser = optparse.OptionParser("%prog [options] quoted_message") |
+ parser.add_option('--noforce', |
+ dest='force', action='store_false', |
+ default=True, |
+ help='Dont force to close tree if it is already closed.') |
+ options, args = parser.parse_args() |
+ if not args: |
+ print >> sys.stderr, 'missing tree close message.' |
+ sys.exit(post_status(options.force, args[0])) |