Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(134)

Side by Side Diff: recipe_modules/step/api.py

Issue 2320223002: Revert of Track step nesting in StreamEngine. (Closed) Base URL: https://github.com/luci/recipes-py@emit-initial-properties
Patch Set: Created 4 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 # Copyright 2013 The LUCI Authors. All rights reserved. 1 # Copyright 2013 The LUCI Authors. All rights reserved.
2 # Use of this source code is governed under the Apache License, Version 2.0 2 # Use of this source code is governed under the Apache License, Version 2.0
3 # that can be found in the LICENSE file. 3 # that can be found in the LICENSE file.
4 4
5 import contextlib 5 import contextlib
6 6
7 from recipe_engine import recipe_api 7 from recipe_engine import recipe_api
8 8
9 9
10 # Inherit from RecipeApiPlain because the only thing which is a step is 10 # Inherit from RecipeApiPlain because the only thing which is a step is
11 # run_from_dict() 11 # run_from_dict()
12 class StepApi(recipe_api.RecipeApiPlain): 12 class StepApi(recipe_api.RecipeApiPlain):
13 def __init__(self, **kwargs): 13 def __init__(self, **kwargs):
14 super(StepApi, self).__init__(**kwargs) 14 super(StepApi, self).__init__(**kwargs)
15 self._step_names = {} 15 self._step_names = {}
16 self._seen_steps = set()
17 16
18 EXCEPTION = 'EXCEPTION' 17 EXCEPTION = 'EXCEPTION'
19 FAILURE = 'FAILURE' 18 FAILURE = 'FAILURE'
20 SUCCESS = 'SUCCESS' 19 SUCCESS = 'SUCCESS'
21 WARNING = 'WARNING' 20 WARNING = 'WARNING'
22 21
23 @property 22 @property
24 def StepFailure(self): 23 def StepFailure(self):
25 """ See recipe_api.py for docs. """ 24 """ See recipe_api.py for docs. """
26 return recipe_api.StepFailure 25 return recipe_api.StepFailure
(...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after
129 if cmd is not None: 128 if cmd is not None:
130 command = list(wrapper) 129 command = list(wrapper)
131 command += cmd 130 command += cmd
132 kwargs['cmd'] = command 131 kwargs['cmd'] = command
133 132
134 kwargs['timeout'] = timeout 133 kwargs['timeout'] = timeout
135 kwargs['ok_ret'] = ok_ret 134 kwargs['ok_ret'] = ok_ret
136 135
137 # Obtain information from composite step parent. 136 # Obtain information from composite step parent.
138 compositor = recipe_api._STEP_CONTEXT 137 compositor = recipe_api._STEP_CONTEXT
139 138 name = compositor.get_with_context('name', name)
140 # Calculate our full step name. If a step already has that name, add an
141 # index to the end of it.
142 #
143 # Note that another step could exist with that index already added to it
144 # by the user. If this happens, we'll continue appending indexes until we
145 # have a unique step name.
146 while True:
147 full_name = compositor.get_with_context('name', name)
148 if full_name not in self._seen_steps:
149 break
150
151 step_count = self._step_names.setdefault(full_name, 1) + 1
152 self._step_names[full_name] = step_count
153 name = "%s (%d)" % (name, step_count)
154 self._seen_steps.add(full_name)
155
156 if 'cwd' not in kwargs: 139 if 'cwd' not in kwargs:
157 kwargs['cwd'] = compositor.get('cwd') 140 kwargs['cwd'] = compositor.get('cwd')
158 kwargs['env'] = compositor.get_with_context('env', kwargs.get('env', {})) 141 kwargs['env'] = compositor.get_with_context('env', kwargs.get('env', {}))
159 kwargs['infra_step'] = compositor.get_with_context( 142 kwargs['infra_step'] = compositor.get_with_context(
160 'infra_step', bool(infra_step)) 143 'infra_step', bool(infra_step))
161 kwargs['step_nest_level'] = compositor.get_with_context('nest_level', 0) 144 kwargs['step_nest_level'] = compositor.get_with_context('nest_level', 0)
162 kwargs['name'] = full_name 145
146 # Disambiguate repeated names
147 step_count = self._step_names.setdefault(name, 0) + 1
148 self._step_names[name] = step_count
149 if step_count > 1:
150 name = "%s (%d)" % (name, step_count)
151 kwargs['name'] = name
163 152
164 schema = self.make_config() 153 schema = self.make_config()
165 schema.set_val(kwargs) 154 schema.set_val(kwargs)
166 return self.run_from_dict(self._engine.create_step(schema)) 155 return self.run_from_dict(self._engine.create_step(schema))
167 156
168 # TODO(martiniss) delete, and make generator_script use **kwargs on step() 157 # TODO(martiniss) delete, and make generator_script use **kwargs on step()
169 @recipe_api.composite_step 158 @recipe_api.composite_step
170 def run_from_dict(self, dct): 159 def run_from_dict(self, dct):
171 return self._engine.run_step(dct) 160 return self._engine.run_step(dct)
OLDNEW
« no previous file with comments | « recipe_modules/generator_script/example.expected/nested.json ('k') | recipe_modules/step/config.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698