Clear decorators if we skip a function

JukkaL · JukkaL · commit 8785da917a4a · 2019-11-08T14:23:29.000Z
diff --git a/mypy/stubgen.py b/mypy/stubgen.py
@@ -538,11 +538,10 @@ def visit_mypy_file(self, o: MypyFile) -> None:
                 self.add('#   %s\n' % name)
 
     def visit_func_def(self, o: FuncDef, is_abstract: bool = False) -> None:
-        if self.is_private_name(o.name(), o.fullname()):
-            return
-        if self.is_not_in_all(o.name()):
-            return
-        if self.is_recorded_name(o.name()):
+        if (self.is_private_name(o.name(), o.fullname())
+                or self.is_not_in_all(o.name())
+                or self.is_recorded_name(o.name())):
+            self.clear_decorators()
             return
         if not self._indent and self._state not in (EMPTY, FUNC) and not o.is_awaitable_coroutine:
             self.add('\n')
diff --git a/test-data/unit/stubgen.test b/test-data/unit/stubgen.test
@@ -1262,6 +1262,24 @@ class F:
 @t.coroutine
 def g(): ...
 
+[case testCoroutineSpecialCase_import]
+import asyncio
+
+__all__ = ['C']
+
+@asyncio.coroutine
+def f():
+    pass
+
+class C:
+    def f(self):
+        pass
+[out]
+import asyncio
+
+class C:
+    def f(self) -> None: ...
+
 -- Tests for stub generation from semantically analyzed trees.
 -- These tests are much slower, so use the `_semanal` suffix only when needed.
 
