From 22ae6b184ff35a54ccec44bff215d8bc77b68ec7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Elan=20Ruusam=C3=A4e?= Date: Sun, 14 Jan 2024 17:22:18 +0200 Subject: [PATCH] Use ClickException rather exit to stop program after rate_limit exceeded --- plextraktsync/decorators/rate_limit.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/plextraktsync/decorators/rate_limit.py b/plextraktsync/decorators/rate_limit.py index 040fcabe1de..5a82d4cc846 100644 --- a/plextraktsync/decorators/rate_limit.py +++ b/plextraktsync/decorators/rate_limit.py @@ -1,6 +1,7 @@ from functools import wraps from time import sleep +from click import ClickException from trakt.errors import RateLimitException from plextraktsync.factory import logger @@ -23,13 +24,12 @@ def wrapper(*args, **kwargs): except RateLimitException as e: if retry == retries: logger.error(f"Trakt Error: {e}") - logger.error( - "Trakt API didn't respond properly, script will abort now. Please try again later." - ) logger.error( f"Last call: {fn.__module__}.{fn.__name__}({args[1:]}, {kwargs})" ) - exit(1) + raise ClickException( + "Trakt API didn't respond properly, script will abort now. Please try again later." + ) seconds = e.retry_after retry += 1