PR: Support mypy

MANIFEST.in

@@ -3,3 +3,4 @@ include CHANGELOG.md
 include LICENSE.txt
 include README.md
 recursive-include qtpy/tests *.py *.ui
+include qtpy/py.typed

README.md

@@ -71,6 +71,24 @@ conda install qtpy
 ```
 
 
+### mypy
+
+A CLI is offered to help with usage of QtPy including the subcommand
+`qtpy mypy args` which generates command line arguments similar to the
+following which help guide mypy through which library QtPy would have used
+so that mypy can get the proper underlying type hints.
+
+```
+--always-false=PYQT4 --always-false=PYQT5 --always-false=PYSIDE --always-true=PYSIDE2
+```
+
+Use such as:
+
+```bash
+env/bin/mypy --package mypackage $(env/bin/qtpy mypy args)
+```
+
+
 ## Contributing
 
 Everyone is welcome to contribute!

qtpy/__init__.py

@@ -220,8 +220,10 @@ class PythonQtWarning(Warning):
     warnings.warn('Selected binding "{}" could not be found, '
                   'using "{}"'.format(initial_api, API), RuntimeWarning)
 
-API_NAME = {'pyqt5': 'PyQt5', 'pyqt': 'PyQt4', 'pyqt4': 'PyQt4',
-            'pyside': 'PySide', 'pyside2':'PySide2'}[API]
+API_NAMES = {'pyqt5': 'PyQt5', 'pyqt': 'PyQt4', 'pyqt4': 'PyQt4',
+            'pyside': 'PySide', 'pyside2':'PySide2'}
+APIS = sorted(name.upper() for name in set(API_NAMES.values()))
+API_NAME = API_NAMES[API]
 
 if PYQT4:
     import sip

qtpy/__main__.py

@@ -0,0 +1,3 @@
+import qtpy.cli
+
+qtpy.cli.cli()

qtpy/cli.py

@@ -0,0 +1,42 @@
+import click
+
+
+@click.group()
+def cli():
+    """Features in support of development with QtPy."""
+
+    pass
+
+
+@cli.group()
+def mypy():
+    """Features in support of running mypy"""
+
+    pass
+
+
+@mypy.command()
+def args():
+    """Generate command line arguments for using mypy with QtPy.
+
+    This will generate strings similar to the following which help guide mypy
+    through which library QtPy would have used so that mypy can get the proper
+    underlying type hints.
+
+        --always-false=PYQT4 --always-false=PYQT5 --always-false=PYSIDE --always-true=PYSIDE2
+
+    Use such as:
+
+        env/bin/mypy --package mypackage $(env/bin/qtpy mypy args)
+    """
+
+    options = {False: '--always-false', True: '--always-true'}
+
+    import qtpy
+
+    apis_active = {name: qtpy.API.lower() == name.lower() for name in qtpy.APIS}
+    print(' '.join(
+        f'{options[is_active]}={name.upper()}'
+        for name, is_active
+        in apis_active.items()
+    ))

qtpy/tests/test_cli.py

@@ -0,0 +1,45 @@
+from __future__ import absolute_import
+
+import subprocess
+import sys
+
+import pytest
+
+import qtpy
+
+
+subcommands = [
+    ['mypy'],
+    ['mypy', 'args'],
+]
+
+
+@pytest.mark.parametrize(
+    argnames=['subcommand'],
+    argvalues=[[subcommand] for subcommand in subcommands],
+    ids=[' '.join(subcommand) for subcommand in subcommands],
+)
+def test_cli_help_does_not_fail(subcommand):
+    # .check_call() over .run(..., check=True) because of py2
+    subprocess.check_call(
+        [sys.executable, '-m', 'qtpy', *subcommand, '--help'],
+    )
+
+
+def test_cli_mypy_args():
+    output = subprocess.check_output(
+        [sys.executable, '-m', 'qtpy', 'mypy', 'args'],
+    )
+
+    if qtpy.PYQT4:
+        expected = b'--always-true=PYQT4 --always-false=PYQT5 --always-false=PYSIDE --always-false=PYSIDE2\n'
+    elif qtpy.PYQT5:
+        expected = b'--always-false=PYQT4 --always-true=PYQT5 --always-false=PYSIDE --always-false=PYSIDE2\n'
+    elif qtpy.PYSIDE:
+        expected = b'--always-false=PYQT4 --always-false=PYQT5 --always-true=PYSIDE --always-false=PYSIDE2\n'
+    elif qtpy.PYSIDE2:
+        expected = b'--always-false=PYQT4 --always-false=PYQT5 --always-false=PYSIDE --always-true=PYSIDE2\n'
+    else:
+        assert False, 'No valid API to test'
+
+    assert output == expected

setup.py

@@ -47,5 +47,8 @@
         'Programming Language :: Python :: 3',
         'Programming Language :: Python :: 3.3',
         'Programming Language :: Python :: 3.4',
-        'Programming Language :: Python :: 3.5']
+        'Programming Language :: Python :: 3.5'],
+    extras_require={'cli': ['click']},
+    entry_points={'console_scripts': 'qtpy = qtpy.cli:cli [cli]'},
+    include_package_data=True
 )