Skip to content

Commit 0eeae21

Browse files
reducing complexity of methods
1 parent 7c0d03c commit 0eeae21

File tree

2 files changed

+20
-25
lines changed

2 files changed

+20
-25
lines changed

backend/backend/celery_db_retry.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -105,16 +105,14 @@ def _log_retry_attempt(
105105
)
106106

107107

108-
def _handle_non_sqlalchemy_error(error: BaseException) -> None:
109-
"""Handle non-SQLAlchemy errors by logging and re-raising."""
108+
def _log_non_sqlalchemy_error(error: BaseException) -> None:
109+
"""Log non-SQLAlchemy errors."""
110110
logger.debug(LogMessages.format_message(LogMessages.NON_RETRYABLE_ERROR, error=error))
111-
raise
112111

113112

114-
def _handle_non_retryable_error(error: BaseException) -> None:
115-
"""Handle non-retryable errors by logging and re-raising."""
113+
def _log_non_retryable_error(error: BaseException) -> None:
114+
"""Log non-retryable errors."""
116115
logger.debug(LogMessages.format_message(LogMessages.NON_RETRYABLE_ERROR, error=error))
117-
raise
118116

119117

120118
def _execute_retry_attempt(
@@ -146,16 +144,15 @@ def _execute_retry_attempt(
146144
return retry_count, delay
147145

148146

149-
def _handle_max_retries_exceeded(error: BaseException, total_retries: int) -> None:
150-
"""Handle the case when max retries are exceeded."""
147+
def _log_max_retries_exceeded(error: BaseException, total_retries: int) -> None:
148+
"""Log when max retries are exceeded."""
151149
logger.error(
152150
LogMessages.format_message(
153151
LogMessages.MAX_RETRIES_EXCEEDED,
154152
total=total_retries,
155153
error=error,
156154
)
157155
)
158-
raise
159156

160157

161158
def _process_celery_exception(
@@ -172,13 +169,15 @@ def _process_celery_exception(
172169
None: if should not retry (will re-raise)
173170
"""
174171
if not _is_sqlalchemy_error(error):
175-
_handle_non_sqlalchemy_error(error)
172+
_log_non_sqlalchemy_error(error)
173+
raise
176174

177175
# Use centralized error classification
178176
error_type, needs_refresh = DatabaseErrorPatterns.classify_error(error)
179177

180178
if not DatabaseErrorPatterns.is_retryable_error(error_type):
181-
_handle_non_retryable_error(error)
179+
_log_non_retryable_error(error)
180+
raise
182181

183182
# Get appropriate retry settings for this error type
184183
current_settings = _get_retry_settings_for_error(
@@ -190,8 +189,8 @@ def _process_celery_exception(
190189
error, error_type, needs_refresh, retry_count, current_settings
191190
)
192191
else:
193-
_handle_max_retries_exceeded(error, current_settings["max_retries"] + 1)
194-
return None # This line will never be reached due to exception, but added for completeness
192+
_log_max_retries_exceeded(error, current_settings["max_retries"] + 1)
193+
raise
195194

196195

197196
def _log_success(retry_count: int) -> None:

backend/utils/db_retry.py

Lines changed: 8 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -314,24 +314,20 @@ def operation():
314314
return decorator
315315

316316

317-
def _handle_context_non_retryable_error(error: BaseException) -> None:
318-
"""Handle non-retryable errors in context manager."""
317+
def _log_context_non_retryable_error(error: BaseException) -> None:
318+
"""Log non-retryable errors in context manager."""
319319
logger.debug(LogMessages.format_message(LogMessages.NON_RETRYABLE_ERROR, error=error))
320-
raise
321320

322321

323-
def _handle_context_max_retries_exceeded(
324-
error: BaseException, total_retries: int
325-
) -> None:
326-
"""Handle max retries exceeded in context manager."""
322+
def _log_context_max_retries_exceeded(error: BaseException, total_retries: int) -> None:
323+
"""Log max retries exceeded in context manager."""
327324
logger.error(
328325
LogMessages.format_message(
329326
LogMessages.MAX_RETRIES_EXCEEDED,
330327
total=total_retries,
331328
error=error,
332329
)
333330
)
334-
raise
335331

336332

337333
def _update_context_retry_settings(
@@ -426,7 +422,8 @@ def db_retry_context(
426422
error_type, needs_refresh, use_extended = _classify_database_error(e)
427423

428424
if not DatabaseErrorPatterns.is_retryable_error(error_type):
429-
_handle_context_non_retryable_error(e)
425+
_log_context_non_retryable_error(e)
426+
raise
430427

431428
# Update settings for extended retry if needed
432429
current_settings = _update_context_retry_settings(
@@ -447,9 +444,8 @@ def db_retry_context(
447444
time.sleep(delay)
448445
continue
449446
else:
450-
_handle_context_max_retries_exceeded(
451-
e, current_settings["max_retries"] + 1
452-
)
447+
_log_context_max_retries_exceeded(e, current_settings["max_retries"] + 1)
448+
raise
453449

454450

455451
def retry_database_operation(

0 commit comments

Comments
 (0)