mirror of
https://github.com/goharbor/harbor.git
synced 2024-12-21 08:07:59 +01:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
|
import datetime
|
||
|
|
import logging
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
LOG = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
def wait_for(func, timeout, delay, *args, **kargs):
|
||
|
|
"""Decorator for waiting for until a function finished running."""
|
||
|
|
|
||
|
|
poll_timeout = timeout
|
||
|
|
poll_sleep_retry = delay
|
||
|
|
|
||
|
|
begin_poll = datetime.datetime.now()
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
return func(*args, **kargs)
|
||
|
|
break
|
||
|
|
except Exception as e:
|
||
|
|
if (datetime.datetime.now() - begin_poll).seconds > poll_timeout:
|
||
|
|
LOG.exception('Time out after %s seconds.' % poll_timeout)
|
||
|
|
raise TimeoutError('Timed out after %s seconds. Reason: '
|
||
|
|
'%s' % (poll_timeout, e))
|
||
|
|
else:
|
||
|
|
LOG.debug('Sleeping %s seconds before retrying'
|
||
|
|
'' % poll_sleep_retry)
|
||
|
|
time.sleep(poll_sleep_retry)
|
||
|
|
|
||
|
|
|
||
|
|
def safe_run(cmd, msg, sleep_time=180):
|
||
|
|
exit_code = shell.local(cmd)[0]
|
||
|
|
if exit_code:
|
||
|
|
LOG.warning('Failed to %s. Retry it after %s seconds' %
|
||
|
|
(msg, sleep_time))
|
||
|
|
time.sleep(sleep_time)
|
||
|
|
shell.local(cmd, raise_error=True)
|