Type annotations for pip._internal.commands.freeze

src/pip/_internal/commands/freeze.py

@@ -1,6 +1,3 @@
-# The following comment should be removed at some point in the future.
-# mypy: disallow-untyped-defs=False
-
 from __future__ import absolute_import
 
 import sys
@@ -8,12 +5,18 @@ import sys
 from pip._internal.cache import WheelCache
 from pip._internal.cli import cmdoptions
 from pip._internal.cli.base_command import Command
+from pip._internal.cli.status_codes import SUCCESS
 from pip._internal.models.format_control import FormatControl
 from pip._internal.operations.freeze import freeze
 from pip._internal.utils.compat import stdlib_pkgs
+from pip._internal.utils.typing import MYPY_CHECK_RUNNING
 
 DEV_PKGS = {'pip', 'setuptools', 'distribute', 'wheel'}
 
+if MYPY_CHECK_RUNNING:
+    from optparse import Values
+    from typing import Any, List
+
 
 class FreezeCommand(Command):
     """
@@ -27,6 +30,7 @@ class FreezeCommand(Command):
     log_streams = ("ext://sys.stderr", "ext://sys.stderr")
 
     def __init__(self, *args, **kw):
+        # type: (*Any, **Any) -> None
         super(FreezeCommand, self).__init__(*args, **kw)
 
         self.cmd_opts.add_option(
@@ -75,6 +79,7 @@ class FreezeCommand(Command):
         self.parser.insert_option_group(0, self.cmd_opts)
 
     def run(self, options, args):
+        # type: (Values, List[str]) -> int
         format_control = FormatControl(set(), set())
         wheel_cache = WheelCache(options.cache_dir, format_control)
         skip = set(stdlib_pkgs)
@@ -97,3 +102,4 @@ class FreezeCommand(Command):
 
         for line in freeze(**freeze_kwargs):
             sys.stdout.write(line + '\n')
+        return SUCCESS