From e6af58a4cc5c3ce8146d6ee3eacc1dec30813eb9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Me=C3=9Fmer?=
 <smessmer@users.noreply.github.com>
Date: Fri, 26 May 2017 03:32:17 +0100
Subject: [PATCH] Fix type stubs in locks.pyi (#1250)

---
 stdlib/3.4/asyncio/locks.pyi | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/stdlib/3.4/asyncio/locks.pyi b/stdlib/3.4/asyncio/locks.pyi
index fe53479b8..685402511 100644
--- a/stdlib/3.4/asyncio/locks.pyi
+++ b/stdlib/3.4/asyncio/locks.pyi
@@ -1,8 +1,9 @@
-from typing import Any, Callable, Generator, Iterable, Iterator, List, TypeVar, Union, Optional
+from typing import Any, Callable, Generator, Iterable, Iterator, List, Type, TypeVar, Union, Optional
 
 from .coroutines import coroutine
 from .events import AbstractEventLoop
-from .futures import Future
+from .futures import Future, Awaitable
+from types import TracebackType
 
 _T = TypeVar('_T')
 
@@ -17,18 +18,18 @@ class _ContextManagerMixin(Future[_ContextManager]):
     # Apparently this exists to *prohibit* use as a context manager.
     def __enter__(self) -> object: ...
     def __exit__(self, *args: Any) -> None: ...
-    def __aenter__(self): ...
-    def __aexit__(self, exc_type, exc, tb): ...
+    def __aenter__(self) -> Awaitable[None]: ...
+    def __aexit__(self, exc_type: Optional[Type[BaseException]], exc: Optional[BaseException], tb: Optional[TracebackType]) -> Awaitable[None]: ...
 
 class Lock(_ContextManagerMixin):
-    def __init__(self, *, loop: AbstractEventLoop = None) -> None: ...
+    def __init__(self, *, loop: Optional[AbstractEventLoop] = ...) -> None: ...
     def locked(self) -> bool: ...
     @coroutine
     def acquire(self) -> Generator[Any, None, bool]: ...
     def release(self) -> None: ...
 
 class Event:
-    def __init__(self, *, loop: AbstractEventLoop = None) -> None: ...
+    def __init__(self, *, loop: Optional[AbstractEventLoop] = ...) -> None: ...
     def is_set(self) -> bool: ...
     def set(self) -> None: ...
     def clear(self) -> None: ...
@@ -36,7 +37,7 @@ class Event:
     def wait(self) -> Generator[Any, None, bool]: ...
 
 class Condition(_ContextManagerMixin):
-    def __init__(self, lock: Lock = None, *, loop: AbstractEventLoop = None) -> None: ...
+    def __init__(self, lock: Lock = None, *, loop: Optional[AbstractEventLoop] = ...) -> None: ...
     def locked(self) -> bool: ...
     @coroutine
     def acquire(self) -> Generator[Any, None, bool]: ...