1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#!/usr/bin/env python -u
"""action_config_script.py
Demonstrate actions and config.
"""
import os
import sys
import time
sys.path.insert(1, os.path.dirname(sys.path[0]))
from mozharness.base.script import BaseScript
# ActionsConfigExample {{{1
class ActionsConfigExample(BaseScript):
config_options = [[
['--beverage', ],
{"action": "store",
"dest": "beverage",
"type": "string",
"help": "Specify your beverage of choice",
}
], [
['--ship-style', ],
{"action": "store",
"dest": "ship_style",
"type": "choice",
"choices": ["1", "2", "3"],
"help": "Specify the type of ship",
}
], [
['--long-sleep-time', ],
{"action": "store",
"dest": "long_sleep_time",
"type": "int",
"help": "Specify how long to sleep",
}
]]
def __init__(self, require_config_file=False):
super(ActionsConfigExample, self).__init__(
config_options=self.config_options,
all_actions=[
'clobber',
'nap',
'ship-it',
],
default_actions=[
'clobber',
'nap',
'ship-it',
],
require_config_file=require_config_file,
config={
'beverage': "kool-aid",
'long_sleep_time': 3600,
'ship_style': "1",
}
)
def _sleep(self, sleep_length, interval=5):
self.info("Sleeping %d seconds..." % sleep_length)
counter = 0
while counter + interval <= sleep_length:
sys.stdout.write(".")
try:
time.sleep(interval)
except:
print
self.error("Impatient, are we?")
sys.exit(1)
counter += interval
print()
self.info("Ok, done.")
def _ship1(self):
self.info("""
_~
_~ )_)_~
)_))_))_)
_!__!__!_
\______t/
~~~~~~~~~~~~~
""")
def _ship2(self):
self.info("""
_4 _4
_)_))_)
_)_)_)_)
_)_))_))_)_
\_=__=__=_/
~~~~~~~~~~~~~
""")
def _ship3(self):
self.info("""
,;;:;,
;;;;;
,:;;:; ,'=.
;:;:;' .=" ,'_\\
':;:;,/ ,__:=@
';;:; =./)_
`"=\\_ )_"`
``'"`
""")
def nap(self):
for var_name in self.config.keys():
if var_name.startswith("random_config_key"):
self.info("This is going to be %s!" % self.config[var_name])
sleep_time = self.config['long_sleep_time']
if sleep_time > 60:
self.info("Ok, grab a %s. This is going to take a while." % self.config['beverage'])
else:
self.info("This will be quick, but grab a %s anyway." % self.config['beverage'])
self._sleep(self.config['long_sleep_time'])
def ship_it(self):
name = "_ship%s" % self.config['ship_style']
if hasattr(self, name):
getattr(self, name)()
# __main__ {{{1
if __name__ == '__main__':
actions_config_example = ActionsConfigExample()
actions_config_example.run_and_exit()
|
