From d260bb488930659a6abe6a5705a49adb922b8974 Mon Sep 17 00:00:00 2001
From: Harmen <harm@dds.nl>
Date: Sat, 24 Jun 2017 18:11:12 +0200
Subject: [PATCH] support for `with socket.create_connection` (#1429)

---
 stdlib/2and3/socket.pyi | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/stdlib/2and3/socket.pyi b/stdlib/2and3/socket.pyi
index 79e03cf21..37488fd11 100644
--- a/stdlib/2and3/socket.pyi
+++ b/stdlib/2and3/socket.pyi
@@ -6,7 +6,8 @@
 # see: http://nullege.com/codes/search/socket
 # adapted for Python 2.7 by Michal Pokorny
 import sys
-from typing import Any, Tuple, List, Optional, Union, overload
+from typing import Any, Tuple, List, Optional, Union, overload, TypeVar
+
 
 # ----- variables and constants -----
 
@@ -491,6 +492,7 @@ class timeout(error):
 
 # TODO AF_PACKET and AF_BLUETOOTH address objects
 
+_SelfT = TypeVar('_SelfT', bound=socket)
 
 # ----- classes -----
 class socket:
@@ -505,6 +507,10 @@ class socket:
         def __init__(self, family: int = ..., type: int = ...,
                      proto: int = ..., fileno: Optional[int] = ...) -> None: ...
 
+    if sys.version_info >= (3, 2):
+        def __enter__(self: _SelfT) -> _SelfT: ...
+        def __exit__(self, *args: Any) -> None: ...
+
     # --- methods ---
     # second tuple item is an address
     def accept(self) -> Tuple['socket', Any]: ...