3 from allmydata.scripts.common import BasedirMixin, BaseOptions
4 from allmydata.util.assertutil import precondition
5 from allmydata.util.encodingutil import listdir_unicode, quote_output
7 class CreateStatsGathererOptions(BasedirMixin, BaseOptions):
11 ["node-directory", "d", None, "Specify which directory the stats-gatherer should be created in. [no default]"],
14 def getSynopsis(self):
15 return "Usage: %s create-stats-gatherer [options] NODEDIR" % (self.command_name,)
18 stats_gatherer_tac = """
21 from allmydata import stats
22 from twisted.application import service
25 g = stats.StatsGathererService(verbose=verbose)
27 application = service.Application('allmydata_stats_gatherer')
28 g.setServiceParent(application)
32 def create_stats_gatherer(config, out=sys.stdout, err=sys.stderr):
33 basedir = config['basedir']
34 # This should always be called with an absolute Unicode basedir.
35 precondition(isinstance(basedir, unicode), basedir)
37 if os.path.exists(basedir):
38 if listdir_unicode(basedir):
39 print >>err, "The base directory %s is not empty." % quote_output(basedir)
40 print >>err, "To avoid clobbering anything, I am going to quit now."
41 print >>err, "Please use a different directory, or empty this one."
43 # we're willing to use an empty directory
46 f = open(os.path.join(basedir, "tahoe-stats-gatherer.tac"), "wb")
47 f.write(stats_gatherer_tac)
52 ["create-stats-gatherer", None, CreateStatsGathererOptions, "Create a stats-gatherer service."],
56 "create-stats-gatherer": create_stats_gatherer,