ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-02-20 17:52:50 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

cli: add --profile flag

Browse Source
This commit is contained in:
Zooko O'Whielacronx 2007-09-21 15:03:47 -07:00
parent 20079e3019
commit 3e4616f4ad
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/allmydata/scripts/startstop_node.py
View File

@ -9,6 +9,9 @@ class StartOptions(BasedirMixin, usage.Options):
optParameters = [
["basedir", "C", None, "which directory to start the node in"],
]
optFlags = [
["profile", "p", "whether to run under the Python profiler, putting results in \"profiling_results.prof\""],
]
class StopOptions(BasedirMixin, usage.Options):
optParameters = [
@ -25,7 +28,7 @@ class RestartOptions(BasedirMixin, usage.Options):
"of 'restart'"],
]
def do_start(basedir, out=sys.stdout, err=sys.stderr):
def do_start(basedir, profile=False, out=sys.stdout, err=sys.stderr):
print >>out, "STARTING", basedir
if os.path.exists(os.path.join(basedir, "client.tac")):
tac = "client.tac"
@ -53,6 +56,8 @@ def do_start(basedir, out=sys.stdout, err=sys.stderr):
fileutil.make_dirs(os.path.join(basedir, "logs"))
cmd.extend(["-y", tac, "--logfile", os.path.join("logs", "twistd.log")])
if profile:
cmd.extend(["--profile=profiling_results.prof", "--savestats",])
curdir = os.getcwd()
try:
os.chdir(basedir)
@ -100,7 +105,7 @@ def do_stop(basedir, out=sys.stdout, err=sys.stderr):
def start(config, stdout, stderr):
rc = 0
for basedir in config['basedirs']:
rc = do_start(basedir, stdout, stderr) or rc
rc = do_start(basedir, config['profile'], stdout, stderr) or rc
return rc
def stop(config, stdout, stderr):