Create mypy_extensions package.

Author: davidfstr

extensions/mypy_extensions/__init__.py

@@ -0,0 +1,18 @@
+from typing import cast, Dict, Type, TypeVar
+
+
+_T = TypeVar('_T')
+
+
+def TypedDict(typename: str, fields: Dict[str, Type[_T]]) -> Type[dict]:
+    """TypedDict creates a dictionary type that expects all of its
+    instances to have a certain common set of keys, with each key
+    associated with a value of a consistent type. This expectation
+    is not checked at runtime but is only enforced by typecheckers.
+    """
+    def new_dict(*args, **kwargs):
+        return dict(*args, **kwargs)
+
+    new_dict.__name__ = typename  # type: ignore  # https://github.com/python/mypy/issues/708
+    new_dict.__supertype__ = dict  # type: ignore  # https://github.com/python/mypy/issues/708
+    return cast(Type[dict], new_dict)

extensions/mypy_extensions/typing.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+
+from distutils.core import setup
+import sys
+
+if '..' not in sys.path:
+    sys.path.insert(0, '..')
+import setup as mypy_setup
+
+description = 'Experimental type system extensions for programs checked with the mypy typechecker.'
+long_description = '''
+Mypy Extensions
+===============
+
+The "mypy_extensions.typing" module defines experimental extensions to the
+standard "typing" module that are supported by the mypy typechecker.
+'''.lstrip()
+
+
+setup(
+    name='mypy_extensions',
+    version=mypy_setup.setup_kwargs['version'],
+    description=description,
+    long_description=long_description,
+    author=mypy_setup.setup_kwargs['author'],
+    author_email=mypy_setup.setup_kwargs['author_email'],
+    url=mypy_setup.setup_kwargs['url'],
+    license=mypy_setup.setup_kwargs['license'],
+    platforms=mypy_setup.setup_kwargs['platforms'],
+    package_dir={'mypy_extensions': 'mypy_extensions'},
+    py_modules=['mypy_extensions'],
+    packages=['mypy_extensions'],
+    scripts=[],
+    data_files=mypy_setup.find_data_files('mypy_extensions', ['*.py', '*.pyi']),
+    classifiers=mypy_setup.setup_kwargs['classifiers'],
+)

extensions/setup.py

@@ -1,18 +1,28 @@
 """Utilities for verifying git integrity."""
 
+from __future__ import print_function
+
 # Used also from setup.py, so don't pull in anything additional here (like mypy or typing):
 import os
 import pipes
 import subprocess
 import sys
 
 
-def is_git_repo(dir: str) -> bool:
+# NOTE: Cannot use typing.TYPE_CHECKING guard here because cannot import
+#       "typing" in this file at all.
+if False:
+    from typing import Iterator
+
+
+def is_git_repo(dir):
+    # type: (str) -> bool
     """Is the given directory version-controlled with git?"""
     return os.path.exists(os.path.join(dir, ".git"))
 
 
-def have_git() -> bool:
+def have_git():
+    # type: () -> bool
     """Can we run the git executable?"""
     try:
         subprocess.check_output(["git", "--help"])
@@ -23,7 +33,8 @@ def have_git() -> bool:
         return False
 
 
-def get_submodules(dir: str):
+def get_submodules(dir):
+    # type: (str) -> Iterator[str]
     """Return a list of all git top-level submodules in a given directory."""
     # It would be nicer to do
     # "git submodule foreach 'echo MODULE $name $path $sha1 $toplevel'"
@@ -41,67 +52,77 @@ def get_submodules(dir: str):
         yield name.decode(sys.getfilesystemencoding())
 
 
-def git_revision(dir: str) -> bytes:
+def git_revision(dir):
+    # type: (str) -> bytes
     """Get the SHA-1 of the HEAD of a git repository."""
     return subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=dir).strip()
 
 
-def submodule_revision(dir: str, submodule: str) -> bytes:
+def submodule_revision(dir, submodule):
+    # type: (str, str) -> bytes
     """Get the SHA-1 a submodule is supposed to have."""
     output = subprocess.check_output(["git", "ls-files", "-s", submodule], cwd=dir).strip()
     # E.g.: "160000 e4a7edb949e0b920b16f61aeeb19fc3d328f3012 0       typeshed"
     return output.split()[1]
 
 
-def is_dirty(dir: str) -> bool:
+def is_dirty(dir):
+    # type: (str) -> bool
     """Check whether a git repository has uncommitted changes."""
     output = subprocess.check_output(["git", "status", "-uno", "--porcelain"], cwd=dir)
     return output.strip() != b""
 
 
-def has_extra_files(dir: str) -> bool:
+def has_extra_files(dir):
+    # type: (str) -> bool
     """Check whether a git repository has untracked files."""
     output = subprocess.check_output(["git", "clean", "--dry-run", "-d"], cwd=dir)
     return output.strip() != b""
 
 
