def __init__(self, buildmaster_host, port, name, passwd, basedir,
keepalive, usePTY, keepaliveTimeout=None, umask=None,
maxdelay=300, numcpus=None, unicode_encoding=None,
allow_shutdown=None):
# note: keepaliveTimeout is ignored, but preserved here for
# backward-compatibility
service.MultiService.__init__(self)
BuildWorkerBase.__init__(self, name, basedir, usePTY, umask=umask, unicode_encoding=unicode_encoding)
if keepalive == 0:
keepalive = None
self.numcpus = numcpus
self.shutdown_loop = None
if allow_shutdown == 'signal':
if not hasattr(signal, 'SIGHUP'):
raise ValueError("Can't install signal handler")
elif allow_shutdown == 'file':
self.shutdown_file = os.path.join(basedir, 'shutdown.stamp')
self.shutdown_mtime = 0
self.allow_shutdown = allow_shutdown
bf = self.bf = BotFactory(buildmaster_host, port, keepalive, maxdelay)
bf.startLogin(credentials.UsernamePassword(name, passwd), client=self.bot)
self.connection = c = internet.TCPClient(buildmaster_host, port, bf)
c.setServiceParent(self)
def startService(self):
BuildWorkerBase.startService(self)
if self.allow_shutdown == 'signal':
log.msg("Setting up SIGHUP handler to initiate shutdown")
signal.signal(signal.SIGHUP, self._handleSIGHUP)
elif self.allow_shutdown == 'file':
log.msg("Watching %s's mtime to initiate shutdown" % self.shutdown_file)
if os.path.exists(self.shutdown_file):
self.shutdown_mtime = os.path.getmtime(self.shutdown_file)
self.shutdown_loop = l = task.LoopingCall(self._checkShutdownFile)
l.start(interval=10)
def startService(self):
# importing here to avoid dependency on buildbot master package
from buildbot.buildworker.protocols.null import Connection
yield BuildWorkerBase.startService(self)
self.workername = self.name
conn = Connection(self.parent, self)
# I don't have a master property, but my parent has.
master = self.parent.master
res = yield master.buildworkers.newConnection(conn, self.name)
if res:
yield self.parent.attached(conn)