[ADD] support context manager on Client

dgravitate · Nov 3, 2022 · c87acd1 · c87acd1
1 parent 9c5a016
commit c87acd1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/zeep/client.py b/src/zeep/client.py
@@ -214,6 +214,12 @@ def _get_service(self, name: typing.Optional[str]) -> str:
             service = next(iter(self.wsdl.services.values()), None)
         return service
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type=None, exc_value=None, traceback=None):
+        self.transport.close()
+
 
 class AsyncClient(Client):
     _default_transport = AsyncTransport