Update stubutil.py is_c_module with .pyd check (python#5284)

AlexArcPy · gvanrossum · commit f7f3887f2d3a · 2018-06-27T15:34:08.000-07:00
Fixes python#5283.
diff --git a/mypy/stubutil.py b/mypy/stubutil.py
@@ -1,5 +1,6 @@
 import re
 import sys
+import os
 
 from typing import Optional, Tuple, Sequence, MutableSequence, List, MutableMapping, IO
 from types import ModuleType
@@ -87,7 +88,8 @@ def find_unique_signatures(sigs: Sequence[Sig]) -> List[Sig]:
 
 
 def is_c_module(module: ModuleType) -> bool:
-    return '__file__' not in module.__dict__ or module.__dict__['__file__'].endswith('.so')
+    return ('__file__' not in module.__dict__ or
+            os.path.splitext(module.__dict__['__file__'])[-1] in ['.so', '.pyd'])
 
 
 def write_header(file: IO[str], module_name: Optional[str] = None,
