Skip to content

Commit cb4e815

Browse files
Make UploadProgressTracker public
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 34a613f commit cb4e815

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

cycode/cyclient/cycode_client_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def _should_retry_exception(exception: BaseException) -> bool:
9292
return is_request_error or is_server_error
9393

9494

95-
class _UploadProgressTracker:
95+
class UploadProgressTracker:
9696
"""File-like wrapper that tracks bytes read during upload and fires a progress callback."""
9797

9898
def __init__(self, data: bytes, callback: Optional[Callable[[int, int], None]]) -> None:
@@ -176,7 +176,7 @@ def _send_multipart(
176176
hide_response_content_log: bool,
177177
) -> Response:
178178
# Wrap the body in a fresh tracker each attempt so bytes_read starts from zero.
179-
tracker = _UploadProgressTracker(body, on_upload_progress)
179+
tracker = UploadProgressTracker(body, on_upload_progress)
180180
headers = self.get_request_headers({'Content-Type': content_type})
181181
try:
182182
response = _get_request_function()(

cycode/cyclient/scan_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
)
1616
from cycode.cli.files_collector.models.in_memory_zip import InMemoryZip
1717
from cycode.cyclient import models
18-
from cycode.cyclient.cycode_client_base import CycodeClientBase, _UploadProgressTracker
18+
from cycode.cyclient.cycode_client_base import CycodeClientBase, UploadProgressTracker
1919
from cycode.cyclient.logger import logger
2020

2121
if TYPE_CHECKING:
@@ -150,7 +150,7 @@ def upload_to_presigned_post(
150150
all_files['file'] = ('multiple_files_scan.zip', zip_file.read(), 'application/octet-stream')
151151

152152
prepared = requests.Request('POST', 'https://dummy', files=all_files).prepare()
153-
tracker = _UploadProgressTracker(prepared.body, on_upload_progress)
153+
tracker = UploadProgressTracker(prepared.body, on_upload_progress)
154154

155155
try:
156156
# We are not using Cycode client, as we are calling aws S3.

0 commit comments

Comments
 (0)