Fix linting

Signed-off-by: Andre Kurait <akurait@amazon.com>

TrafficCapture/dockerSolution/src/main/docker/migrationConsole/lib/console_link/console_link/models/snapshot.py

@@ -248,8 +248,7 @@ def get_snapshot_status_message(snapshot_info: Dict) -> str:
     stats = snapshot_info.get("stats", {})
 
     total_size_bytes = stats.get("total", {}).get("size_in_bytes", 0)
-    processed_size_bytes = stats.get("processed", {}).get("size_in_bytes",
-        # Processed size may not be returned for SUCCESS snapshots, in which case we set it to the full size
+    processed_size_bytes = stats.get("processed", {}).get("size_in_bytes", 
         total_size_bytes if snapshot_state == "SUCCESS" else 0
     )
 
@@ -274,11 +273,10 @@ def get_snapshot_status_message(snapshot_info: Dict) -> str:
     remaining_duration = ((duration_millis / percent_completed) * (100 - percent_completed)) if percent_completed > 0 else None
 
     # Calculate throughput
-    throughput_mib_per_sec = (
-        (processed_size_bytes / (1024 ** 2)) / (duration_millis / 1000)
-        if duration_millis > 0
-        else None
-    )
+    if duration_millis > 0:
+        throughput_mib_per_sec = (processed_size_bytes / (1024 * 1024)) / (duration_millis / 1000)
+    else:
+        throughput_mib_per_sec = 0
 
     # Determine anticipated duration remaining and throughput format
     if duration_millis > 1000 and percent_completed > 0 and remaining_duration is not None:
@@ -289,16 +287,18 @@ def get_snapshot_status_message(snapshot_info: Dict) -> str:
         throughput_mib_per_sec = "N/A (not enough data to compute)"
 
     # Compile status message
-    return (f"Snapshot is {snapshot_state}.\n"
-            f"Percent completed: {percent_completed:.2f}%\n"
-            f"Data GiB done: {processed_size_gib:.3f}/{total_size_gib:.3f}\n"
-            f"Total shards: {total_shards}\n"
-            f"Successful shards: {successful_shards}\n"
-            f"Failed shards: {failed_shards}\n"
-            f"Start time: {start_time_formatted}\n"
-            f"Duration: {duration_formatted}\n"
-            f"Anticipated duration remaining: {anticipated_remaining_formatted}\n"
-            f"Throughput: {throughput_mib_per_sec}")
+    return (
+        f"Snapshot is {snapshot_state}.\n"
+        f"Percent completed: {percent_completed:.2f}%\n"
+        f"Data GiB done: {processed_size_gib:.3f}/{total_size_gib:.3f}\n"
+        f"Total shards: {total_shards}\n"
+        f"Successful shards: {successful_shards}\n"
+        f"Failed shards: {failed_shards}\n"
+        f"Start time: {start_time_formatted}\n"
+        f"Duration: {duration_formatted}\n"
+        f"Anticipated duration remaining: {anticipated_remaining_formatted}\n"
+        f"Throughput: {throughput_mib_per_sec}"
+    )
 
 
 def get_snapshot_status_full(cluster: Cluster, snapshot: str,

TrafficCapture/dockerSolution/src/main/docker/migrationConsole/lib/console_link/tests/test_snapshot.py

@@ -9,7 +9,8 @@
 from console_link.models.factories import (UnsupportedSnapshotError,
                                            get_snapshot)
 from console_link.models.snapshot import (FileSystemSnapshot, S3Snapshot,
-                                          Snapshot, delete_snapshot)
+                                          Snapshot, delete_snapshot,
+                                          get_snapshot_status_message)
 from tests.utils import create_valid_cluster
 
 
@@ -419,7 +420,7 @@ def test_handling_extra_args(mocker, request, snapshot_fixture):
 
 import unittest
 from unittest.mock import MagicMock
-from console_link.models.snapshot import get_snapshot_status_full, get_snapshot_status_message
+
 
 class TestSnapshot(unittest.TestCase):
     def setUp(self):