Update entrypoint.py

Changes in backoff module

Author: BGarber42

entrypoint.py

@@ -4,6 +4,8 @@
 import sys
 
 import backoff
+from backoff import expo  # Import the 'expo' for exponential backoff
+from backoff.types import Details
 
 
 # Configure logging
@@ -15,10 +17,20 @@
 logger = logging.getLogger(__name__)
 
 
+def _log_backoff(details: Details) -> None:
+    """Logs backoff attempts."""
+    logger.warning(
+        f"Backing off {details['wait']:0.1f}s after {details['tries']} tries "
+        f"for function: {details['target'].__qualname__} "
+        f"with args: {details['args']} and kwargs: {details['kwargs']}"
+    )
+
+
 @backoff.on_exception(
-    backoff.expo,
+    expo,  # Use expo for exponential backoff
     (ConnectionRefusedError, socket.timeout),  # Handle timeout explicitly
     max_time=float(os.environ.get("INPUT_MAXTIME", "60")),  # Default to "60" and ensure numeric
+    on_backoff=_log_backoff,
     logger=logger,  # Integrate backoff with the configured logger
 )
 def netcat(host: str, port: int, content: str | None = None) -> None:
@@ -34,7 +46,7 @@ def netcat(host: str, port: int, content: str | None = None) -> None:
     try:
         with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
             s.settimeout(10)  # Add a timeout to the socket operations
-            s.connect((host, port))
+            s.connect((host, int(port)))
             if content:
                 s.sendall(content.encode())
             s.shutdown(socket.SHUT_WR)  # Inform server no more data will be sent
@@ -62,12 +74,12 @@ def main() -> None:
     Main function to execute the netcat operation and set output.
     """
     my_host = os.environ["INPUT_REMOTEHOST"]
-    my_port = int(os.environ["INPUT_REMOTEPORT"]) # Type conversion and error handling
+    my_port = os.environ["INPUT_REMOTEPORT"] # keep as string
     try:
-        netcat(my_host, my_port)
-    except backoff.exceptions.MaxRetriesError:  # Specific exception from backoff
-        logger.error("Max retries reached. Failed to connect.")
-        print(f"::error::Timed out waiting for connection after multiple retries")  # Github Action error annotation
+        netcat(my_host, int(my_port)) # Convert to int here
+    except backoff.exceptions.PermanentError as e: # Correct Exception type
+        logger.error(f"Failed to connect after retries: {e}")
+        print(f"::error::Failed to connect after retries: {e}")  # Github Action error annotation
         sys.exit(1)  # Indicate failure to github actions
     except Exception as e:
         logger.error(f"An unexpected error occurred: {e}")