--- /dev/null
+
+import os, sys
+from twisted.python import usage
+from allmydata.scripts.common import NoDefaultBasedirMixin
+
+class CreateClientOptions(NoDefaultBasedirMixin, usage.Options):
+ optParameters = [
+ ["basedir", "C", None, "which directory to create the client in"],
+ ]
+
+class CreateIntroducerOptions(NoDefaultBasedirMixin, usage.Options):
+ optParameters = [
+ ["basedir", "C", None, "which directory to create the introducer in"],
+ ]
+
+client_tac = """
+# -*- python -*-
+
+from allmydata import client
+from twisted.application import service
+
+c = client.Client()
+
+application = service.Application("allmydata_client")
+c.setServiceParent(application)
+"""
+
+introducer_tac = """
+# -*- python -*-
+
+from allmydata import introducer_and_vdrive
+from twisted.application import service
+
+c = introducer_and_vdrive.IntroducerAndVdrive()
+
+application = service.Application("allmydata_introducer")
+c.setServiceParent(application)
+"""
+
+def create_client(basedir, config, out=sys.stdout, err=sys.stderr):
+ if os.path.exists(basedir):
+ if os.listdir(basedir):
+ print >>err, "The base directory already exists: %s" % basedir
+ print >>err, "To avoid clobbering anything, I am going to quit now"
+ print >>err, "Please use a different directory, or delete this one"
+ return -1
+ # we're willing to use an empty directory
+ else:
+ os.mkdir(basedir)
+ f = open(os.path.join(basedir, "client.tac"), "w")
+ f.write(client_tac)
+ f.close()
+ print >>out, "client created in %s" % basedir
+ print >>out, " please copy introducer.furl and vdrive.furl into the directory"
+
+def create_introducer(basedir, config, out=sys.stdout, err=sys.stderr):
+ if os.path.exists(basedir):
+ if os.listdir(basedir):
+ print >>err, "The base directory already exists: %s" % basedir
+ print >>err, "To avoid clobbering anything, I am going to quit now"
+ print >>err, "Please use a different directory, or delete this one"
+ return -1
+ # we're willing to use an empty directory
+ else:
+ os.mkdir(basedir)
+ f = open(os.path.join(basedir, "introducer.tac"), "w")
+ f.write(introducer_tac)
+ f.close()
+ print >>out, "introducer created in %s" % basedir
+
+subCommands = [
+ ["create-client", None, CreateClientOptions, "Create a client node."],
+ ["create-introducer", None, CreateIntroducerOptions, "Create a introducer node."],
+
+]
+
+dispatch = {
+ "create-client": create_client,
+ "create-introducer": create_introducer,
+ }
-import os, subprocess, sys, signal, time
+import sys
from cStringIO import StringIO
from twisted.python import usage
-from twisted.python.procutils import which
-from allmydata.scripts import debug
-from allmydata.scripts.common import BasedirMixin, NoDefaultBasedirMixin
-
-def testtwistd(loc):
- try:
- return subprocess.call(["python", loc,], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
- except:
- return -1
-
-twistd = None
-if not twistd:
- for maybetwistd in which("twistd"):
- ret = testtwistd(maybetwistd)
- if ret == 0:
- twistd = maybetwistd
- break
-
-if not twistd:
- for maybetwistd in which("twistd.py"):
- ret = testtwistd(maybetwistd)
- if ret == 0:
- twistd = maybetwistd
- break
-
-if not twistd:
- maybetwistd = os.path.join(sys.prefix, 'Scripts', 'twistd')
- ret = testtwistd(maybetwistd)
- if ret == 0:
- twistd = maybetwistd
-
-if not twistd:
- maybetwistd = os.path.join(sys.prefix, 'Scripts', 'twistd.py')
- ret = testtwistd(maybetwistd)
- if ret == 0:
- twistd = maybetwistd
-
-if not twistd:
- print "Can't find twistd (it comes with Twisted). Aborting."
- sys.exit(1)
-
-class StartOptions(BasedirMixin, usage.Options):
- optParameters = [
- ["basedir", "C", None, "which directory to start the node in"],
- ]
-
-class StopOptions(BasedirMixin, usage.Options):
- optParameters = [
- ["basedir", "C", None, "which directory to stop the node in"],
- ]
-
-class RestartOptions(BasedirMixin, usage.Options):
- optParameters = [
- ["basedir", "C", None, "which directory to restart the node in"],
- ]
- optFlags = [
- ["force", "f", "if the node is not already running, start it "
- "instead of complaining that you should have used 'start' instead "
- "of 'restart'"],
- ]
-
-class CreateClientOptions(NoDefaultBasedirMixin, usage.Options):
- optParameters = [
- ["basedir", "C", None, "which directory to create the client in"],
- ]
-
-class CreateIntroducerOptions(NoDefaultBasedirMixin, usage.Options):
- optParameters = [
- ["basedir", "C", None, "which directory to create the introducer in"],
- ]
-
-client_tac = """
-# -*- python -*-
-
-from allmydata import client
-from twisted.application import service
-
-c = client.Client()
-
-application = service.Application("allmydata_client")
-c.setServiceParent(application)
-"""
-
-introducer_tac = """
-# -*- python -*-
-
-from allmydata import introducer_and_vdrive
-from twisted.application import service
-
-c = introducer_and_vdrive.IntroducerAndVdrive()
-
-application = service.Application("allmydata_introducer")
-c.setServiceParent(application)
-"""
+from allmydata.scripts import debug, create_node, startstop_node
class Options(usage.Options):
synopsis = "Usage: allmydata <command> [command options]"
["quiet", "q", "operate silently"],
]
- subCommands = [
- ["create-client", None, CreateClientOptions, "Create a client node."],
- ["create-introducer", None, CreateIntroducerOptions, "Create a introducer node."],
- ["start", None, StartOptions, "Start a node (of any type)."],
- ["stop", None, StopOptions, "Stop a node."],
- ["restart", None, RestartOptions, "Restart a node."],
- ] + debug.subCommands
+ subCommands = []
+ subCommands += create_node.subCommands
+ subCommands += startstop_node.subCommands
+ subCommands += debug.subCommands
def postOptions(self):
if not hasattr(self, 'subOptions'):
stdout = StringIO()
rc = 0
- if command == "create-client":
- for basedir in so.basedirs:
- rc = create_client(basedir, so, stdout, stderr) or rc
- elif command == "create-introducer":
- for basedir in so.basedirs:
- rc = create_introducer(basedir, so, stdout, stderr) or rc
- elif command == "start":
+ if command in create_node.dispatch:
for basedir in so.basedirs:
- rc = start(basedir, so, stdout, stderr) or rc
- elif command == "stop":
- for basedir in so.basedirs:
- rc = stop(basedir, so, stdout, stderr) or rc
- elif command == "restart":
- for basedir in so.basedirs:
- rc = stop(basedir, so, stdout, stderr) or rc
- if rc == 2 and so['force']:
- print >>stderr, "ignoring couldn't-stop"
- rc = 0
- if rc:
- print >>stderr, "not restarting"
- return rc
- for basedir in so.basedirs:
- rc = start(basedir, so, stdout, stderr) or rc
+ f = create_node.dispatch[command]
+ rc = f(basedir, so, stdout, stderr) or rc
+ elif command in startstop_node.dispatch:
+ rc = startstop_node.dispatch[command](so, stdout, stderr)
elif command in debug.dispatch:
rc = debug.dispatch[command](so, stdout, stderr)
def run():
rc = runner(sys.argv[1:])
sys.exit(rc)
-
-def create_client(basedir, config, out=sys.stdout, err=sys.stderr):
- if os.path.exists(basedir):
- if os.listdir(basedir):
- print >>err, "The base directory already exists: %s" % basedir
- print >>err, "To avoid clobbering anything, I am going to quit now"
- print >>err, "Please use a different directory, or delete this one"
- return -1
- # we're willing to use an empty directory
- else:
- os.mkdir(basedir)
- f = open(os.path.join(basedir, "client.tac"), "w")
- f.write(client_tac)
- f.close()
- print >>out, "client created in %s" % basedir
- print >>out, " please copy introducer.furl and vdrive.furl into the directory"
-
-def create_introducer(basedir, config, out=sys.stdout, err=sys.stderr):
- if os.path.exists(basedir):
- if os.listdir(basedir):
- print >>err, "The base directory already exists: %s" % basedir
- print >>err, "To avoid clobbering anything, I am going to quit now"
- print >>err, "Please use a different directory, or delete this one"
- return -1
- # we're willing to use an empty directory
- else:
- os.mkdir(basedir)
- f = open(os.path.join(basedir, "introducer.tac"), "w")
- f.write(introducer_tac)
- f.close()
- print >>out, "introducer created in %s" % basedir
-
-def start(basedir, config, out=sys.stdout, err=sys.stderr):
- print >>out, "STARTING", basedir
- if os.path.exists(os.path.join(basedir, "client.tac")):
- tac = "client.tac"
- type = "client"
- elif os.path.exists(os.path.join(basedir, "introducer.tac")):
- tac = "introducer.tac"
- type = "introducer"
- else:
- print >>err, "%s does not look like a node directory" % basedir
- if not os.path.isdir(basedir):
- print >>err, " in fact, it doesn't look like a directory at all!"
- sys.exit(1)
- rc = subprocess.call(["python", twistd, "-y", tac,], cwd=basedir)
- if rc == 0:
- print >>out, "%s node probably started" % type
- return 0
- else:
- print >>err, "%s node probably not started" % type
- return 1
-
-def stop(basedir, config, out=sys.stdout, err=sys.stderr):
- print >>out, "STOPPING", basedir
- pidfile = os.path.join(basedir, "twistd.pid")
- if not os.path.exists(pidfile):
- print >>err, "%s does not look like a running node directory (no twistd.pid)" % basedir
- return 2
- pid = open(pidfile, "r").read()
- pid = int(pid)
-
- timer = 0
- os.kill(pid, signal.SIGTERM)
- 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 >>out, "process %d is dead" % pid
- return
- timer += 1
- time.sleep(1)
- print >>err, "never saw process go away"
- return 1
--- /dev/null
+
+import os, sys, signal, time, subprocess
+from twisted.python import usage
+from allmydata.scripts.common import BasedirMixin
+from twisted.python.procutils import which
+
+class StartOptions(BasedirMixin, usage.Options):
+ optParameters = [
+ ["basedir", "C", None, "which directory to start the node in"],
+ ]
+
+class StopOptions(BasedirMixin, usage.Options):
+ optParameters = [
+ ["basedir", "C", None, "which directory to stop the node in"],
+ ]
+
+class RestartOptions(BasedirMixin, usage.Options):
+ optParameters = [
+ ["basedir", "C", None, "which directory to restart the node in"],
+ ]
+ optFlags = [
+ ["force", "f", "if the node is not already running, start it "
+ "instead of complaining that you should have used 'start' instead "
+ "of 'restart'"],
+ ]
+
+
+def testtwistd(loc):
+ try:
+ return subprocess.call(["python", loc,], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+ except:
+ return -1
+
+twistd = None
+if not twistd:
+ for maybetwistd in which("twistd"):
+ ret = testtwistd(maybetwistd)
+ if ret == 0:
+ twistd = maybetwistd
+ break
+
+if not twistd:
+ for maybetwistd in which("twistd.py"):
+ ret = testtwistd(maybetwistd)
+ if ret == 0:
+ twistd = maybetwistd
+ break
+
+if not twistd:
+ maybetwistd = os.path.join(sys.prefix, 'Scripts', 'twistd')
+ ret = testtwistd(maybetwistd)
+ if ret == 0:
+ twistd = maybetwistd
+
+if not twistd:
+ maybetwistd = os.path.join(sys.prefix, 'Scripts', 'twistd.py')
+ ret = testtwistd(maybetwistd)
+ if ret == 0:
+ twistd = maybetwistd
+
+if not twistd:
+ print "Can't find twistd (it comes with Twisted). Aborting."
+ sys.exit(1)
+
+
+
+def do_start(basedir, config, out=sys.stdout, err=sys.stderr):
+ print >>out, "STARTING", basedir
+ if os.path.exists(os.path.join(basedir, "client.tac")):
+ tac = "client.tac"
+ type = "client"
+ elif os.path.exists(os.path.join(basedir, "introducer.tac")):
+ tac = "introducer.tac"
+ type = "introducer"
+ else:
+ print >>err, "%s does not look like a node directory" % basedir
+ if not os.path.isdir(basedir):
+ print >>err, " in fact, it doesn't look like a directory at all!"
+ sys.exit(1)
+ rc = subprocess.call(["python", twistd, "-y", tac,], cwd=basedir)
+ if rc == 0:
+ print >>out, "%s node probably started" % type
+ return 0
+ else:
+ print >>err, "%s node probably not started" % type
+ return 1
+
+def do_stop(basedir, config, out=sys.stdout, err=sys.stderr):
+ print >>out, "STOPPING", basedir
+ pidfile = os.path.join(basedir, "twistd.pid")
+ if not os.path.exists(pidfile):
+ print >>err, "%s does not look like a running node directory (no twistd.pid)" % basedir
+ return 2
+ pid = open(pidfile, "r").read()
+ pid = int(pid)
+
+ timer = 0
+ os.kill(pid, signal.SIGTERM)
+ 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 >>out, "process %d is dead" % pid
+ return
+ timer += 1
+ time.sleep(1)
+ print >>err, "never saw process go away"
+ return 1
+
+def start(config, stdout, stderr):
+ rc = 0
+ for basedir in config['basedirs']:
+ rc = do_start(basedir, config, stdout, stderr) or rc
+ return rc
+
+def stop(config, stdout, stderr):
+ rc = 0
+ for basedir in config['basedirs']:
+ rc = do_stop(basedir, config, stdout, stderr) or rc
+ return rc
+
+def restart(config, stdout, stderr):
+ rc = 0
+ for basedir in config['basedirs']:
+ rc = do_stop(basedir, config, stdout, stderr) or rc
+ if rc == 2 and config['force']:
+ print >>stderr, "ignoring couldn't-stop"
+ rc = 0
+ if rc:
+ print >>stderr, "not restarting"
+ return rc
+ for basedir in config['basedirs']:
+ rc = do_start(basedir, config, stdout, stderr) or rc
+ return rc
+
+
+subCommands = [
+ ["start", None, StartOptions, "Start a node (of any type)."],
+ ["stop", None, StopOptions, "Stop a node."],
+ ["restart", None, RestartOptions, "Restart a node."],
+]
+
+dispatch = {
+ "start": start,
+ "stop": stop,
+ "restart": restart,
+ }
projects / tahoe-lafs / tahoe-lafs.git / commitdiff