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

Side by Side Diff: build/android/pylib/utils/timeout_retry.py

Issue 132463007: Enable presubmit pylint in build/android. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: rebase vs tot and only disabling F0401 in specific spots Created 6 years, 10 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 Chromium Authors. All rights reserved. 1 # Copyright 2013 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be 2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file. 3 # found in the LICENSE file.
4 4
5 """A utility to run functions with timeouts and retries.""" 5 """A utility to run functions with timeouts and retries."""
6 # pylint: disable=W0702
6 7
7 import functools
8 import threading 8 import threading
9 9
10 import reraiser_thread 10 from pylib.utils import reraiser_thread
11 import watchdog_timer 11 from pylib.utils import watchdog_timer
12 12
13 13
14 def Run(func, timeout, retries, args=[], kwargs={}): 14 def Run(func, timeout, retries, args=None, kwargs=None):
15 """Runs the passed function in a separate thread with timeouts and retries. 15 """Runs the passed function in a separate thread with timeouts and retries.
16 16
17 Args: 17 Args:
18 func: the function to be wrapped. 18 func: the function to be wrapped.
19 timeout: the timeout in seconds for each try. 19 timeout: the timeout in seconds for each try.
20 retries: the number of retries. 20 retries: the number of retries.
21 args: list of positional args to pass to |func|. 21 args: list of positional args to pass to |func|.
22 kwargs: dictionary of keyword args to pass to |func|. 22 kwargs: dictionary of keyword args to pass to |func|.
23 23
24 Returns: 24 Returns:
25 The return value of func(*args, **kwargs). 25 The return value of func(*args, **kwargs).
26 """ 26 """
27 if not args:
28 args = []
29 if not kwargs:
30 kwargs = {}
31
27 # The return value uses a list because Python variables are references, not 32 # The return value uses a list because Python variables are references, not
28 # values. Closures make a copy of the reference, so updating the closure's 33 # values. Closures make a copy of the reference, so updating the closure's
29 # reference wouldn't update where the original reference pointed. 34 # reference wouldn't update where the original reference pointed.
30 ret = [None] 35 ret = [None]
31 def RunOnTimeoutThread(): 36 def RunOnTimeoutThread():
32 ret[0] = func(*args, **kwargs) 37 ret[0] = func(*args, **kwargs)
33 38
34 while True: 39 while True:
35 try: 40 try:
36 name = 'TimeoutThread-for-%s' % threading.current_thread().name 41 name = 'TimeoutThread-for-%s' % threading.current_thread().name
37 thread_group = reraiser_thread.ReraiserThreadGroup( 42 thread_group = reraiser_thread.ReraiserThreadGroup(
38 [reraiser_thread.ReraiserThread(RunOnTimeoutThread, name=name)]) 43 [reraiser_thread.ReraiserThread(RunOnTimeoutThread, name=name)])
39 thread_group.StartAll() 44 thread_group.StartAll()
40 thread_group.JoinAll(watchdog_timer.WatchdogTimer(timeout)) 45 thread_group.JoinAll(watchdog_timer.WatchdogTimer(timeout))
41 return ret[0] 46 return ret[0]
42 except: 47 except:
43 if retries <= 0: 48 if retries <= 0:
44 raise 49 raise
45 retries -= 1 50 retries -= 1
OLDNEW
« no previous file with comments | « build/android/pylib/utils/test_environment.py ('k') | build/android/pylib/utils/timeout_retry_unittest.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698