feat(error-handling): enhance error handling with detailed logging and flexible decorator

Author: Einswilli

forklet/infrastructure/error_handler.py

@@ -11,23 +11,17 @@
 from forklet.infrastructure.logger import logger
 
 
-
-
 ####
 ##      DOWNLOAD ERROR MODEL
 #####
 class DownloadError(Exception):
     """Base exception for download-related errors."""
-    
-    def __init__(
-        self, 
-        message: str, 
-        original_error: Optional[Exception] = None
-    ):
+
+    def __init__(self, message: str, original_error: Optional[Exception] = None):
         super().__init__(message)
         self.original_error = original_error
         self.message = message
-    
+
     def __str__(self) -> str:
         if self.original_error:
             return f"{self.message} (Original: {self.original_error})"
@@ -39,6 +33,7 @@ def __str__(self) -> str:
 #####
 class RateLimitError(DownloadError):
     """Exception raised when rate limits are exceeded."""
+
     pass
 
 
@@ -47,6 +42,7 @@ class RateLimitError(DownloadError):
 #####
 class AuthenticationError(DownloadError):
     """Exception raised for authentication failures."""
+
     pass
 
 
@@ -55,20 +51,20 @@ class AuthenticationError(DownloadError):
 #####
 class RepositoryNotFoundError(DownloadError):
     """Exception raised when repository is not found."""
-    pass
 
+    pass
 
 
 ####
-##      RROR HANDLER UTILITIES
+##      ERROR HANDLER UTILITIES
 #####
 def handle_api_error(func: Callable) -> Callable:
     """
     Decorator to handle API errors and convert to appropriate exceptions.
-    
+
     Args:
         func: Function to decorate
-        
+
     Returns:
         Decorated function
     """
@@ -81,8 +77,7 @@ def wrapper(*args, **kwargs) -> Any:
 
         # Cse of GH Exceptions
         except GithubException as e:
-
-            if e.status == 403 and 'rate limit' in str(e).lower():
+            if e.status == 403 and "rate limit" in str(e).lower():
                 raise RateLimitError("GitHub API rate limit exceeded", e) from e
 
             elif e.status == 401 or e.status == 403:
@@ -96,45 +91,39 @@ def wrapper(*args, **kwargs) -> Any:
 
         # Request Exceptions
         except httpx.RequestError as e:
-
-            if '429' in str(e) or 'rate limit' in str(e).lower():
+            if "429" in str(e) or "rate limit" in str(e).lower():
                 raise RateLimitError("Rate limit exceeded", e) from e
 
             else:
                 raise DownloadError(f"Network error: {e}", e) from e
 
         except Exception as e:
             raise DownloadError(f"Unexpected error: {e}", e) from e
-    
+
     return wrapper
 
 
 def retry_on_error(max_retries: int = 3) -> Callable:
     """
     Decorator to retry operations on specific errors.
-    
+
     Args:
         max_retries: Maximum number of retry attempts
-        
+
     Returns:
-        Decorated function
+        Decorator function
     """
 
     def decorator(func: Callable) -> Callable:
-        
+
         @functools.wraps(func)
         def wrapper(*args, **kwargs) -> Any:
             last_exception = None
-            
+
             for attempt in range(max_retries + 1):
                 try:
                     return func(*args, **kwargs)
-                except (
-                    RateLimitError, 
-                    httpx.RequestError, 
-                    ConnectionError
-                ) as e:
-                    
+                except (RateLimitError, httpx.RequestError, ConnectionError) as e:
                     last_exception = e
                     if attempt < max_retries:
                         logger.warning(
@@ -147,7 +136,9 @@ def wrapper(*args, **kwargs) -> Any:
                     # Don't retry on other errors
                     logger.error(f"Non-retryable error: {e}")
                     raise
-            
+
             raise last_exception or Exception("All retry attempts failed")
+
         return wrapper
+
     return decorator