added test

fixed flake errors
renamed variables/attributes

Author: Patrik Kopkan

src/pip/_internal/commands/wheel.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from __future__ import absolute_import, with_statement
+from __future__ import absolute_import
 
 import logging
 import os
@@ -105,7 +105,6 @@ def __init__(self, *args, **kw):
             help="stores the filenames of wheels in file of given path"
         )
 
-
         index_opts = cmdoptions.make_option_group(
             cmdoptions.index_group,
             self.parser,
@@ -178,7 +177,7 @@ def run(self, options, args):
                         build_options=options.build_options or [],
                         global_options=options.global_options or [],
                         no_clean=options.no_clean,
-                        path_to_save_wheelnames = options.path_to_wheelnames
+                        path_to_wheelnames=options.path_to_wheelnames
                     )
                     build_failures = wb.build(
                         requirement_set.requirements.values(), session=session,
@@ -187,15 +186,14 @@ def run(self, options, args):
                         raise CommandError(
                             "Failed to build one or more wheels"
                         )
-                    if len(wb.wheel_filenames) !=0:
-                        try:
-                            with open(options.path_to_wheelnames, 'w') as file:
-                                output = ''
-                                for wheel_filename in wb.wheel_filenames:
-                                    output = wheel_filename + os.linesep
-                                file.write(output)
-                        except EnvironmentError as e:
-                            logger.error(str(e))
+                    if wb.path_to_wheelnames is not None:
+                        if len(wb.wheel_filenames) != 0:
+                            with open(wb.path_to_wheelnames, 'w') as file:
+                                file.write(
+                                    os.linesep.join(
+                                        wb.wheel_filenames
+                                    ) + os.linesep
+                                )
 
                 except PreviousBuildDirError:
                     options.no_clean = True

src/pip/_internal/wheel.py

@@ -851,8 +851,8 @@ def __init__(
         wheel_cache,  # type: WheelCache
         build_options=None,  # type: Optional[List[str]]
         global_options=None,  # type: Optional[List[str]]
-        no_clean=False, # type: bool
-        path_to_save_wheelnames=None # type: str
+        no_clean=False,  # type: bool
+        path_to_wheelnames=None  # Optional[str]
     ):
         # type: (...) -> None
         self.finder = finder
@@ -864,7 +864,7 @@ def __init__(
         self.build_options = build_options or []
         self.global_options = global_options or []
         self.no_clean = no_clean
-        self.path_to_save_wheelnames = path_to_save_wheelnames
+        self.path_to_wheelnames = path_to_wheelnames
         self.wheel_filenames = []
 
     def _build_one(self, req, output_dir, python_tag=None):

tests/functional/test_wheel.py

@@ -246,3 +246,22 @@ def test_legacy_wheels_are_not_confused_with_other_files(script, tmpdir, data):
     wheel_file_name = 'simplewheel-1.0-py%s-none-any.whl' % pyversion[0]
     wheel_file_path = script.scratch / wheel_file_name
     assert wheel_file_path in result.files_created, result.stdout
+
+
+def test_pip_option_save_wheel_name(script, data):
+    """
+    Test 'pip wheel ' success.
+    """
+    script.pip(
+        'wheel', '--no-index', '-f', data.find_links,
+        'require_simple==1.0',
+        '--save-wheel-name', './wheelnames',
+    )
+
+    wheel_file_names = ['require_simple-1.0-py%s-none-any.whl' % pyversion[0],
+                        'simple-3.0-py%s-none-any.whl' % pyversion[0],
+                        ]
+    wheelnames_path = script.scratch_path / 'wheelnames'
+    wheelnames_file = open(wheelnames_path, 'r')
+    wheelnames_entries = (wheelnames_file.read()).splitlines()
+    assert wheel_file_names == wheelnames_entries