types

Author: dklimpel

podman/api/parse_utils.py

@@ -8,7 +8,7 @@
 from typing import Any, Optional, Union
 from collections.abc import Iterator
 
-from requests import Response
+from podman.api.client import APIResponse
 from .output_utils import demux_output
 
 
@@ -67,7 +67,7 @@ def prepare_cidr(value: Union[ipaddress.IPv4Network, ipaddress.IPv6Network]) ->
     return str(value.network_address), base64.b64encode(value.netmask.packed).decode("utf-8")
 
 
-def frames(response: Response) -> Iterator[bytes]:
+def frames(response: APIResponse) -> Iterator[bytes]:
     """Returns each frame from multiplexed payload, all results are expected in the payload.
 
     The stdout and stderr frames are undifferentiated as they are returned.
@@ -84,7 +84,7 @@ def frames(response: Response) -> Iterator[bytes]:
 
 
 def stream_frames(
-    response: Response, demux: bool = False
+    response: APIResponse, demux: bool = False
 ) -> Iterator[Union[bytes, tuple[bytes, bytes]]]:
     """Returns each frame from multiplexed streamed payload.
 
@@ -111,7 +111,7 @@ def stream_frames(
 
 
 def stream_helper(
-    response: Response, decode_to_json: bool = False
+    response: APIResponse, decode_to_json: bool = False
 ) -> Union[Iterator[bytes], Iterator[dict[str, Any]]]:
     """Helper to stream results and optionally decode to json"""
     for value in response.iter_lines():