From aec1d1eecc9c2f9da2c10e4639135d89c3a6b774 Mon Sep 17 00:00:00 2001 From: Brian Warner Date: Tue, 5 Dec 2006 12:25:23 -0700 Subject: [PATCH] add bin/allmydata to create/stop/start nodes --- allmydata/scripts/__init__.py | 0 allmydata/scripts/runner.py | 160 ++++++++++++++++++++++++++++++++++ bin/allmydata | 4 + setup.py | 3 +- 4 files changed, 166 insertions(+), 1 deletion(-) create mode 100644 allmydata/scripts/__init__.py create mode 100644 allmydata/scripts/runner.py create mode 100644 bin/allmydata diff --git a/allmydata/scripts/__init__.py b/allmydata/scripts/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/allmydata/scripts/runner.py b/allmydata/scripts/runner.py new file mode 100644 index 00000000..ceea9c51 --- /dev/null +++ b/allmydata/scripts/runner.py @@ -0,0 +1,160 @@ +#! /usr/bin/python + +import os, sys, signal, time +from twisted.python import usage + +class StartOptions(usage.Options): + optParameters = [ + ["basedir", "C", ".", "which directory to start the node in"], + ] + +class StopOptions(usage.Options): + optParameters = [ + ["basedir", "C", ".", "which directory to stop the node in"], + ] + +class RestartOptions(usage.Options): + optParameters = [ + ["basedir", "C", ".", "which directory to restart the node in"], + ] + +class CreateClientOptions(usage.Options): + optParameters = [ + ["basedir", "C", ".", "which directory to create the client in"], + ] +class CreateQueenOptions(usage.Options): + optParameters = [ + ["basedir", "C", ".", "which directory to create the queen in"], + ] + +client_tac = """ +# -*- python -*- + +from allmydata import client +from twisted.application import service + +c = client.Client() + +application = service.Application("allmydata_client") +c.setServiceParent(application) +""" + +queen_tac = """ +# -*- python -*- + +from allmydata import queen +from twisted.application import service + +c = queen.Queen() + +application = service.Application("allmydata_queen") +c.setServiceParent(application) +""" + +class Options(usage.Options): + synopsis = "Usage: allmydata [command options]" + + subcommands = [ + ["create-client", None, CreateClientOptions], + ["create-queen", None, CreateQueenOptions], + ["start", None, StartOptions], + ["stop", None, StopOptions], + ["restart", None, RestartOptions], + ] + + def postOptions(self): + if not hasattr(self, 'subOptions'): + raise usage.UsageError("must specify a command") + +def run(): + config = Options() + try: + config.parseOptions() + except usage.error, e: + print "%s: %s" % (sys.argv[0], e) + print + c = getattr(config, 'subOptions', config) + print str(c) + sys.exit(1) + + command = config.subCommand + so = config.subOptions + + if command == "create-client": + rc = create_client(so) + elif command == "create-queen": + rc = create_queen(so) + elif command == "start": + rc = start(so) + elif command == "stop": + rc = stop(so) + elif command == "restart": + rc = restart(so) + rc = rc or 0 + sys.exit(rc) + +def create_client(config): + basedir = config['basedir'] + os.mkdir(basedir) + f = open(os.path.join(basedir, "client.tac"), "w") + f.write(client_tac) + f.close() + print "client created, please copy roster_pburl into the directory" + +def create_queen(config): + basedir = config['basedir'] + os.mkdir(basedir) + f = open(os.path.join(basedir, "queen.tac"), "w") + f.write(queen_tac) + f.close() + print "queen created" + +def start(config): + basedir = config['basedir'] + if os.path.exists(os.path.join(basedir, "client.tac")): + tac = "client.tac" + type = "client" + elif os.path.exists(os.path.join(basedir, "queen.tac")): + tac = "queen.tac" + type = "queen" + else: + print "%s does not look like a node directory" % basedir + sys.exit(1) + os.chdir(basedir) + rc = os.system("twistd -y %s" % tac) + if rc == 0: + print "node probably started" + else: + print "node probably not started" + return 1 + +def stop(config): + basedir = config['basedir'] + pidfile = os.path.join(basedir, "twistd.pid") + if not os.path.exists(pidfile): + print "%s does not look like a running node directory (no twistd.pid)" % basedir + return 1 + pid = open(pidfile, "r").read() + pid = int(pid) + + timer = 0 + os.kill(pid, signal.TERM) + time.sleep(0.1) + while timer < 5: + # poll once per second until twistd.pid goes away, up to 5 seconds + try: + os.kill(pid, 0) + except OSError: + print "process %d is dead" % pid + return + timer += 1 + time.sleep(1) + print "never saw process go away" + return 1 + +def restart(config): + rc = stop(config) + if rc: + print "not restarting" + return rc + return start(config) diff --git a/bin/allmydata b/bin/allmydata new file mode 100644 index 00000000..57d4ebc6 --- /dev/null +++ b/bin/allmydata @@ -0,0 +1,4 @@ +#!/usr/bin/python + +from allmydata.scripts import runner +runner.run() diff --git a/setup.py b/setup.py index c6e55d27..bb384b24 100644 --- a/setup.py +++ b/setup.py @@ -7,7 +7,8 @@ setup( name="AllMyData", version="0.0.1", #packages=find_packages('.'), - packages=["allmydata", "allmydata/test", "allmydata/util"], + packages=["allmydata", "allmydata/test", "allmydata/util", "allmydata/scripts"], + scripts = ["bin/allmydata"], package_data={ 'allmydata': ['web/*.xhtml'] }, description="AllMyData (tahoe2)", ) -- 2.45.2