Mercurial > hg > autobot
annotate autobot/template.py @ 289:754cda797873
update
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Tue, 15 Apr 2014 21:37:18 -0700 |
parents | 72645086a0b7 |
children | 45fd58949f8a |
rev | line source |
---|---|
1 | 1 #!/usr/bin/env python |
2 | |
3 """ | |
271
13107cdc1e33
move to mozinfo and start adding email
Jeff Hammel <jhammel@mozilla.com>
parents:
259
diff
changeset
|
4 makeitso templates for the A*Team's buildbot |
1 | 5 """ |
6 | |
271
13107cdc1e33
move to mozinfo and start adding email
Jeff Hammel <jhammel@mozilla.com>
parents:
259
diff
changeset
|
7 import mozinfo |
12
848569b7f91a
use subdirectories for the master and slave
Jeff Hammel <jhammel@mozilla.com>
parents:
11
diff
changeset
|
8 import os |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
9 import string |
1 | 10 import sys |
6
4ba6ba323871
stub out invocation of the template; this should eventually move to its own class, as we will want to subclass MakeItSoCLI, /me thinks
Jeff Hammel <jhammel@mozilla.com>
parents:
5
diff
changeset
|
11 from makeitso.cli import MakeItSoCLI |
23
a32a7055f3e6
combine variables for buildbot master+slave
Jeff Hammel <jhammel@mozilla.com>
parents:
22
diff
changeset
|
12 from makeitso.template import assemble |
4
eb289a46f4d3
make buildbot master template look close to what it should
Jeff Hammel <jhammel@mozilla.com>
parents:
3
diff
changeset
|
13 from makeitso.template import MakeItSoTemplate |
eb289a46f4d3
make buildbot master template look close to what it should
Jeff Hammel <jhammel@mozilla.com>
parents:
3
diff
changeset
|
14 from makeitso.template import Variable |
59
e66165f2f31b
complete automagic imports
Jeff Hammel <jhammel@mozilla.com>
parents:
46
diff
changeset
|
15 from projects import descriptions |
16
9ad3a9e220bc
get factory from command line
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
16 from projects import factories |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
17 from StringIO import StringIO |
289 | 18 from subprocess import check_call as call |
1 | 19 |
198
1a50b63c73a3
template now uses the slave os
Jeff Hammel <jhammel@mozilla.com>
parents:
190
diff
changeset
|
20 |
62
2a5fa6876e48
update template to conform to the new CLI
Jeff Hammel <jhammel@mozilla.com>
parents:
59
diff
changeset
|
21 def print_descriptions(): |
42
6981dcad3b2c
--list-factories now works
Jeff Hammel <jhammel@mozilla.com>
parents:
41
diff
changeset
|
22 """print factory descriptions""" |
6981dcad3b2c
--list-factories now works
Jeff Hammel <jhammel@mozilla.com>
parents:
41
diff
changeset
|
23 |
6981dcad3b2c
--list-factories now works
Jeff Hammel <jhammel@mozilla.com>
parents:
41
diff
changeset
|
24 buffer = StringIO() |
190
6b0587a217d3
nicer printing for --list-factories
Jeff Hammel <jhammel@mozilla.com>
parents:
155
diff
changeset
|
25 print >> buffer, 'Factories::\n' |
42
6981dcad3b2c
--list-factories now works
Jeff Hammel <jhammel@mozilla.com>
parents:
41
diff
changeset
|
26 for key in sorted(factories.keys()): |
6981dcad3b2c
--list-factories now works
Jeff Hammel <jhammel@mozilla.com>
parents:
41
diff
changeset
|
27 print >> buffer, '%s:' % key |
190
6b0587a217d3
nicer printing for --list-factories
Jeff Hammel <jhammel@mozilla.com>
parents:
155
diff
changeset
|
28 print >> buffer, descriptions[key].rstrip() |
42
6981dcad3b2c
--list-factories now works
Jeff Hammel <jhammel@mozilla.com>
parents:
41
diff
changeset
|
29 print >> buffer |
190
6b0587a217d3
nicer printing for --list-factories
Jeff Hammel <jhammel@mozilla.com>
parents:
155
diff
changeset
|
30 return buffer.getvalue().strip() |
42
6981dcad3b2c
--list-factories now works
Jeff Hammel <jhammel@mozilla.com>
parents:
41
diff
changeset
|
31 |
6981dcad3b2c
--list-factories now works
Jeff Hammel <jhammel@mozilla.com>
parents:
41
diff
changeset
|
32 |
1 | 33 class AutobotMasterTemplate(MakeItSoTemplate): |
34 name = 'autobot-master' | |
4
eb289a46f4d3
make buildbot master template look close to what it should
Jeff Hammel <jhammel@mozilla.com>
parents:
3
diff
changeset
|
35 description = 'template for the autotools buildbot master' |
12
848569b7f91a
use subdirectories for the master and slave
Jeff Hammel <jhammel@mozilla.com>
parents:
11
diff
changeset
|
36 templates = [os.path.join('template', 'master')] |
207
dc56c0de73bc
name the slave to the os by default
Jeff Hammel <jhammel@mozilla.com>
parents:
198
diff
changeset
|
37 vars = [Variable('slave', 'buildslave name', os_type or 'slave'), |
198
1a50b63c73a3
template now uses the slave os
Jeff Hammel <jhammel@mozilla.com>
parents:
190
diff
changeset
|
38 Variable('os', 'buildslave os (win, mac, linux)', os_type), |
5
b7c521f53bda
document what the passwd variable is for
Jeff Hammel <jhammel@mozilla.com>
parents:
4
diff
changeset
|
39 Variable('passwd', 'buildslave password', default='passwd'), |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
40 Variable('slaveport', 'port to talk to slaves on', default=9010, cast=int), |
249
76b4f58df3ee
* enable a public HTML waterfall; * try to add an irc bot
Jeff Hammel <jhammel@mozilla.com>
parents:
240
diff
changeset
|
41 Variable('htmlport', 'port for waterfall display', default=8010, cast=int), |
250 | 42 Variable('publichtmlport', 'port for public waterfall display', default=8011, cast=int), |
258
8ed63380052e
use test bot and dont try to connect to irc when testing
Jeff Hammel <jhammel@mozilla.com>
parents:
257
diff
changeset
|
43 Variable('botname', 'name of IRC bot', default='autobot-test'), # use autobot in production |
289 | 44 Variable('ircHost', 'IRC host'), |
271
13107cdc1e33
move to mozinfo and start adding email
Jeff Hammel <jhammel@mozilla.com>
parents:
259
diff
changeset
|
45 Variable('ircChannels', 'channels for the IRC bot ("-" for none)', default='#ateam-test'), # use #ateam in production |
13107cdc1e33
move to mozinfo and start adding email
Jeff Hammel <jhammel@mozilla.com>
parents:
259
diff
changeset
|
46 Variable('domain', 'autobot domain name', default='localhost'), |
272 | 47 Variable('recipients', 'email recipients for all projects', '') |
252 | 48 ] |
4
eb289a46f4d3
make buildbot master template look close to what it should
Jeff Hammel <jhammel@mozilla.com>
parents:
3
diff
changeset
|
49 |
73
d033843da35a
templates work again maybe
Jeff Hammel <jhammel@mozilla.com>
parents:
69
diff
changeset
|
50 def pre(self, variables, output): |
66
cbcb96142ed6
stubs towards specifying multiple factories
Jeff Hammel <jhammel@mozilla.com>
parents:
62
diff
changeset
|
51 factory = variables.get('factories') |
13 | 52 if factory: |
66
cbcb96142ed6
stubs towards specifying multiple factories
Jeff Hammel <jhammel@mozilla.com>
parents:
62
diff
changeset
|
53 assert(set(factory).issubset(factories.keys())), 'Factory must be one of: ' % ', '.join(factories.keys()) |
13 | 54 elif self.interactive: |
62
2a5fa6876e48
update template to conform to the new CLI
Jeff Hammel <jhammel@mozilla.com>
parents:
59
diff
changeset
|
55 print print_descriptions() |
16
9ad3a9e220bc
get factory from command line
Jeff Hammel <jhammel@mozilla.com>
parents:
13
diff
changeset
|
56 sys.stdout.write('Enter factory: ') |
13 | 57 factory = raw_input() |
58 assert factory in factories, 'Factory must be one of: ' % ', '.join(factories.keys()) | |
66
cbcb96142ed6
stubs towards specifying multiple factories
Jeff Hammel <jhammel@mozilla.com>
parents:
62
diff
changeset
|
59 variables['factories'] = [factory] |
13 | 60 else: |
61 raise AssertionError("No factory provided") | |
1 | 62 |
87 | 63 # stringify for .ini file |
64 variables['factories'] = ' '.join([i for i in variables['factories']]) | |
86
2616a5f49bd3
switch to .ini files for configuration
Jeff Hammel <jhammel@mozilla.com>
parents:
74
diff
changeset
|
65 |
25
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
66 def create(self, output, variables): |
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
67 command = ['buildbot', 'create-master', output] |
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
68 print ' '.join(command) |
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
69 call(command) |
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
70 |
18
016ffd7bfd99
can now actually make a (admittedly non-working) buildmaster
Jeff Hammel <jhammel@mozilla.com>
parents:
17
diff
changeset
|
71 def post(self, variables): |
17
baee795ad6df
stub out actually calling buildbot
Jeff Hammel <jhammel@mozilla.com>
parents:
16
diff
changeset
|
72 """ |
baee795ad6df
stub out actually calling buildbot
Jeff Hammel <jhammel@mozilla.com>
parents:
16
diff
changeset
|
73 called after the template is applied |
baee795ad6df
stub out actually calling buildbot
Jeff Hammel <jhammel@mozilla.com>
parents:
16
diff
changeset
|
74 """ |
26
fdddf883c19d
first successful run of create-autobot script
Jeff Hammel <jhammel@mozilla.com>
parents:
25
diff
changeset
|
75 self.create(self.output, variables) |
17
baee795ad6df
stub out actually calling buildbot
Jeff Hammel <jhammel@mozilla.com>
parents:
16
diff
changeset
|
76 |
66
cbcb96142ed6
stubs towards specifying multiple factories
Jeff Hammel <jhammel@mozilla.com>
parents:
62
diff
changeset
|
77 |
1 | 78 class AutobotSlaveTemplate(MakeItSoTemplate): |
79 name = 'autobot-slave' | |
4
eb289a46f4d3
make buildbot master template look close to what it should
Jeff Hammel <jhammel@mozilla.com>
parents:
3
diff
changeset
|
80 description = 'template for the autotools buildbot slave' |
12
848569b7f91a
use subdirectories for the master and slave
Jeff Hammel <jhammel@mozilla.com>
parents:
11
diff
changeset
|
81 templates = [os.path.join('template', 'slave')] |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
82 vars = [Variable('master', 'host of the master', default='localhost'), |
207
dc56c0de73bc
name the slave to the os by default
Jeff Hammel <jhammel@mozilla.com>
parents:
198
diff
changeset
|
83 Variable('slave', 'buildslave name', os_type or 'slave'), |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
84 Variable('passwd', 'buildslave password', default='passwd'), |
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
85 Variable('slaveport', 'port to talk to slaves on', default=9010)] |
1 | 86 |
25
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
87 def create(self, output, variables): |
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
88 command = ['buildslave', 'create-slave', output, |
132
da7af78edcb5
ensure the slaveport is an int
Jeff Hammel <jhammel@mozilla.com>
parents:
87
diff
changeset
|
89 '%s:%d' % (variables['master'], int(variables['slaveport'])), |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
90 variables['slave'], |
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
91 variables['passwd'] ] |
21 | 92 print ' '.join(command) |
93 call(command) | |
25
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
94 |
73
d033843da35a
templates work again maybe
Jeff Hammel <jhammel@mozilla.com>
parents:
69
diff
changeset
|
95 def post(self, variables, output): |
d033843da35a
templates work again maybe
Jeff Hammel <jhammel@mozilla.com>
parents:
69
diff
changeset
|
96 self.create(output, variables) |
25
785b075be979
post method should work now [untested]
Jeff Hammel <jhammel@mozilla.com>
parents:
24
diff
changeset
|
97 |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
98 |
23
a32a7055f3e6
combine variables for buildbot master+slave
Jeff Hammel <jhammel@mozilla.com>
parents:
22
diff
changeset
|
99 class AutobotTemplate(AutobotMasterTemplate, AutobotSlaveTemplate): |
22
a7e2f5b2a7e9
stub out a combined template for the master + slave
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
100 name = 'autobot' |
a7e2f5b2a7e9
stub out a combined template for the master + slave
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
101 description = 'template for the autotools buildbot master+slave' |
a7e2f5b2a7e9
stub out a combined template for the master + slave
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
102 templates = ['template'] |
23
a32a7055f3e6
combine variables for buildbot master+slave
Jeff Hammel <jhammel@mozilla.com>
parents:
22
diff
changeset
|
103 vars = assemble(AutobotMasterTemplate, AutobotSlaveTemplate) |
145 | 104 vars.append(Variable('debug', 'whether to deploy in debug mode', default=True, cast='eval')) |
22
a7e2f5b2a7e9
stub out a combined template for the master + slave
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
105 |
73
d033843da35a
templates work again maybe
Jeff Hammel <jhammel@mozilla.com>
parents:
69
diff
changeset
|
106 def post(self, variables, output): |
d033843da35a
templates work again maybe
Jeff Hammel <jhammel@mozilla.com>
parents:
69
diff
changeset
|
107 AutobotMasterTemplate.create(self, os.path.join(output, 'master'), variables) |
240
6023ae1d13e6
remove trailing whitespace
Jeff Hammel <jhammel@mozilla.com>
parents:
207
diff
changeset
|
108 AutobotSlaveTemplate.create(self, os.path.join(output, 'slave'), variables) |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
109 |
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
110 |
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
111 class ProjectTemplate(MakeItSoTemplate): |
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
112 """ |
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
113 template for creating new autobot project stubs for testing |
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
114 """ |
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
115 name = 'autobot-project' |
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
116 templates = [os.path.join('projects', 'project.template')] |
155 | 117 vars = [Variable('description', 'description of the project'), |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
118 Variable('repo', 'repository location of the project')] |
24 | 119 |
44
481f0bb18787
stub out installing the packages, for hg anyway
Jeff Hammel <jhammel@mozilla.com>
parents:
43
diff
changeset
|
120 # CLI front end class |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
121 |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
122 class AutobotMasterCLI(MakeItSoCLI): |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
123 """ |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
124 command line handler for the master |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
125 """ |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
126 def parser(self): |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
127 parser = MakeItSoCLI.parser(self) |
66
cbcb96142ed6
stubs towards specifying multiple factories
Jeff Hammel <jhammel@mozilla.com>
parents:
62
diff
changeset
|
128 parser.add_option('-f', '--factory', dest='factories', |
cbcb96142ed6
stubs towards specifying multiple factories
Jeff Hammel <jhammel@mozilla.com>
parents:
62
diff
changeset
|
129 default=[], action='append', |
74
d94fbaac7c2c
stub out being able to find scripts cross-platform in a virtualenv
Jeff Hammel <jhammel@mozilla.com>
parents:
73
diff
changeset
|
130 help="factory to use ('--factory -' for all)") |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
131 parser.add_option('--list-factories', dest='_list_factories', |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
132 default=False, action='store_true', |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
133 help="list available factories") |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
134 return parser |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
135 |
62
2a5fa6876e48
update template to conform to the new CLI
Jeff Hammel <jhammel@mozilla.com>
parents:
59
diff
changeset
|
136 def parse(self, args=None, parser=None, options=None): |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
137 |
240
6023ae1d13e6
remove trailing whitespace
Jeff Hammel <jhammel@mozilla.com>
parents:
207
diff
changeset
|
138 # parse the command line |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
139 if not parser or not options or not args: |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
140 parser = self.parser() |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
141 options, args = parser.parse_args() |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
142 |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
143 # list the factories |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
144 if options._list_factories: |
62
2a5fa6876e48
update template to conform to the new CLI
Jeff Hammel <jhammel@mozilla.com>
parents:
59
diff
changeset
|
145 print print_descriptions() |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
146 parser.exit() |
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
147 |
74
d94fbaac7c2c
stub out being able to find scripts cross-platform in a virtualenv
Jeff Hammel <jhammel@mozilla.com>
parents:
73
diff
changeset
|
148 # check for all factories |
d94fbaac7c2c
stub out being able to find scripts cross-platform in a virtualenv
Jeff Hammel <jhammel@mozilla.com>
parents:
73
diff
changeset
|
149 if options.factories == ['-']: |
d94fbaac7c2c
stub out being able to find scripts cross-platform in a virtualenv
Jeff Hammel <jhammel@mozilla.com>
parents:
73
diff
changeset
|
150 options.factories = factories.keys() |
d94fbaac7c2c
stub out being able to find scripts cross-platform in a virtualenv
Jeff Hammel <jhammel@mozilla.com>
parents:
73
diff
changeset
|
151 |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
152 # call the parent |
62
2a5fa6876e48
update template to conform to the new CLI
Jeff Hammel <jhammel@mozilla.com>
parents:
59
diff
changeset
|
153 return MakeItSoCLI.parse(self, args, parser, options) |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
154 |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
155 |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
156 class ProjectTemplateCLI(MakeItSoCLI): |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
157 """CLI front end for project template""" |
240
6023ae1d13e6
remove trailing whitespace
Jeff Hammel <jhammel@mozilla.com>
parents:
207
diff
changeset
|
158 |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
159 def __init__(self): |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
160 self.usage = '%prog [options] project <output>' |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
161 MakeItSoCLI.__init__(self, ProjectTemplate) |
240
6023ae1d13e6
remove trailing whitespace
Jeff Hammel <jhammel@mozilla.com>
parents:
207
diff
changeset
|
162 |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
163 def parse(self, args=None, parser=None, options=None): |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
164 |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
165 # parse the arguments |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
166 parser = self.parser() |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
167 options, args = parser.parse_args(args=args) |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
168 |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
169 if not(args): |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
170 parser.print_usage() |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
171 parser.exit() |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
172 |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
173 # deploy to the correct place |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
174 if len(args) not in (1,2): |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
175 parser.error("Please provide a single project") |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
176 project = args[0].lower() |
240
6023ae1d13e6
remove trailing whitespace
Jeff Hammel <jhammel@mozilla.com>
parents:
207
diff
changeset
|
177 |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
178 # ensure the argument is a python path |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
179 assert set(project).issubset(string.letters) |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
180 |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
181 # get the variables |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
182 variables = self.get_variables(options) |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
183 variables['project'] = project |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
184 |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
185 # get the output or use the location in the source |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
186 if len(args) == 2: |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
187 output = args[1] |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
188 else: |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
189 here = os.path.dirname(os.path.abspath(__file__)) |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
190 projectdir = os.path.join(here, 'projects', project) |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
191 if os.path.exists(projectdir): |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
192 assert os.path.isdir(projectdir) |
69
849bbc2e5572
first use of project template -> devicemanager
Jeff Hammel <jhammel@mozilla.com>
parents:
68
diff
changeset
|
193 else: |
849bbc2e5572
first use of project template -> devicemanager
Jeff Hammel <jhammel@mozilla.com>
parents:
68
diff
changeset
|
194 os.makedirs(projectdir) |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
195 output = os.path.join(projectdir, '__init__.py') |
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
196 assert not os.path.exists(output), "Project '%s' already exists" % project |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
197 |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
198 return variables, output |
44
481f0bb18787
stub out installing the packages, for hg anyway
Jeff Hammel <jhammel@mozilla.com>
parents:
43
diff
changeset
|
199 |
289 | 200 |
44
481f0bb18787
stub out installing the packages, for hg anyway
Jeff Hammel <jhammel@mozilla.com>
parents:
43
diff
changeset
|
201 ### console_script front-ends |
481f0bb18787
stub out installing the packages, for hg anyway
Jeff Hammel <jhammel@mozilla.com>
parents:
43
diff
changeset
|
202 |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
203 def create_master(args=sys.argv[1:]): |
41
42ef457708de
stub out adding factories to our specialty CLI class
Jeff Hammel <jhammel@mozilla.com>
parents:
37
diff
changeset
|
204 cli = AutobotMasterCLI(AutobotMasterTemplate) |
62
2a5fa6876e48
update template to conform to the new CLI
Jeff Hammel <jhammel@mozilla.com>
parents:
59
diff
changeset
|
205 cli(*args) |
1 | 206 |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
207 def create_slave(args=sys.argv[1:]): |
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
208 cli = MakeItSoCLI(AutobotSlaveTemplate) |
62
2a5fa6876e48
update template to conform to the new CLI
Jeff Hammel <jhammel@mozilla.com>
parents:
59
diff
changeset
|
209 cli(*args) |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
210 |
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
211 def create_autobot(args=sys.argv[1:]): |
43
19d555001292
use our CLI class for the create-autobot too
Jeff Hammel <jhammel@mozilla.com>
parents:
42
diff
changeset
|
212 cli = AutobotMasterCLI(AutobotTemplate) |
62
2a5fa6876e48
update template to conform to the new CLI
Jeff Hammel <jhammel@mozilla.com>
parents:
59
diff
changeset
|
213 cli(*args) |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
214 |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
215 def create_project(args=sys.argv[1:]): |
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
216 cli = ProjectTemplateCLI() |
68
fbd4a34d8e42
make project creation template work
Jeff Hammel <jhammel@mozilla.com>
parents:
67
diff
changeset
|
217 cli(*args) |
67
e8dabfd24c42
stub for a template to create new autobot factories; unfinished
Jeff Hammel <jhammel@mozilla.com>
parents:
66
diff
changeset
|
218 |
1 | 219 if __name__ == '__main__': |
289 | 220 # default: create master |
20
30ec24255ce9
stub out having a slave template too
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
221 create_master() |