Type annotations for pip._internal.commands.freeze
-# 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
