diff options
Diffstat (limited to 'contrib/fb303/py/fb303_scripts/fb303_simple_mgmt.py')
| -rw-r--r-- | contrib/fb303/py/fb303_scripts/fb303_simple_mgmt.py | 39 |
1 files changed, 19 insertions, 20 deletions
diff --git a/contrib/fb303/py/fb303_scripts/fb303_simple_mgmt.py b/contrib/fb303/py/fb303_scripts/fb303_simple_mgmt.py index 4f8ce9933..4b1c25728 100644 --- a/contrib/fb303/py/fb303_scripts/fb303_simple_mgmt.py +++ b/contrib/fb303/py/fb303_scripts/fb303_simple_mgmt.py @@ -19,7 +19,8 @@ # under the License. # -import sys, os +import sys +import os from optparse import OptionParser from thrift.Thrift import * @@ -31,11 +32,12 @@ from thrift.protocol import TBinaryProtocol from fb303 import * from fb303.ttypes import * + def service_ctrl( - command, - port, - trans_factory = None, - prot_factory = None): + command, + port, + trans_factory=None, + prot_factory=None): """ service_ctrl is a generic function to execute standard fb303 functions @@ -66,19 +68,19 @@ def service_ctrl( return 3 # scalar commands - if command in ["version","alive","name"]: + if command in ["version", "alive", "name"]: try: - result = fb303_wrapper(command, port, trans_factory, prot_factory) + result = fb303_wrapper(command, port, trans_factory, prot_factory) print result return 0 except: - print "failed to get ",command + print "failed to get ", command return 3 # counters if command in ["counters"]: try: - counters = fb303_wrapper('counters', port, trans_factory, prot_factory) + counters = fb303_wrapper('counters', port, trans_factory, prot_factory) for counter in counters: print "%s: %d" % (counter, counters[counter]) return 0 @@ -86,11 +88,10 @@ def service_ctrl( print "failed to get counters" return 3 - # Only root should be able to run the following commands if os.getuid() == 0: # async commands - if command in ["stop","reload"] : + if command in ["stop", "reload"]: try: fb303_wrapper(command, port, trans_factory, prot_factory) return 0 @@ -98,23 +99,21 @@ def service_ctrl( print "failed to tell the service to ", command return 3 else: - if command in ["stop","reload"]: + if command in ["stop", "reload"]: print "root privileges are required to stop or reload the service." return 4 print "The following commands are available:" - for command in ["counters","name","version","alive","status"]: + for command in ["counters", "name", "version", "alive", "status"]: print "\t%s" % command print "The following commands are available for users with root privileges:" - for command in ["stop","reload"]: + for command in ["stop", "reload"]: print "\t%s" % command + return 0 - return 0; - - -def fb303_wrapper(command, port, trans_factory = None, prot_factory = None): +def fb303_wrapper(command, port, trans_factory=None, prot_factory=None): sock = TSocket.TSocket('localhost', port) # use input transport factory if provided @@ -179,11 +178,11 @@ def main(): # parse command line options parser = OptionParser() - commands=["stop","counters","status","reload","version","name","alive"] + commands = ["stop", "counters", "status", "reload", "version", "name", "alive"] parser.add_option("-c", "--command", dest="command", help="execute this API", choices=commands, default="status") - parser.add_option("-p","--port",dest="port",help="the service's port", + parser.add_option("-p", "--port", dest="port", help="the service's port", default=9082) (options, args) = parser.parse_args() |