Add support for running with Python 3.6 and 3.7

.travis.yml

@@ -13,8 +13,18 @@ script:
   # Linux Precise and Trusty cannot install Python 3.7 due to an outdated OpenSSL,
   # so we set this variable to allow skipping Python 3.7 on these shards.
   - python37_present="$(if which python3.7 >/dev/null; then echo 'true'; else echo 'false'; fi)"
-  - ./ci.py --pants-version unspecified
-  - ./ci.py --pants-version config
+  - ./ci.py --pants-version unspecified --python-version unspecified
+  # TODO: these entries will not work until 1.15.0 is cut, as ./pants uses 1.14.0 currently
+  # when pants_version is unspecified. Add this test back once 1.15.0 is released.
+  # - ./ci.py --pants-version unspecified --python-version 2.7
+  # - ./ci.py --pants-version unspecified --python-version 3.6
+  # - if [[ "${python37_present}" = 'true' ]]; then
+  #    ./ci.py --pants-version unspecified --python-version 3.7; fi
+  - ./ci.py --pants-version config --python-version unspecified
+  - ./ci.py --pants-version config --python-version 2.7
+  - ./ci.py --pants-version config --python-version 3.6
+  - if [[ "${python37_present}" = 'true' ]]; then
+      ./ci.py --pants-version config --python-version 3.7; fi
 
 osx_setup: &osx_setup
   os: osx

ci.py

@@ -22,9 +22,19 @@ def __str__(self):
       return self.value
 
 
+class PythonVersion(Enum):
+  unspecified = "unspecified"
+  py27 = "2.7"
+  py36 = "3.6"
+  py37 = "3.7"
+
+  def __str__(self):
+    return self.value
+
+
 def main() -> None:
   args = create_parser().parse_args()
-  run_tests(test_pants_version=args.pants_version)
+  run_tests(test_pants_version=args.pants_version, test_python_version=args.python_version)
 
 
 def create_parser() -> argparse.ArgumentParser:
@@ -37,41 +47,57 @@ def create_parser() -> argparse.ArgumentParser:
       required=True,
       help="Pants version to configure ./pants to use."
   )
+  parser.add_argument(
+      "--python-version",
+      action="store",
+      type=PythonVersion,
+      choices=list(PythonVersion),
+      required=True,
+      help="Python version to configure ./pants to use."
+  )
   return parser
 
 
-def run_tests(*, test_pants_version: PantsVersion) -> None:
+def run_tests(*, test_pants_version: PantsVersion, test_python_version: PythonVersion) -> None:
   version_command = ["./pants", "--version"]
   list_command = ["./pants", "list", "::"]
   env_with_pantsd = {**os.environ, "PANTS_ENABLE_PANTSD": "True"}
   with setup_pants_version(test_pants_version):
-    subprocess.run(version_command, check=True)
-    subprocess.run(list_command, check=True)
-    subprocess.run(version_command, env=env_with_pantsd, check=True)
-    subprocess.run(list_command, env=env_with_pantsd, check=True)
+    with setup_python_version(test_python_version):
+      subprocess.run(version_command, check=True)
+      subprocess.run(list_command, check=True)
+      subprocess.run(version_command, env=env_with_pantsd, check=True)
+      subprocess.run(list_command, env=env_with_pantsd, check=True)
 
 
 @contextmanager
 def setup_pants_version(test_pants_version: PantsVersion):
   """Modify pants.ini to allow the pants version to be unspecified or keep what was originally there."""
-  original_config = read_config()
   updated_config = read_config()
   config_entry = "pants_version"
   if test_pants_version == PantsVersion.unspecified:
     updated_config.remove_option(GLOBAL_SECTION, config_entry)
     # NB: We also remove plugins as they refer to the pants_version.
     updated_config.remove_option(GLOBAL_SECTION, "plugins")
-    write_config(updated_config)
   elif test_pants_version == PantsVersion.config:
     if config_entry not in original_config[GLOBAL_SECTION]:
       raise ValueError("You requested to use the pants_version from pants.ini for this test, but pants.ini "
                        "does not include a pants_version!")
