Python Decorator – Как использовать возвращаемые значения функции в функции декоратора ⇐ Python

[Anonymous]

Я не знаю, как это решить. Я тестирую отправку запросов GET с помощью модуля запросов Python. Идея состоит в том, что если я получу ответ.status_code больше 500, мне придется повторить запрос.
На данный момент я реализовал решение повторить запрос независимо от кода состояния, и все работает отлично. Однако я изо всех сил пытаюсь передать status_code исходного соединения в функции main() в функцию декоратора retry_requests.
У меня есть следующий рабочий код ниже.

Код: Выделить всё

import requests, time

def retry_requests(func, retries=3):
def retry_wrapper(*args, **kwargs):
attempts = 0
while attempts < retries:
try:
return func(*args, **kwargs)
except requests.exceptions.RequestException as e:
print(f"Error as {e}.")
attempts +=1
time.sleep(3)
return retry_wrapper

@retry_requests
def get_request():
response = requests.get("https://this-url-does-not-exist-sorry-jwsgudkaad")
resp = response.json()
status_code = response.status_code
return resp, status_code

def main():
resp, status_code = get_request()

if __name__ == '__main__':
main()

Я хочу передать код status_code в декоратор, чтобы иметь возможность ссылаться на него в цикле While. Итак, это будет: `

Код: Выделить всё

while attempts < retries and status_code > 5:

У меня есть обходной путь, с помощью которого я могу установить ENV или глобальную переменную с кодом состояния. Однако мне было интересно, можно ли реализовать мою первоначальную идею в самой функции декоратора.

Код: Выделить всё

TypeError: retry_requests() missing 1 required positional argument: 'status_code'

Подробнее здесь: https://stackoverflow.com/questions/793 ... tor-functi