-def warn_no_git_executable() -> None:
+def warn_no_git_executable():
+    # type: () -> None
     print("Warning: Couldn't check git integrity. "
           "git executable not in path.", file=sys.stderr)
 
 
-def warn_dirty(dir: str) -> None:
+def warn_dirty(dir):
+    # type: (str) -> None
     print("Warning: git module '{}' has uncommitted changes.".format(dir),
           file=sys.stderr)
     print("Go to the directory", file=sys.stderr)
     print("  {}".format(dir), file=sys.stderr)
     print("and commit or reset your changes", file=sys.stderr)
 
 
-def warn_extra_files(dir: str) -> None:
+def warn_extra_files(dir):
+    # type: (str) -> None
     print("Warning: git module '{}' has untracked files.".format(dir),
           file=sys.stderr)
     print("Go to the directory", file=sys.stderr)
     print("  {}".format(dir), file=sys.stderr)
     print("and add & commit your new files.", file=sys.stderr)
 
 
-def chdir_prefix(dir: str) -> str:
+def chdir_prefix(dir):
+    # type: (str) -> str
     """Return the command to change to the target directory, plus '&&'."""
     if os.path.relpath(dir) != ".":
         return "cd " + pipes.quote(dir) + " && "
     else:
         return ""
 
 
-def error_submodule_not_initialized(name: str, dir: str) -> None:
+def error_submodule_not_initialized(name, dir):
+    # type: (str, str) -> None
     print("Submodule '{}' not initialized.".format(name), file=sys.stderr)
     print("Please run:", file=sys.stderr)
     print("  {}git submodule update --init {}".format(
         chdir_prefix(dir), name), file=sys.stderr)
 
 
-def error_submodule_not_updated(name: str, dir: str) -> None:
+def error_submodule_not_updated(name, dir):
+    # type: (str, str) -> None
     print("Submodule '{}' not updated.".format(name), file=sys.stderr)
     print("Please run:", file=sys.stderr)
     print("  {}git submodule update {}".format(
@@ -110,7 +131,8 @@ def error_submodule_not_updated(name: str, dir: str) -> None:
     print(" then run \"git add {}\" to silence this check)".format(name), file=sys.stderr)
 
 
-def verify_git_integrity_or_abort(datadir: str) -> None:
+def verify_git_integrity_or_abort(datadir):
+    # type: (str) -> None
     """Verify the (submodule) integrity of a git repository.
 
     Potentially output warnings/errors (to stderr), and exit with status 1

mypy/git.py

@@ -5,9 +5,10 @@
 import os.path
 import sys
 
-if sys.version_info < (3, 2, 0):
-    sys.stderr.write("ERROR: You need Python 3.2 or later to use mypy.\n")
-    exit(1)
+if __name__ == '__main__':
+    if sys.version_info < (3, 2, 0):
+        sys.stderr.write("ERROR: You need Python 3.2 or later to use mypy.\n")
+        exit(1)
 
 from distutils.core import setup
 from distutils.command.build_py import build_py
@@ -79,9 +80,7 @@ def run(self):
 
 
 data_files = []
-
 data_files += find_data_files('typeshed', ['*.py', '*.pyi'])
-
 data_files += find_data_files('xml', ['*.xsd', '*.xslt', '*.css'])
 
 classifiers = [
@@ -105,20 +104,24 @@ def run(self):
 if os.name == 'nt':
     scripts.append('scripts/mypy.bat')
 
-setup(name='mypy-lang',
-      version=version,
-      description=description,
-      long_description=long_description,
-      author='Jukka Lehtosalo',
-      author_email='[email protected]',
-      url='http://www.mypy-lang.org/',
-      license='MIT License',
-      platforms=['POSIX'],
-      package_dir=package_dir,
-      py_modules=['typing'] if sys.version_info < (3, 5, 0) else [],
-      packages=['mypy'],
-      scripts=scripts,
-      data_files=data_files,
-      classifiers=classifiers,
-      cmdclass={'build_py': CustomPythonBuild},
-      )
+setup_kwargs = dict(
+    name='mypy-lang',
+    version=version,
+    description=description,
+    long_description=long_description,
+    author='Jukka Lehtosalo',
+    author_email='[email protected]',
+    url='http://www.mypy-lang.org/',
+    license='MIT License',
+    platforms=['POSIX'],
+    package_dir=package_dir,
+    py_modules=['typing'] if sys.version_info < (3, 5, 0) else [],
+    packages=['mypy'],
+    scripts=scripts,
+    data_files=data_files,
+    classifiers=classifiers,
+    cmdclass={'build_py': CustomPythonBuild},
+)
+
+if __name__ == '__main__':
+    setup(**setup_kwargs)