-  try:
+  with temporarily_rewrite_config(updated_config):
+    yield
+
+
+@contextmanager
+def setup_python_version(test_python_version: PythonVersion):
+  """Modify pants.ini to allow the Python version to be unspecified or change to what was requested."""
+  updated_config = read_config()
+  config_entry = "pants_runtime_python_version"
+  if test_python_version == PythonVersion.unspecified:
+    updated_config.remove_option(GLOBAL_SECTION, config_entry)
+  else:
+    updated_config[GLOBAL_SECTION][config_entry] = test_python_version.value
+  with temporarily_rewrite_config(updated_config):
     yield
-  except subprocess.CalledProcessError:
-    raise
-  finally:
-    write_config(original_config)
 
 
 def read_config() -> configparser.ConfigParser:
@@ -85,5 +111,15 @@ def write_config(config: configparser.ConfigParser) -> None:
     config.write(f)
 
 
+@contextmanager
+def temporarily_rewrite_config(updated_config: configparser.ConfigParser) -> None:
+  original_config = read_config()
+  write_config(updated_config)
+  try:
+    yield
+  finally:
+    write_config(original_config)
+
+
 if __name__ == "__main__":
   main()

pants

@@ -13,7 +13,7 @@
 
 set -eou pipefail
 
-PYTHON_BIN_NAME="${PYTHON:-python2.7}"
+PYTHON_BIN_NAME="${PYTHON:-unspecified}"
 
 PANTS_HOME="${PANTS_HOME:-${XDG_CACHE_HOME:-$HOME/.cache}/pants/setup}"
 PANTS_BOOTSTRAP="${PANTS_HOME}/bootstrap-$(uname -s)-$(uname -m)"
@@ -62,8 +62,25 @@ function get_pants_ini_config_value {
   sed -ne "/${prefix}/ s#${prefix}##p" pants.ini
 }
 
+function get_python_major_minor_version {
+  python_exe="$1"
+  "$python_exe" <<EOF
+import sys
+major_minor_version = ''.join(str(version_num) for version_num in sys.version_info[0:2])
+print(major_minor_version)
+EOF
+}
+
+function check_interpreter_on_path {
+  python="$1"
+  if ! which "${python}" >/dev/null; then
+    die "Could not find ${python}. Please ensure ${python} is on your PATH."
+  fi
+}
+
 # The high-level flow:
-# 1.) Resolve the Python interpreter, defaulting to `python2.7`.
+# 1.) Resolve the Python interpreter, first reading from the env var $PYTHON,
+#     then reading from pants.ini, then defaulting to `python2.7`.
 # 2.) Grab pants version from pants.ini or default to latest.
 # 3.) Check for a venv via a naming/path convention and execute if found.
 # 4.) Otherwise create venv and re-exec self.
@@ -72,11 +89,17 @@ function get_pants_ini_config_value {
 # are installed and up to date.
 
 function determine_pants_runtime_python_version {
-  if which "${PYTHON_BIN_NAME}" >/dev/null; then
-    which "${PYTHON_BIN_NAME}"
+  if [[ "${PYTHON_BIN_NAME}" != 'unspecified' ]]; then
+    python_bin_name="${PYTHON_BIN_NAME}"
   else
-    die "Could not find ${PYTHON_BIN_NAME}. Please ensure ${PYTHON_BIN_NAME} is on your PATH."
+    interpreter_version="$(get_pants_ini_config_value 'pants_runtime_python_version')"
+    if [[ -z "${interpreter_version}" ]]; then
+      interpreter_version='2.7'
+    fi
+    python_bin_name="python${interpreter_version}"
   fi
+  check_interpreter_on_path "${python_bin_name}"
+  which "${python_bin_name}"
 }
 
 # TODO(John Sirois): GC race loser tmp dirs leftover from bootstrap_XXX
@@ -107,7 +130,8 @@ function bootstrap_pants {
     pants_version="unspecified"
   fi
 
-  target_folder_name="${pants_version}"
+  python_major_minor_version="$(get_python_major_minor_version "${python}")"
+  target_folder_name="${pants_version}_py${python_major_minor_version}"
 
   if [[ ! -d "${PANTS_BOOTSTRAP}/${target_folder_name}" ]]; then
     (