blob: 89799a0d15f54adb8053a46a16a7bafc1b7c9865 [file] [log] [blame]
Kuang-che Wu6e4beca2018-06-27 17:45:02 +08001# -*- coding: utf-8 -*-
Kuang-che Wu88875db2017-07-20 10:47:53 +08002# Copyright 2017 The Chromium OS Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5"""Bisect command line interface."""
6
7from __future__ import print_function
8import argparse
9import datetime
Kuang-che Wu8b654092018-11-09 17:56:25 +080010import json
Kuang-che Wu88875db2017-07-20 10:47:53 +080011import logging
12import os
13import re
Kuang-che Wu88875db2017-07-20 10:47:53 +080014import textwrap
15import time
16
17from bisect_kit import common
Kuang-che Wu385279d2017-09-27 14:48:28 +080018from bisect_kit import configure
Kuang-che Wu88875db2017-07-20 10:47:53 +080019from bisect_kit import core
Kuang-che Wue121fae2018-11-09 16:18:39 +080020from bisect_kit import errors
Kuang-che Wu88875db2017-07-20 10:47:53 +080021from bisect_kit import strategy
22from bisect_kit import util
23
24logger = logging.getLogger(__name__)
25
Kuang-che Wu88875db2017-07-20 10:47:53 +080026DEFAULT_SESSION_NAME = 'default'
27DEFAULT_CONFIDENCE = 0.999
28
29
30class ArgTypeError(argparse.ArgumentTypeError):
31 """An error for argument validation failure.
32
33 This not only tells users the argument is wrong but also gives correct
34 example. The main purpose of this error is for argtype_multiplexer, which
35 cascades examples from multiple ArgTypeError.
36 """
37
38 def __init__(self, msg, example):
39 self.msg = msg
40 if isinstance(example, list):
41 self.example = example
42 else:
43 self.example = [example]
44 full_msg = '%s (example value: %s)' % (self.msg, ', '.join(self.example))
45 super(ArgTypeError, self).__init__(full_msg)
46
47
48def argtype_notempty(s):
49 """Validates argument is not an empty string.
50
51 Args:
52 s: string to validate.
53
54 Raises:
55 ArgTypeError if argument is empty string.
56 """
57 if not s:
58 msg = 'should not be empty'
59 raise ArgTypeError(msg, 'foo')
60 return s
61
62
63def argtype_int(s):
64 """Validate argument is a number.
65
66 Args:
67 s: string to validate.
68
69 Raises:
70 ArgTypeError if argument is not a number.
71 """
72 try:
73 return str(int(s))
74 except ValueError:
75 raise ArgTypeError('should be a number', '123')
76
77
78def argtype_multiplexer(*args):
79 r"""argtype multiplexer
80
81 This function takes a list of argtypes or regex patterns and creates a new
82 function matching them. Moreover, it gives error message with examples.
83
Kuang-che Wubaaa4532018-08-15 17:08:10 +080084 Examples:
Kuang-che Wu88875db2017-07-20 10:47:53 +080085 >>> argtype = argtype_multiplexer(argtype_int, r'^r\d+$')
86 >>> argtype('123')
87 123
88 >>> argtype('r456')
89 r456
90 >>> argtype('hello')
91 ArgTypeError: Invalid argument (example value: 123, r\d+$)
92
93 Args:
94 *args: list of argtypes or regex pattern.
95
96 Returns:
97 A new argtype function which matches *args.
98 """
99
100 def validate(s):
101 examples = []
102 for t in args:
103 if isinstance(t, str):
104 if re.match(t, s):
105 return s
106 examples.append(t)
107 continue
108 try:
109 return t(s)
110 except ArgTypeError as e:
111 examples += e.example
112
113 msg = 'Invalid argument'
114 raise ArgTypeError(msg, examples)
115
116 return validate
117
118
119def argtype_multiplier(argtype):
120 """A new argtype that supports multiplier suffix of the given argtype.
121
Kuang-che Wubaaa4532018-08-15 17:08:10 +0800122 Examples:
Kuang-che Wu88875db2017-07-20 10:47:53 +0800123 Supports the given argtype accepting "foo" as argument, this function
124 generates a new argtype function which accepts argument like "foo*3".
125
126 Returns:
127 A new argtype function which returns (arg, times) where arg is accepted
128 by input `argtype` and times is repeating count. Note that if multiplier is
129 omitted, "times" is 1.
130 """
131
132 def helper(s):
133 m = re.match(r'^(.+)\*(\d+)$', s)
134 try:
135 if m:
136 return argtype(m.group(1)), int(m.group(2))
Kuang-che Wu68db08a2018-03-30 11:50:34 +0800137 return argtype(s), 1
Kuang-che Wu88875db2017-07-20 10:47:53 +0800138 except ArgTypeError as e:
139 # It should be okay to gives multiplier example only for the first one
140 # because it is just "example", no need to enumerate all possibilities.
141 raise ArgTypeError(e.msg, e.example + [e.example[0] + '*3'])
142
143 return helper
144
145
146def argtype_dir_path(s):
147 """Validate argument is an existing directory.
148
149 Args:
150 s: string to validate.
151
152 Raises:
153 ArgTypeError if the path is not a directory.
154 """
155 if not os.path.exists(s):
156 raise ArgTypeError('should be an existing directory', '/path/to/somewhere')
157 if not os.path.isdir(s):
158 raise ArgTypeError('should be a directory', '/path/to/somewhere')
159
160 # Normalize, trim trailing path separators.
161 if len(s) > 1 and s[-1] == os.path.sep:
162 s = s[:-1]
163 return s
164
165
166def _collect_bisect_result_values(values, line):
167 """Collect bisect result values from output line.
168
169 Args:
170 values: Collected values are appending to this list.
171 line: One line of output string.
172 """
173 m = re.match(r'^BISECT_RESULT_VALUES=(.+)', line)
174 if m:
175 try:
176 values.extend(map(float, m.group(1).split()))
177 except ValueError:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800178 raise errors.InternalError(
Kuang-che Wu88875db2017-07-20 10:47:53 +0800179 'BISECT_RESULT_VALUES should be list of floats: %r' % m.group(1))
180
181
Kuang-che Wu88518882017-09-22 16:57:25 +0800182def check_executable(program):
183 """Checks whether a program is executable.
184
185 Args:
186 program: program path in question
187
188 Returns:
189 string as error message if `program` is not executable, or None otherwise.
190 It will return None if unable to determine as well.
191 """
192 returncode = util.call('which', program)
193 if returncode == 127: # No 'which' on this platform, skip the check.
194 return None
195 if returncode == 0: # is executable
196 return None
197
198 hint = ''
199 if not os.path.exists(program):
200 hint = 'Not in PATH?'
201 elif not os.path.isfile(program):
202 hint = 'Not a file'
203 elif not os.access(program, os.X_OK):
204 hint = 'Forgot to chmod +x?'
205 elif '/' not in program:
206 hint = 'Forgot to prepend "./" ?'
207 return '%r is not executable. %s' % (program, hint)
208
209
Kuang-che Wu88875db2017-07-20 10:47:53 +0800210def do_evaluate(evaluate_cmd, domain, rev):
211 """Invokes evaluator command.
212
213 The `evaluate_cmd` can get the target revision from the environment variable
214 named 'BISECT_REV'.
215
216 The result is determined according to the exit code of evaluator:
217 0: 'old'
218 1..124: 'new'
219 125: 'skip'
220 126, 127: fatal error
221 terminated by signal: fatal error
222
223 p.s. the definition of result is compatible with git-bisect(1).
224
225 It also extracts additional values from evaluate_cmd's stdout lines which
226 match the following format:
227 BISECT_RESULT_VALUES=<float>[, <float>]*
228
229 Args:
230 evaluate_cmd: evaluator command.
231 domain: a bisect_kit.core.Domain instance.
232 rev: version to evaluate.
233
234 Returns:
235 (result, values):
236 result is one of 'old', 'new', 'skip'.
237 values are additional collected values, like performance score.
238
239 Raises:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800240 errors.ExecutionFatalError if evaluator returned fatal error code.
Kuang-che Wu88875db2017-07-20 10:47:53 +0800241 """
242 env = os.environ.copy()
243 env['BISECT_REV'] = rev
244 domain.setenv(env, rev)
245
246 values = []
247 p = util.Popen(
248 evaluate_cmd,
249 env=env,
250 stdout_callback=lambda line: _collect_bisect_result_values(values, line))
251 returncode = p.wait()
252 if returncode < 0 or returncode > 125:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800253 raise errors.ExecutionFatalError('eval failed: %s' % returncode)
Kuang-che Wu88875db2017-07-20 10:47:53 +0800254
255 if returncode == 0:
256 return 'old', values
257 if returncode == 125:
258 return 'skip', values
259 return 'new', values
260
261
262def do_switch(switch_cmd, domain, rev):
263 """Invokes switcher command.
264
265 The `switch_cmd` can get the target revision from the environment variable
266 named 'BISECT_REV'.
267
268 The result is determined according to the exit code of switcher:
269 0: switch succeeded
270 1..125: 'skip'
271 126, 127: fatal error
272 terminated by signal: fatal error
273
274 In other words, any non-fatal errors are considered as 'skip'.
275
276 Args:
277 switch_cmd: switcher command.
278 domain: a bisect_kit.core.Domain instance.
279 rev: version to switch.
280
281 Returns:
282 None if switch successfully, 'skip' otherwise.
283
284 Raises:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800285 errors.ExecutionFatalError if switcher returned fatal error code.
Kuang-che Wu88875db2017-07-20 10:47:53 +0800286 """
287 env = os.environ.copy()
288 env['BISECT_REV'] = rev
289 domain.setenv(env, rev)
290
291 returncode = util.call(*switch_cmd, env=env)
292 if returncode < 0 or returncode > 125:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800293 raise errors.ExecutionFatalError('switch failed: %s' % returncode)
Kuang-che Wu88875db2017-07-20 10:47:53 +0800294
295 if returncode != 0:
296 return 'skip'
297 return None
298
299
Kuang-che Wu68db08a2018-03-30 11:50:34 +0800300class BisectorCommandLine(object):
Kuang-che Wu88875db2017-07-20 10:47:53 +0800301 """Bisector command line interface.
302
303 The typical usage pattern:
304
305 if __name__ == '__main__':
Kuang-che Wu68db08a2018-03-30 11:50:34 +0800306 BisectorCommandLine(CustomDomain).main()
Kuang-che Wu88875db2017-07-20 10:47:53 +0800307
308 where CustomDomain is a derived class of core.BisectDomain. See
Kuang-che Wu02170592018-07-09 21:42:44 +0800309 bisect_list.py as example.
Kuang-che Wu88875db2017-07-20 10:47:53 +0800310
311 If you need to control the bisector using python code, the easier way is
312 passing command line arguments to main() function. For example,
313 bisector = Bisector(CustomDomain)
314 bisector.main('init', '--old', '123', '--new', '456')
315 bisector.main('config', 'switch', 'true')
316 bisector.main('config', 'eval', 'true')
317 bisector.main('run')
318 """
319
320 def __init__(self, domain_cls):
321 self.domain_cls = domain_cls
322 self.domain = None
323 self.states = None
324 self.strategy = None
325
326 @property
327 def config(self):
328 return self.states.config
329
Kuang-che Wu8b654092018-11-09 17:56:25 +0800330 def _add_sample(self, rev, status, **kwargs):
Kuang-che Wu88875db2017-07-20 10:47:53 +0800331 idx = self.states.rev2idx(rev)
Kuang-che Wu8b654092018-11-09 17:56:25 +0800332 self.states.add_sample(idx, status, **kwargs)
Kuang-che Wu88875db2017-07-20 10:47:53 +0800333 self.strategy.update(idx, status)
334
335 def cmd_reset(self, _opts):
336 """Resets bisect session and clean up saved result."""
337 self.states.reset()
338
339 def cmd_init(self, opts):
340 """Initializes bisect session.
341
342 See init command's help message for more detail.
343 """
344 config, revlist = self.domain_cls.init(opts)
Kuang-che Wu42551dd2018-01-16 17:27:20 +0800345 logger.info('found %d revs to bisect', len(revlist))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800346 logger.debug('revlist %r', revlist)
347 if 'new' not in config:
348 config['new'] = opts.new
349 if 'old' not in config:
350 config['old'] = opts.old
351 assert len(revlist) >= 2
352 assert config['new'] in revlist
353 assert config['old'] in revlist
354 old_idx = revlist.index(config['old'])
355 new_idx = revlist.index(config['new'])
356 assert old_idx < new_idx
357
358 config.update(confidence=opts.confidence, noisy=opts.noisy)
359
360 self.states.init(config, revlist)
361 self.states.save()
362
363 def _switch_and_eval(self, rev, prev_rev=None):
364 """Switches and evaluates given version.
365
366 If current version equals to target, switch step will be skip.
367
368 Args:
369 rev: Target version.
370 prev_rev: Previous version.
371
372 Returns:
373 (step, status, values):
374 step: Last step executed ('switch' or 'eval').
375 status: Execution result ('old', 'new', or 'skip').
376 values: Collected values from eval step. None if last step is 'switch'.
377 """
Kuang-che Wu88875db2017-07-20 10:47:53 +0800378 if prev_rev != rev:
379 logger.debug('switch to rev=%s', rev)
380 t0 = time.time()
381 status = do_switch(self.config['switch'], self.domain, rev)
382 t1 = time.time()
383 if status == 'skip':
384 logger.debug('switch failed => skip')
385 return 'switch', status, None
386 self.states.data['stats']['switch_count'] += 1
387 self.states.data['stats']['switch_time'] += t1 - t0
388
389 logger.debug('eval rev=%s', rev)
390 t0 = time.time()
391 status, values = do_evaluate(self.config['eval'], self.domain, rev)
392 t1 = time.time()
393 if status == 'skip':
394 return 'eval', status, values
395 self.states.data['stats']['eval_count'] += 1
396 self.states.data['stats']['eval_time'] += t1 - t0
397
398 return 'eval', status, values
399
Kuang-che Wu889f68e2018-10-29 14:12:13 +0800400 def _next_idx_iter(self, opts, force):
Kuang-che Wu88875db2017-07-20 10:47:53 +0800401 if opts.revs:
402 for rev in opts.revs:
403 idx = self.states.rev2idx(rev)
404 logger.info('try idx=%d rev=%s (command line specified)', idx, rev)
405 yield idx, rev
406 if opts.once:
407 break
408 else:
Kuang-che Wu889f68e2018-10-29 14:12:13 +0800409 while force or not self.strategy.is_done():
Kuang-che Wu88875db2017-07-20 10:47:53 +0800410 idx = self.strategy.next_idx()
411 rev = self.states.idx2rev(idx)
412 logger.info('try idx=%d rev=%s', idx, rev)
413 yield idx, rev
Kuang-che Wu889f68e2018-10-29 14:12:13 +0800414 force = False
Kuang-che Wu88875db2017-07-20 10:47:53 +0800415 if opts.once:
416 break
417
418 def cmd_run(self, opts):
419 """Performs bisection.
420
421 See run command's help message for more detail.
422
423 Raises:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800424 errors.VerificationFailed: The bisection range is verified false. We
Kuang-che Wu88875db2017-07-20 10:47:53 +0800425 expect 'old' at the first rev and 'new' at last rev.
Kuang-che Wue121fae2018-11-09 16:18:39 +0800426 errors.UnableToProceed: Too many errors to narrow down further the
427 bisection range.
Kuang-che Wu88875db2017-07-20 10:47:53 +0800428 """
Kuang-che Wu8b654092018-11-09 17:56:25 +0800429 # Set dummy values in case exception raised before loop.
430 idx, rev = -1, None
431 try:
432 assert self.config.get('switch')
433 assert self.config.get('eval')
434 self.strategy.rebuild()
Kuang-che Wu88875db2017-07-20 10:47:53 +0800435
Kuang-che Wu8b654092018-11-09 17:56:25 +0800436 prev_rev = None
437 force = opts.force
438 for idx, rev in self._next_idx_iter(opts, force):
439 if not force:
440 # Bail out if bisection range is unlikely true in order to prevent
441 # wasting time. This is necessary because some configurations (say,
442 # confidence) may be changed before cmd_run() and thus the bisection
443 # range becomes not acceptable.
444 self.strategy.check_verification_range()
Kuang-che Wu88875db2017-07-20 10:47:53 +0800445
Kuang-che Wu8b654092018-11-09 17:56:25 +0800446 step, status, values = self._switch_and_eval(rev, prev_rev=prev_rev)
447 logger.info('rev=%s => status %s', rev, status)
448 force = False
449
450 self._add_sample(rev, status, values=values)
451 self.states.save()
452
453 # Bail out if bisection range is unlikely true.
Kuang-che Wu889f68e2018-10-29 14:12:13 +0800454 self.strategy.check_verification_range()
Kuang-che Wu88875db2017-07-20 10:47:53 +0800455
Kuang-che Wu8b654092018-11-09 17:56:25 +0800456 if status == 'skip':
457 current_state = self.states.get(idx)
458 if current_state['skip'] > (
459 current_state['old'] + current_state['new'] + 1) * 5:
460 message = 'too much "skip" for rev=%r' % rev
461 raise errors.UnableToProceed(message)
Kuang-che Wu88875db2017-07-20 10:47:53 +0800462
Kuang-che Wu8b654092018-11-09 17:56:25 +0800463 self.strategy.show_summary()
464
465 if step == 'switch' and status == 'skip':
466 # Previous switch failed and thus the current version is unknown. Set
467 # it None, so next switch operation won't be bypassed (due to
468 # optimization).
469 prev_rev = None
470 else:
471 prev_rev = rev
472
473 logger.info('done')
474 old_idx, new_idx = self.strategy.get_range()
475 self.states.add_history('done')
Kuang-che Wu889f68e2018-10-29 14:12:13 +0800476 self.states.save()
Kuang-che Wu8b654092018-11-09 17:56:25 +0800477 except Exception as e:
478 exception_name = e.__class__.__name__
479 self.states.add_history(
480 'failed', text='%s: %s' % (exception_name, e), index=idx, rev=rev)
481 self.states.save()
482 raise
483 finally:
484 if rev:
485 # progress so far
486 old_idx, new_idx = self.strategy.get_range()
487 self.states.add_history(
488 'range',
489 old=self.states.idx2rev(old_idx),
490 new=self.states.idx2rev(new_idx))
491 self.states.save()
Kuang-che Wu88875db2017-07-20 10:47:53 +0800492
493 def cmd_view(self, opts):
494 """Shows current progress and candidates."""
495 self.strategy.rebuild()
496 # Rebuild twice in order to re-estimate noise.
497 self.strategy.rebuild()
498 self.strategy.show_summary(more=opts.more)
499 left, right = self.strategy.get_range()
Kuang-che Wu81aecc02018-10-31 19:37:32 +0800500 self.domain.view(self.states.data['revlist'], self.states.idx2rev(left),
501 self.states.idx2rev(right))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800502
503 def current_status(self, session=None, session_base=None):
504 """Gets current bisect status.
505
506 Returns:
507 A dict describing current status. It contains following items:
508 inited: True iff the session file is initialized (init command has been
509 invoked). If not, below items are omitted.
510 old: Start of current estimated range.
511 new: End of current estimated range.
Kuang-che Wu8b654092018-11-09 17:56:25 +0800512 verified: The bisect range is already verified.
Kuang-che Wu88875db2017-07-20 10:47:53 +0800513 estimated_noise: New estimated noise.
514 done: True if bisection is done, otherwise False.
515 """
516 self._create_states(session=session, session_base=session_base)
517 if self.states.load():
518 self.strategy = strategy.NoisyBinarySearch(
519 self.states.rev_info,
520 self.states.rev2idx(self.config['old']),
521 self.states.rev2idx(self.config['new']),
522 confidence=self.config['confidence'],
523 observation=self.config['noisy'])
524 self.strategy.rebuild()
525 left, right = self.strategy.get_range()
526 estimated_noise = self.strategy.get_noise_observation()
527
528 result = dict(
529 inited=True,
530 old=self.states.idx2rev(left),
531 new=self.states.idx2rev(right),
Kuang-che Wu8b654092018-11-09 17:56:25 +0800532 verified=self.strategy.is_range_verified(),
Kuang-che Wudd7f6f02018-06-28 18:19:30 +0800533 estimated_noise=estimated_noise,
Kuang-che Wu88875db2017-07-20 10:47:53 +0800534 done=self.strategy.is_done())
535 else:
536 result = dict(inited=False)
537 return result
538
Kuang-che Wu8b654092018-11-09 17:56:25 +0800539 def cmd_log(self, opts):
Kuang-che Wu88875db2017-07-20 10:47:53 +0800540 """Prints what has been done so far."""
Kuang-che Wu8b654092018-11-09 17:56:25 +0800541 history = []
Kuang-che Wu88875db2017-07-20 10:47:53 +0800542 for entry in self.states.data['history']:
Kuang-che Wu8b654092018-11-09 17:56:25 +0800543 if opts.before and entry['timestamp'] >= opts.before:
544 continue
545 if opts.after and entry['timestamp'] <= opts.after:
546 continue
547 history.append(entry)
548
549 if opts.json:
550 print(json.dumps(history, indent=2))
551 return
552
553 for entry in history:
554 entry_time = datetime.datetime.fromtimestamp(int(entry['timestamp']))
555 if entry.get('event', 'sample') == 'sample':
556 print('{datetime} {rev} {status} {values} {comment}'.format(
557 datetime=entry_time,
558 rev=entry['rev'],
559 status=entry['status'] + ('*%d' % entry['times']
560 if entry.get('times', 1) > 1 else ''),
561 values=entry.get('values', ''),
562 comment=entry.get('comment', '')))
563 else:
564 print('%s %r' % (entry_time, entry))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800565
566 def cmd_next(self, _opts):
567 """Prints next suggested rev to bisect."""
568 self.strategy.rebuild()
569 if self.strategy.is_done():
570 print('done')
571 return
572
573 idx = self.strategy.next_idx()
574 rev = self.states.idx2rev(idx)
575 print(rev)
576
577 def cmd_switch(self, opts):
578 """Switches to given rev without eval."""
579 assert self.config.get('switch')
580
581 self.strategy.rebuild()
582
583 if opts.rev == 'next':
584 idx = self.strategy.next_idx()
585 rev = self.states.idx2rev(idx)
586 else:
Kuang-che Wu752228c2018-09-05 13:54:22 +0800587 rev = self.domain_cls.intra_revtype(opts.rev)
Kuang-che Wu88875db2017-07-20 10:47:53 +0800588 assert rev
589
590 logger.info('switch to %s', rev)
591 status = do_switch(self.config['switch'], self.domain, rev)
592 if status:
593 print('switch failed')
594
595 def _add_revs_status_helper(self, revs, status):
596 self.strategy.rebuild()
597 for rev, times in revs:
Kuang-che Wu8b654092018-11-09 17:56:25 +0800598 self._add_sample(rev, status, times=times, comment='manual')
Kuang-che Wu88875db2017-07-20 10:47:53 +0800599 self.states.save()
600
601 def cmd_new(self, opts):
602 """Tells bisect engine the said revs have "new" behavior."""
603 logger.info('set [%s] as new', opts.revs)
604 self._add_revs_status_helper(opts.revs, 'new')
605
606 def cmd_old(self, opts):
607 """Tells bisect engine the said revs have "old" behavior."""
608 logger.info('set [%s] as old', opts.revs)
609 self._add_revs_status_helper(opts.revs, 'old')
610
611 def cmd_skip(self, opts):
612 """Tells bisect engine the said revs have "skip" behavior."""
613 logger.info('set [%s] as skip', opts.revs)
614 self._add_revs_status_helper(opts.revs, 'skip')
615
616 def _create_states(self, session=None, session_base=None):
617 if not session:
618 session = DEFAULT_SESSION_NAME
619 if not session_base:
Kuang-che Wu41e8b592018-09-25 17:01:30 +0800620 session_base = configure.get('SESSION_BASE', common.DEFAULT_SESSION_BASE)
Kuang-che Wu88875db2017-07-20 10:47:53 +0800621
622 session_file = os.path.join(session_base, session, self.domain_cls.__name__)
623
624 if self.states:
625 assert self.states.session_file == session_file
626 else:
Kuang-che Wuc5781932018-10-05 00:30:19 +0800627 self.states = core.BisectStates(session_file)
Kuang-che Wu88875db2017-07-20 10:47:53 +0800628
629 def cmd_config(self, opts):
630 """Configures additional setting.
631
632 See config command's help message for more detail.
633 """
634 self.states.load()
635 self.domain = self.domain_cls(self.states.config)
636 if not opts.value:
637 print(self.states.config[opts.key])
638 return
639
640 if opts.key in ['switch', 'eval']:
Kuang-che Wu88518882017-09-22 16:57:25 +0800641 result = check_executable(opts.value[0])
642 if result:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800643 raise errors.ArgumentError('%s command' % opts.key, result)
Kuang-che Wu88518882017-09-22 16:57:25 +0800644
Kuang-che Wu88875db2017-07-20 10:47:53 +0800645 self.states.config[opts.key] = opts.value
646
647 elif opts.key == 'confidence':
648 if len(opts.value) != 1:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800649 raise errors.ArgumentError(
650 'confidence value',
651 'expected 1 value, %d values given' % len(opts.value))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800652 try:
653 self.states.config[opts.key] = float(opts.value[0])
654 except ValueError:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800655 raise errors.ArgumentError('confidence value',
656 'invalid float value: %r' % opts.value[0])
Kuang-che Wu88875db2017-07-20 10:47:53 +0800657
658 elif opts.key == 'noisy':
659 if len(opts.value) != 1:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800660 raise errors.ArgumentError(
661 'noisy value',
662 'expected 1 value, %d values given' % len(opts.value))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800663 self.states.config[opts.key] = opts.value[0]
664
665 else:
Kuang-che Wue121fae2018-11-09 16:18:39 +0800666 # unreachable
667 assert 0
Kuang-che Wu88875db2017-07-20 10:47:53 +0800668
669 self.states.save()
670
671 def create_argument_parser(self, prog):
Kuang-che Wub2376262017-11-20 18:05:24 +0800672 if self.domain_cls.help:
673 description = self.domain_cls.help
674 else:
675 description = 'Bisector for %s' % self.domain_cls.__name__
676 description += textwrap.dedent('''
677 When running switcher and evaluator, it will set BISECT_REV environment
678 variable, indicates current rev to switch/evaluate.
679 ''')
680
Kuang-che Wu88875db2017-07-20 10:47:53 +0800681 parser = argparse.ArgumentParser(
682 prog=prog,
683 formatter_class=argparse.RawDescriptionHelpFormatter,
Kuang-che Wub2376262017-11-20 18:05:24 +0800684 description=description)
Kuang-che Wu385279d2017-09-27 14:48:28 +0800685 common.add_common_arguments(parser)
686 parser.add_argument(
Kuang-che Wu88875db2017-07-20 10:47:53 +0800687 '--session_base',
Kuang-che Wu41e8b592018-09-25 17:01:30 +0800688 default=configure.get('SESSION_BASE', common.DEFAULT_SESSION_BASE),
Kuang-che Wu88875db2017-07-20 10:47:53 +0800689 help='Directory to store sessions (default: %(default)r)')
690 parser.add_argument(
691 '--session',
692 default=DEFAULT_SESSION_NAME,
693 help='Session name (default: %(default)r)')
694 subparsers = parser.add_subparsers(
695 dest='command', title='commands', metavar='<command>')
696
697 parser_reset = subparsers.add_parser(
698 'reset', help='Reset bisect session and clean up saved result')
699 parser_reset.set_defaults(func=self.cmd_reset)
700
701 parser_init = subparsers.add_parser(
702 'init',
703 help='Initializes bisect session',
704 formatter_class=argparse.RawDescriptionHelpFormatter,
705 description=textwrap.dedent('''
706 Besides arguments for 'init' command, you also need to set 'switch'
707 and 'eval' command line via 'config' command.
708 $ bisector config switch <switch command and arguments>
709 $ bisector config eval <eval command and arguments>
710
711 The value of --noisy and --confidence could be changed by 'config'
712 command after 'init' as well.
713 '''))
714 parser_init.add_argument(
715 '--old',
716 required=True,
717 type=self.domain_cls.revtype,
718 help='Start of bisect range, which has old behavior')
719 parser_init.add_argument(
720 '--new',
721 required=True,
722 type=self.domain_cls.revtype,
723 help='End of bisect range, which has new behavior')
724 parser_init.add_argument(
725 '--noisy',
726 help='Enable noisy binary search and specify prior result. '
727 'For example, "old=1/10,new=2/3" means old fail rate is 1/10 '
728 'and new fail rate increased to 2/3. '
729 'Skip if not flaky, say, "new=2/3" means old is always good.')
730 parser_init.add_argument(
731 '--confidence',
732 type=float,
733 default=DEFAULT_CONFIDENCE,
734 help='Confidence level (default: %(default)r)')
735 parser_init.set_defaults(func=self.cmd_init)
736 self.domain_cls.add_init_arguments(parser_init)
737
738 parser_config = subparsers.add_parser(
739 'config', help='Configures additional setting')
740 parser_config.add_argument(
741 'key',
742 choices=['switch', 'eval', 'confidence', 'noisy'],
743 metavar='key',
744 help='What config to change. choices=[%(choices)s]')
745 parser_config.add_argument(
746 'value', nargs=argparse.REMAINDER, help='New value')
747 parser_config.set_defaults(func=self.cmd_config)
748
749 parser_run = subparsers.add_parser(
750 'run',
751 help='Performs bisection',
752 formatter_class=argparse.RawDescriptionHelpFormatter,
753 description=textwrap.dedent('''
754 This command does switch and eval to determine candidates having old or
755 new behavior.
756
757 By default, it attempts to try versions in binary search manner until
758 found the first version having new behavior.
759
760 If version numbers are specified on command line, it just tries those
761 versions and record the result.
762
763 Example:
764 Bisect automatically.
765 $ %(prog)s
766
767 Switch and run version "2.13" and "2.14" and then stop.
768 $ %(prog)s 2.13 2.14
769 '''))
770 parser_run.add_argument(
771 '-1', '--once', action='store_true', help='Only run one step')
772 parser_run.add_argument(
Kuang-che Wu889f68e2018-10-29 14:12:13 +0800773 '--force',
774 action='store_true',
775 help="Run at least once even it's already done")
776 parser_run.add_argument(
Kuang-che Wu88875db2017-07-20 10:47:53 +0800777 'revs',
778 nargs='*',
Kuang-che Wu752228c2018-09-05 13:54:22 +0800779 type=self.domain_cls.intra_revtype,
Kuang-che Wu88875db2017-07-20 10:47:53 +0800780 help='revs to switch+eval; '
781 'default is calculating automatically and run until done')
782 parser_run.set_defaults(func=self.cmd_run)
783
784 parser_switch = subparsers.add_parser(
785 'switch', help='Switch to given rev without eval')
786 parser_switch.add_argument(
Kuang-che Wu752228c2018-09-05 13:54:22 +0800787 'rev', type=argtype_multiplexer(self.domain_cls.intra_revtype, 'next'))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800788 parser_switch.set_defaults(func=self.cmd_switch)
789
790 parser_old = subparsers.add_parser(
791 'old', help='Tells bisect engine the said revs have "old" behavior')
792 parser_old.add_argument(
Kuang-che Wu752228c2018-09-05 13:54:22 +0800793 'revs',
794 nargs='+',
795 type=argtype_multiplier(self.domain_cls.intra_revtype))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800796 parser_old.set_defaults(func=self.cmd_old)
797
798 parser_new = subparsers.add_parser(
799 'new', help='Tells bisect engine the said revs have "new" behavior')
800 parser_new.add_argument(
Kuang-che Wu752228c2018-09-05 13:54:22 +0800801 'revs',
802 nargs='+',
803 type=argtype_multiplier(self.domain_cls.intra_revtype))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800804 parser_new.set_defaults(func=self.cmd_new)
805
806 parser_skip = subparsers.add_parser(
807 'skip', help='Tells bisect engine the said revs have "skip" behavior')
808 parser_skip.add_argument(
Kuang-che Wu752228c2018-09-05 13:54:22 +0800809 'revs',
810 nargs='+',
811 type=argtype_multiplier(self.domain_cls.intra_revtype))
Kuang-che Wu88875db2017-07-20 10:47:53 +0800812 parser_skip.set_defaults(func=self.cmd_skip)
813
814 parser_view = subparsers.add_parser(
815 'view', help='Shows current progress and candidates')
816 parser_view.add_argument('--more', action='store_true')
817 parser_view.set_defaults(func=self.cmd_view)
818
819 parser_log = subparsers.add_parser(
820 'log', help='Prints what has been done so far')
Kuang-che Wu8b654092018-11-09 17:56:25 +0800821 parser_log.add_argument('--before', type=float)
822 parser_log.add_argument('--after', type=float)
823 parser_log.add_argument(
824 '--json', action='store_true', help='Machine readable output')
Kuang-che Wu88875db2017-07-20 10:47:53 +0800825 parser_log.set_defaults(func=self.cmd_log)
826
827 parser_next = subparsers.add_parser(
828 'next', help='Prints next suggested rev to bisect')
829 parser_next.set_defaults(func=self.cmd_next)
830
831 return parser
832
833 def main(self, *args, **kwargs):
834 """Command line main function.
835
836 Args:
837 *args: Command line arguments.
838 **kwargs: additional non command line arguments passed by script code.
839 {
840 'prog': Program name; optional.
841 }
842 """
Kuang-che Wu385279d2017-09-27 14:48:28 +0800843 common.init()
Kuang-che Wu88875db2017-07-20 10:47:53 +0800844 parser = self.create_argument_parser(kwargs.get('prog'))
845 opts = parser.parse_args(args or None)
846 common.config_logging(opts)
847
848 self._create_states(session=opts.session, session_base=opts.session_base)
849 if opts.command not in ('init', 'reset', 'config'):
850 self.states.load()
851 self.domain = self.domain_cls(self.states.config)
852 self.strategy = strategy.NoisyBinarySearch(
853 self.states.rev_info,
854 self.states.rev2idx(self.config['old']),
855 self.states.rev2idx(self.config['new']),
856 confidence=self.config['confidence'],
857 observation=self.config['noisy'])
858
859 return opts.func(opts)