Default to python3 when using python_venv under python 2

asottile · asottile · commit 0eaacd7c8e10 · 2018-07-17T18:24:29.000-07:00
diff --git a/pre_commit/languages/python_venv.py b/pre_commit/languages/python_venv.py
@@ -1,6 +1,7 @@
 from __future__ import unicode_literals
 
 import os.path
+import sys
 
 from pre_commit.languages import python
 from pre_commit.util import CalledProcessError
@@ -10,6 +11,13 @@
 ENVIRONMENT_DIR = 'py_venv'
 
 
+def get_default_version():  # pragma: no cover (version specific)
+    if sys.version_info < (3,):
+        return 'python3'
+    else:
+        return python.get_default_version()
+
+
 def orig_py_exe(exe):  # pragma: no cover (platform specific)
     """A -mvenv virtualenv made from a -mvirtualenv virtualenv installs
     packages to the incorrect location.  Attempt to find the _original_ exe
@@ -43,6 +51,5 @@ def make_venv(envdir, python):
     cmd_output(orig_py_exe(python), '-mvenv', envdir, cwd='/')
 
 
-get_default_version = python.get_default_version
 _interface = python.py_interface(ENVIRONMENT_DIR, make_venv)
 in_env, healthy, run_hook, install_environment = _interface
