after.py 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  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. import typing
  17. from pip._vendor.tenacity import _utils
  18. if typing.TYPE_CHECKING:
  19. import logging
  20. from pip._vendor.tenacity import RetryCallState
  21. def after_nothing(retry_state: "RetryCallState") -> None:
  22. """After call strategy that does nothing."""
  23. def after_log(
  24. logger: "logging.Logger",
  25. log_level: int,
  26. sec_format: str = "%0.3f",
  27. ) -> typing.Callable[["RetryCallState"], None]:
  28. """After call strategy that logs to some logger the finished attempt."""
  29. def log_it(retry_state: "RetryCallState") -> None:
  30. logger.log(
  31. log_level,
  32. f"Finished call to '{_utils.get_callback_name(retry_state.fn)}' "
  33. f"after {sec_format % retry_state.seconds_since_start}(s), "
  34. f"this was the {_utils.to_ordinal(retry_state.attempt_number)} time calling it.",
  35. )
  36. return log_it