before_sleep.py 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. # Copyright 2016 Julien Danjou
  2. # Copyright 2016 Joshua Harlow
  3. # Copyright 2013-2014 Ray Holder
  4. #
  5. # Licensed under the Apache License, Version 2.0 (the "License");
  6. # you may not use this file except in compliance with the License.
  7. # You may obtain a copy of the License at
  8. #
  9. # http://www.apache.org/licenses/LICENSE-2.0
  10. #
  11. # Unless required by applicable law or agreed to in writing, software
  12. # distributed under the License is distributed on an "AS IS" BASIS,
  13. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. # See the License for the specific language governing permissions and
  15. # limitations under the License.
  16. from tenacity import _utils
  17. def before_sleep_nothing(retry_state):
  18. """Before call strategy that does nothing."""
  19. def before_sleep_log(logger, log_level):
  20. """Before call strategy that logs to some logger the attempt."""
  21. def log_it(retry_state):
  22. if retry_state.outcome.failed:
  23. verb, value = 'raised', retry_state.outcome.exception()
  24. else:
  25. verb, value = 'returned', retry_state.outcome.result()
  26. logger.log(log_level,
  27. "Retrying %s in %s seconds as it %s %s.",
  28. _utils.get_callback_name(retry_state.fn),
  29. getattr(retry_state.next_action, 'sleep'),
  30. verb, value)
  31. return log_it