Add timeout argument for docker.image.push() method (#929)

* added timeout for push command * add a feature #929 file --------- Co-authored-by: costefan <konstantine.ovchynnikov@apolo.us>
aio-libs · Nov 20, 2024 · a40b374 · a40b374
1 parent ba05d80
commit a40b374
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/CHANGES/929.feature b/CHANGES/929.feature
@@ -0,0 +1 @@
+Added timeout parameter for push method
diff --git a/aiodocker/images.py b/aiodocker/images.py
@@ -157,6 +157,7 @@ async def push(
         auth: Optional[Union[JSONObject, str, bytes]] = None,
         tag: Optional[str] = None,
         stream: Literal[False] = False,
+        timeout: Union[float, Sentinel, None] = SENTINEL,
     ) -> Dict[str, Any]: ...
 
     @overload
@@ -167,6 +168,7 @@ def push(
         auth: Optional[Union[JSONObject, str, bytes]] = None,
         tag: Optional[str] = None,
         stream: Literal[True],
+        timeout: Union[float, Sentinel, None] = SENTINEL,
     ) -> AsyncIterator[Dict[str, Any]]: ...
 
     def push(
@@ -176,6 +178,7 @@ def push(
         auth: Optional[Union[JSONObject, str, bytes]] = None,
         tag: Optional[str] = None,
         stream: bool = False,
+        timeout: Union[float, Sentinel, None] = SENTINEL,
     ) -> Any:
         params = {}
         headers = {
@@ -197,6 +200,7 @@ def push(
             "POST",
             params=params,
             headers=headers,
+            timeout=timeout,
         )
         return self._handle_response(cm, stream)