Source Code for Module buildbot.steps.master

 1  import os, types 
 2  from twisted.python import log, failure, runtime 
 3  from twisted.internet import reactor, defer, task 
 4  from buildbot.process.buildstep import RemoteCommand, BuildStep 
 5  from buildbot.process.buildstep import SUCCESS, FAILURE 
 6  from twisted.internet.protocol import ProcessProtocol 
 7   
 8 -class MasterShellCommand(BuildStep): 
 9      """ 
10      Run a shell command locally - on the buildmaster.  The shell command 
11      COMMAND is specified just as for a RemoteShellCommand.  Note that extra 
12      logfiles are not sopported. 
13      """ 
14      name='MasterShellCommand' 
15      description='Running' 
16      descriptionDone='Ran' 
17   
18 -    def __init__(self, command, **kwargs): 
19          BuildStep.__init__(self, **kwargs) 
20          self.addFactoryArguments(command=command) 
21          self.command=command 
22   
23 -    class LocalPP(ProcessProtocol): 
24 -        def __init__(self, step): 
25              self.step = step 
26   
27 -        def outReceived(self, data): 
28              self.step.stdio_log.addStdout(data) 
29   
30 -        def errReceived(self, data): 
31              self.step.stdio_log.addStderr(data) 
32   
33 -        def processEnded(self, status_object): 
34              self.step.stdio_log.addHeader("exit status %d\n" % status_object.value.exitCode) 
35              self.step.processEnded(status_object) 
36   
37 -    def start(self): 
38          # set up argv 
39          if type(self.command) in types.StringTypes: 
40              if runtime.platformType  == 'win32': 
41                  argv = os.environ['COMSPEC'].split() # allow %COMSPEC% to have args 
42                  if '/c' not in argv: argv += ['/c']  
43                  argv += [self.command] 
44              else: 
45                  # for posix, use /bin/sh. for other non-posix, well, doesn't 
46                  # hurt to try 
47                  argv = ['/bin/sh', '-c', self.command] 
48          else: 
49              if runtime.platformType  == 'win32': 
50                  argv = os.environ['COMSPEC'].split() # allow %COMSPEC% to have args 
51                  if '/c' not in argv: argv += ['/c']  
52                  argv += list(self.command) 
53              else: 
54                  argv = self.command 
55   
56          self.stdio_log = stdio_log = self.addLog("stdio") 
57   
58          if type(self.command) in types.StringTypes: 
59              stdio_log.addHeader(self.command.strip() + "\n\n") 
60          else: 
61              stdio_log.addHeader(" ".join(self.command) + "\n\n") 
62          stdio_log.addHeader("** RUNNING ON BUILDMASTER **\n") 
63          stdio_log.addHeader(" in dir %s\n" % os.getcwd()) 
64          stdio_log.addHeader(" argv: %s\n" % (argv,)) 
65   
66          # TODO add a timeout? 
67          proc = reactor.spawnProcess(self.LocalPP(self), argv[0], argv) 
68          # (the LocalPP object will call processEnded for us) 
69   
70 -    def processEnded(self, status_object): 
71          if status_object.value.exitCode != 0: 
72              self.step_status.setText(["failed (%d)" % status_object.value.exitCode]) 
73              self.finished(FAILURE) 
74          else: 
75              self.step_status.setText(["succeeded"]) 
76              self.finished(SUCCESS) 
77