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")
status_code = response.status_code
return status_code

def main():
status_code = get_request()

if __name__ == '__main__':
main()

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

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

while attempts < retries and status_code > 5:

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

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

def retry_requests(func, retries=3):
def retry_wrapper(status_code, *args, **kwargs):
attempts = 0
while (attempts < retries) and (status_code > 500):
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(status_code)
def get_request():
response = requests.get("https://restful-booker.herokuapp.com/mamsma")
status_code = response.status_code
return status_code

который возвращает ошибку имени NameError: имя «status_code» не определено
Когда я определяю его внутри функции, я получаю TypeError : в функции retry_requests() отсутствует 1 обязательный позиционный аргумент: 'status_code'

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