mirror of
https://github.com/python-poetry/install.python-poetry.org.git
synced 2024-11-28 16:50:59 -05:00
Implement a new bootstrapping method
This commit is contained in:
commit
3921e4bea3
1 changed files with 876 additions and 0 deletions
876
install-poetry.py
Normal file
876
install-poetry.py
Normal file
|
@ -0,0 +1,876 @@
|
||||||
|
"""
|
||||||
|
This script will install Poetry and its dependencies.
|
||||||
|
|
||||||
|
It does, in order:
|
||||||
|
|
||||||
|
- Downloads the virtualenv package to a temporary directory and add it to sys.path.
|
||||||
|
- Creates a virtual environment in the correct OS data dir which will be
|
||||||
|
- `%APPDATA%\\pypoetry` on Windows
|
||||||
|
- ~/Library/Application Support/pypoetry on MacOS
|
||||||
|
- `${XDG_DATA_HOME}/pypoetry` (or `~/.local/share/pypoetry` if it's not set) on UNIX systems
|
||||||
|
- In `${POETRY_HOME}` if it's set.
|
||||||
|
- Installs the latest or given version of Poetry inside this virtual environment.
|
||||||
|
- Installs a `poetry` script in the Python user directory (or `${POETRY_HOME/bin}` if `POETRY_HOME` is set).
|
||||||
|
"""
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
import shutil
|
||||||
|
import site
|
||||||
|
import stat
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
import tempfile
|
||||||
|
|
||||||
|
from contextlib import closing
|
||||||
|
from contextlib import contextmanager
|
||||||
|
from functools import cmp_to_key
|
||||||
|
from io import UnsupportedOperation
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Optional
|
||||||
|
from typing import Tuple
|
||||||
|
from urllib.request import Request
|
||||||
|
from urllib.request import urlopen
|
||||||
|
|
||||||
|
|
||||||
|
SHELL = os.getenv("SHELL", "")
|
||||||
|
WINDOWS = sys.platform.startswith("win") or (sys.platform == "cli" and os.name == "nt")
|
||||||
|
MACOS = sys.platform == "darwin"
|
||||||
|
|
||||||
|
FOREGROUND_COLORS = {
|
||||||
|
"black": 30,
|
||||||
|
"red": 31,
|
||||||
|
"green": 32,
|
||||||
|
"yellow": 33,
|
||||||
|
"blue": 34,
|
||||||
|
"magenta": 35,
|
||||||
|
"cyan": 36,
|
||||||
|
"white": 37,
|
||||||
|
}
|
||||||
|
|
||||||
|
BACKGROUND_COLORS = {
|
||||||
|
"black": 40,
|
||||||
|
"red": 41,
|
||||||
|
"green": 42,
|
||||||
|
"yellow": 43,
|
||||||
|
"blue": 44,
|
||||||
|
"magenta": 45,
|
||||||
|
"cyan": 46,
|
||||||
|
"white": 47,
|
||||||
|
}
|
||||||
|
|
||||||
|
OPTIONS = {"bold": 1, "underscore": 4, "blink": 5, "reverse": 7, "conceal": 8}
|
||||||
|
|
||||||
|
|
||||||
|
def style(fg, bg, options):
|
||||||
|
codes = []
|
||||||
|
|
||||||
|
if fg:
|
||||||
|
codes.append(FOREGROUND_COLORS[fg])
|
||||||
|
|
||||||
|
if bg:
|
||||||
|
codes.append(BACKGROUND_COLORS[bg])
|
||||||
|
|
||||||
|
if options:
|
||||||
|
if not isinstance(options, (list, tuple)):
|
||||||
|
options = [options]
|
||||||
|
|
||||||
|
for option in options:
|
||||||
|
codes.append(OPTIONS[option])
|
||||||
|
|
||||||
|
return "\033[{}m".format(";".join(map(str, codes)))
|
||||||
|
|
||||||
|
|
||||||
|
STYLES = {
|
||||||
|
"info": style("cyan", None, None),
|
||||||
|
"comment": style("yellow", None, None),
|
||||||
|
"success": style("green", None, None),
|
||||||
|
"error": style("red", None, None),
|
||||||
|
"warning": style("yellow", None, None),
|
||||||
|
"b": style(None, None, ("bold",)),
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def is_decorated():
|
||||||
|
if WINDOWS:
|
||||||
|
return (
|
||||||
|
os.getenv("ANSICON") is not None
|
||||||
|
or "ON" == os.getenv("ConEmuANSI")
|
||||||
|
or "xterm" == os.getenv("Term")
|
||||||
|
)
|
||||||
|
|
||||||
|
if not hasattr(sys.stdout, "fileno"):
|
||||||
|
return False
|
||||||
|
|
||||||
|
try:
|
||||||
|
return os.isatty(sys.stdout.fileno())
|
||||||
|
except UnsupportedOperation:
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def is_interactive():
|
||||||
|
if not hasattr(sys.stdin, "fileno"):
|
||||||
|
return False
|
||||||
|
|
||||||
|
try:
|
||||||
|
return os.isatty(sys.stdin.fileno())
|
||||||
|
except UnsupportedOperation:
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def colorize(style, text):
|
||||||
|
if not is_decorated():
|
||||||
|
return text
|
||||||
|
|
||||||
|
return "{}{}\033[0m".format(STYLES[style], text)
|
||||||
|
|
||||||
|
|
||||||
|
def string_to_bool(value):
|
||||||
|
value = value.lower()
|
||||||
|
|
||||||
|
return value in {"true", "1", "y", "yes"}
|
||||||
|
|
||||||
|
|
||||||
|
def data_dir(version: Optional[str] = None) -> Path:
|
||||||
|
if os.getenv("POETRY_HOME"):
|
||||||
|
return Path(os.getenv("POETRY_HOME")).expanduser()
|
||||||
|
|
||||||
|
if WINDOWS:
|
||||||
|
const = "CSIDL_APPDATA"
|
||||||
|
path = os.path.normpath(_get_win_folder(const))
|
||||||
|
path = os.path.join(path, "pypoetry")
|
||||||
|
elif MACOS:
|
||||||
|
path = os.path.expanduser("~/Library/Application Support/pypoetry")
|
||||||
|
if not os.path.isdir(path):
|
||||||
|
path = os.path.expanduser("~/.config/pypoetry")
|
||||||
|
else:
|
||||||
|
path = os.getenv("XDG_DATA_HOME", os.path.expanduser("~/.local/share"))
|
||||||
|
path = os.path.join(path, "pypoetry")
|
||||||
|
|
||||||
|
if version:
|
||||||
|
path = os.path.join(path, version)
|
||||||
|
|
||||||
|
return Path(path)
|
||||||
|
|
||||||
|
|
||||||
|
def bin_dir(version: Optional[str] = None) -> Path:
|
||||||
|
if os.getenv("POETRY_HOME"):
|
||||||
|
return Path(os.getenv("POETRY_HOME"), "bin").expanduser()
|
||||||
|
|
||||||
|
user_base = site.getuserbase()
|
||||||
|
|
||||||
|
if WINDOWS:
|
||||||
|
bin_dir = os.path.join(user_base, "Scripts")
|
||||||
|
else:
|
||||||
|
bin_dir = os.path.join(user_base, "bin")
|
||||||
|
|
||||||
|
return Path(bin_dir)
|
||||||
|
|
||||||
|
|
||||||
|
def _get_win_folder_from_registry(csidl_name):
|
||||||
|
import winreg as _winreg
|
||||||
|
|
||||||
|
shell_folder_name = {
|
||||||
|
"CSIDL_APPDATA": "AppData",
|
||||||
|
"CSIDL_COMMON_APPDATA": "Common AppData",
|
||||||
|
"CSIDL_LOCAL_APPDATA": "Local AppData",
|
||||||
|
}[csidl_name]
|
||||||
|
|
||||||
|
key = _winreg.OpenKey(
|
||||||
|
_winreg.HKEY_CURRENT_USER,
|
||||||
|
r"Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders",
|
||||||
|
)
|
||||||
|
dir, type = _winreg.QueryValueEx(key, shell_folder_name)
|
||||||
|
|
||||||
|
return dir
|
||||||
|
|
||||||
|
|
||||||
|
def _get_win_folder_with_ctypes(csidl_name):
|
||||||
|
import ctypes
|
||||||
|
|
||||||
|
csidl_const = {
|
||||||
|
"CSIDL_APPDATA": 26,
|
||||||
|
"CSIDL_COMMON_APPDATA": 35,
|
||||||
|
"CSIDL_LOCAL_APPDATA": 28,
|
||||||
|
}[csidl_name]
|
||||||
|
|
||||||
|
buf = ctypes.create_unicode_buffer(1024)
|
||||||
|
ctypes.windll.shell32.SHGetFolderPathW(None, csidl_const, None, 0, buf)
|
||||||
|
|
||||||
|
# Downgrade to short path name if have highbit chars. See
|
||||||
|
# <http://bugs.activestate.com/show_bug.cgi?id=85099>.
|
||||||
|
has_high_char = False
|
||||||
|
for c in buf:
|
||||||
|
if ord(c) > 255:
|
||||||
|
has_high_char = True
|
||||||
|
break
|
||||||
|
if has_high_char:
|
||||||
|
buf2 = ctypes.create_unicode_buffer(1024)
|
||||||
|
if ctypes.windll.kernel32.GetShortPathNameW(buf.value, buf2, 1024):
|
||||||
|
buf = buf2
|
||||||
|
|
||||||
|
return buf.value
|
||||||
|
|
||||||
|
|
||||||
|
if WINDOWS:
|
||||||
|
try:
|
||||||
|
from ctypes import windll # noqa
|
||||||
|
|
||||||
|
_get_win_folder = _get_win_folder_with_ctypes
|
||||||
|
except ImportError:
|
||||||
|
_get_win_folder = _get_win_folder_from_registry
|
||||||
|
|
||||||
|
|
||||||
|
@contextmanager
|
||||||
|
def temporary_directory(*args, **kwargs):
|
||||||
|
try:
|
||||||
|
from tempfile import TemporaryDirectory
|
||||||
|
except ImportError:
|
||||||
|
name = tempfile.mkdtemp(*args, **kwargs)
|
||||||
|
|
||||||
|
yield name
|
||||||
|
|
||||||
|
shutil.rmtree(name)
|
||||||
|
else:
|
||||||
|
with TemporaryDirectory(*args, **kwargs) as name:
|
||||||
|
yield name
|
||||||
|
|
||||||
|
|
||||||
|
BIN = """# -*- coding: utf-8 -*-
|
||||||
|
import glob
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
|
||||||
|
site_packages = "{site_packages}"
|
||||||
|
|
||||||
|
sys.path.insert(0, site_packages)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
from {main_module} import main
|
||||||
|
|
||||||
|
main()
|
||||||
|
"""
|
||||||
|
|
||||||
|
BAT = '@echo off\r\n{python_executable} "{poetry_bin}" %*\r\n'
|
||||||
|
|
||||||
|
|
||||||
|
PRE_MESSAGE = """# Welcome to {poetry}!
|
||||||
|
|
||||||
|
This will download and install the latest version of {poetry},
|
||||||
|
a dependency and package manager for Python.
|
||||||
|
|
||||||
|
It will add the `poetry` command to {poetry}'s bin directory, located at:
|
||||||
|
|
||||||
|
{poetry_home_bin}
|
||||||
|
|
||||||
|
You can uninstall at any time by executing this script with the --uninstall option,
|
||||||
|
and these changes will be reverted.
|
||||||
|
"""
|
||||||
|
|
||||||
|
POST_MESSAGE = """{poetry} ({version}) is installed now. Great!
|
||||||
|
|
||||||
|
You can test that everything is set up by executing:
|
||||||
|
|
||||||
|
`{test_command}`
|
||||||
|
"""
|
||||||
|
|
||||||
|
POST_MESSAGE_NOT_IN_PATH = """{poetry} ({version}) is installed now. Great!
|
||||||
|
|
||||||
|
To get started you need {poetry}'s bin directory ({poetry_home_bin}) in your `PATH`
|
||||||
|
environment variable.
|
||||||
|
{configure_message}
|
||||||
|
Alternatively, you can call {poetry} explicitly with `{poetry_executable}`.
|
||||||
|
|
||||||
|
You can test that everything is set up by executing:
|
||||||
|
|
||||||
|
`{test_command}`
|
||||||
|
"""
|
||||||
|
|
||||||
|
POST_MESSAGE_CONFIGURE_UNIX = """
|
||||||
|
Add `export PATH="{poetry_home_bin}:$PATH` to your shell configuration file.
|
||||||
|
"""
|
||||||
|
|
||||||
|
POST_MESSAGE_CONFIGURE_FISH = """
|
||||||
|
You can execute `set -U fish_user_paths {poetry_home_bin} $fish_user_paths`
|
||||||
|
"""
|
||||||
|
|
||||||
|
POST_MESSAGE_CONFIGURE_WINDOWS = """"""
|
||||||
|
|
||||||
|
|
||||||
|
class Cursor:
|
||||||
|
def __init__(self) -> None:
|
||||||
|
self._output = sys.stdout
|
||||||
|
|
||||||
|
def move_up(self, lines: int = 1) -> "Cursor":
|
||||||
|
self._output.write("\x1b[{}A".format(lines))
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def move_down(self, lines: int = 1) -> "Cursor":
|
||||||
|
self._output.write("\x1b[{}B".format(lines))
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def move_right(self, columns: int = 1) -> "Cursor":
|
||||||
|
self._output.write("\x1b[{}C".format(columns))
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def move_left(self, columns: int = 1) -> "Cursor":
|
||||||
|
self._output.write("\x1b[{}D".format(columns))
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def move_to_column(self, column: int) -> "Cursor":
|
||||||
|
self._output.write("\x1b[{}G".format(column))
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def move_to_position(self, column: int, row: int) -> "Cursor":
|
||||||
|
self._output.write("\x1b[{};{}H".format(row + 1, column))
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def save_position(self) -> "Cursor":
|
||||||
|
self._output.write("\x1b7")
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def restore_position(self) -> "Cursor":
|
||||||
|
self._output.write("\x1b8")
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def hide(self) -> "Cursor":
|
||||||
|
self._output.write("\x1b[?25l")
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def show(self) -> "Cursor":
|
||||||
|
self._output.write("\x1b[?25h\x1b[?0c")
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def clear_line(self) -> "Cursor":
|
||||||
|
"""
|
||||||
|
Clears all the output from the current line.
|
||||||
|
"""
|
||||||
|
self._output.write("\x1b[2K")
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def clear_line_after(self) -> "Cursor":
|
||||||
|
"""
|
||||||
|
Clears all the output from the current line after the current position.
|
||||||
|
"""
|
||||||
|
self._output.write("\x1b[K")
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def clear_output(self) -> "Cursor":
|
||||||
|
"""
|
||||||
|
Clears all the output from the cursors' current position
|
||||||
|
to the end of the screen.
|
||||||
|
"""
|
||||||
|
self._output.write("\x1b[0J")
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
def clear_screen(self) -> "Cursor":
|
||||||
|
"""
|
||||||
|
Clears the entire screen.
|
||||||
|
"""
|
||||||
|
self._output.write("\x1b[2J")
|
||||||
|
|
||||||
|
return self
|
||||||
|
|
||||||
|
|
||||||
|
class Installer:
|
||||||
|
METADATA_URL = "https://pypi.org/pypi/poetry/json"
|
||||||
|
VERSION_REGEX = re.compile(
|
||||||
|
r"v?(\d+)(?:\.(\d+))?(?:\.(\d+))?(?:\.(\d+))?"
|
||||||
|
"("
|
||||||
|
"[._-]?"
|
||||||
|
r"(?:(stable|beta|b|rc|RC|alpha|a|patch|pl|p)((?:[.-]?\d+)*)?)?"
|
||||||
|
"([.-]?dev)?"
|
||||||
|
")?"
|
||||||
|
r"(?:\+[^\s]+)?"
|
||||||
|
)
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
version: Optional[str] = None,
|
||||||
|
preview: bool = False,
|
||||||
|
force: bool = False,
|
||||||
|
accept_all: bool = False,
|
||||||
|
git: Optional[str] = None,
|
||||||
|
path: Optional[str] = None,
|
||||||
|
) -> None:
|
||||||
|
self._version = version
|
||||||
|
self._preview = preview
|
||||||
|
self._force = force
|
||||||
|
self._accept_all = accept_all
|
||||||
|
self._git = git
|
||||||
|
self._path = path
|
||||||
|
self._data_dir = data_dir()
|
||||||
|
self._bin_dir = bin_dir()
|
||||||
|
self._cursor = Cursor()
|
||||||
|
|
||||||
|
def allows_prereleases(self) -> bool:
|
||||||
|
return self._preview
|
||||||
|
|
||||||
|
def run(self) -> int:
|
||||||
|
if self._git:
|
||||||
|
version = self._git
|
||||||
|
elif self._path:
|
||||||
|
version = self._path
|
||||||
|
else:
|
||||||
|
version, current_version = self.get_version()
|
||||||
|
|
||||||
|
if version is None:
|
||||||
|
return 0
|
||||||
|
|
||||||
|
self.display_pre_message()
|
||||||
|
self.ensure_directories()
|
||||||
|
|
||||||
|
try:
|
||||||
|
self.install(version)
|
||||||
|
except subprocess.CalledProcessError as e:
|
||||||
|
print(colorize("error", "An error has occured: {}".format(str(e))))
|
||||||
|
print(e.output.decode())
|
||||||
|
|
||||||
|
return e.returncode
|
||||||
|
|
||||||
|
self._write("")
|
||||||
|
self.display_post_message(version)
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
def install(self, version, upgrade=False):
|
||||||
|
"""
|
||||||
|
Installs Poetry in $POETRY_HOME.
|
||||||
|
"""
|
||||||
|
self._write(
|
||||||
|
"Installing {} ({})".format(
|
||||||
|
colorize("info", "Poetry"), colorize("info", version)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
env_path, site_packages = self.make_env(version)
|
||||||
|
self.install_poetry(version, env_path)
|
||||||
|
self.make_bin(site_packages, version)
|
||||||
|
|
||||||
|
self._overwrite(
|
||||||
|
"Installing {} ({}): {}".format(
|
||||||
|
colorize("info", "Poetry"),
|
||||||
|
colorize("b", version),
|
||||||
|
colorize("success", "Done"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
self._data_dir.joinpath("VERSION").write_text(version)
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
def uninstall(self) -> int:
|
||||||
|
if not self._data_dir.exists():
|
||||||
|
self._write(
|
||||||
|
"{} is not currently installed.".format(colorize("info", "Poetry"))
|
||||||
|
)
|
||||||
|
|
||||||
|
return 1
|
||||||
|
|
||||||
|
version = None
|
||||||
|
if self._data_dir.joinpath("VERSION").exists():
|
||||||
|
version = self._data_dir.joinpath("VERSION").read_text().strip()
|
||||||
|
|
||||||
|
if version:
|
||||||
|
self._write(
|
||||||
|
"Removing {} ({})".format(
|
||||||
|
colorize("info", "Poetry"), colorize("b", version)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
self._write("Removing {}".format(colorize("info", "Poetry")))
|
||||||
|
|
||||||
|
shutil.rmtree(str(self._data_dir))
|
||||||
|
for script in ["poetry", "poetry.bat"]:
|
||||||
|
self._bin_dir.joinpath(script).unlink(missing_ok=True)
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
def make_env(self, version: str) -> Tuple[Path, Path]:
|
||||||
|
self._overwrite(
|
||||||
|
"Installing {} ({}): {}".format(
|
||||||
|
colorize("info", "Poetry"),
|
||||||
|
colorize("b", version),
|
||||||
|
colorize("comment", "Creating environment"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
env_path = self._data_dir.joinpath("venv")
|
||||||
|
|
||||||
|
with temporary_directory() as tmp_dir:
|
||||||
|
subprocess.call(
|
||||||
|
[sys.executable, "-m", "pip", "install", "virtualenv", "-t", tmp_dir],
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
stderr=subprocess.STDOUT,
|
||||||
|
)
|
||||||
|
|
||||||
|
sys.path.insert(0, tmp_dir)
|
||||||
|
|
||||||
|
import virtualenv
|
||||||
|
|
||||||
|
virtualenv.cli_run([str(env_path), "--clear"])
|
||||||
|
|
||||||
|
if WINDOWS:
|
||||||
|
return env_path, list(env_path.glob("Lib/site-packages"))[0]
|
||||||
|
|
||||||
|
return env_path, list(env_path.glob("lib/python*/site-packages"))[0]
|
||||||
|
|
||||||
|
def make_bin(self, site_packages: Path, version: str) -> None:
|
||||||
|
self._overwrite(
|
||||||
|
"Installing {} ({}): {}".format(
|
||||||
|
colorize("info", "Poetry"),
|
||||||
|
colorize("b", version),
|
||||||
|
colorize("comment", "Creating script"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
self._bin_dir.mkdir(parents=True, exist_ok=True)
|
||||||
|
|
||||||
|
python_executable = sys.executable
|
||||||
|
|
||||||
|
if WINDOWS:
|
||||||
|
with self._bin_dir.joinpath("poetry.bat").open("w") as f:
|
||||||
|
f.write(
|
||||||
|
BAT.format(
|
||||||
|
python_executable=python_executable,
|
||||||
|
poetry_bin=self._bin_dir.joinpath("poetry"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
# Versions of Poetry prior to 1.2.0 did not have the main()
|
||||||
|
# function at the poetry.console.application level but et he poetry.console one.
|
||||||
|
main_module = "poetry.console.application"
|
||||||
|
version_content = site_packages.joinpath("poetry/__version__.py").read_text(
|
||||||
|
encoding="utf-8"
|
||||||
|
)
|
||||||
|
|
||||||
|
current_version_re = re.match('(?ms).*__version__ = "(.+)".*', version_content)
|
||||||
|
if not current_version_re:
|
||||||
|
self._write(
|
||||||
|
colorize(
|
||||||
|
"warning",
|
||||||
|
"Unable to get the current Poetry version. Assuming None",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
if is_decorated():
|
||||||
|
self._write("")
|
||||||
|
|
||||||
|
current_version = "1.2.0"
|
||||||
|
else:
|
||||||
|
current_version = current_version_re.group(1)
|
||||||
|
|
||||||
|
m = self.VERSION_REGEX.match(current_version)
|
||||||
|
if tuple(int(p) for p in m.groups()[:2]) < (1, 2):
|
||||||
|
main_module = "poetry.console"
|
||||||
|
|
||||||
|
with self._bin_dir.joinpath("poetry").open("w", encoding="utf-8") as f:
|
||||||
|
f.write("#!/usr/bin/env {}\n".format(python_executable))
|
||||||
|
|
||||||
|
if WINDOWS:
|
||||||
|
f.write(
|
||||||
|
BIN.format(
|
||||||
|
site_packages=str(site_packages.resolve()).replace(
|
||||||
|
"\\", "\\\\"
|
||||||
|
),
|
||||||
|
main_module=main_module,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
f.write(
|
||||||
|
BIN.format(
|
||||||
|
site_packages=str(site_packages.resolve()),
|
||||||
|
main_module=main_module,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
if not WINDOWS:
|
||||||
|
# Making the file executable
|
||||||
|
st = os.stat(self._bin_dir.joinpath("poetry").as_posix())
|
||||||
|
os.chmod(
|
||||||
|
self._bin_dir.joinpath("poetry").as_posix(), st.st_mode | stat.S_IEXEC
|
||||||
|
)
|
||||||
|
|
||||||
|
def install_poetry(self, version: str, env_path: Path) -> None:
|
||||||
|
self._overwrite(
|
||||||
|
"Installing {} ({}): {}".format(
|
||||||
|
colorize("info", "Poetry"),
|
||||||
|
colorize("b", version),
|
||||||
|
colorize("comment", "Installing Poetry"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
if WINDOWS:
|
||||||
|
python = env_path.joinpath("Scripts/python.exe")
|
||||||
|
else:
|
||||||
|
python = env_path.joinpath("bin/python")
|
||||||
|
|
||||||
|
if self._git:
|
||||||
|
specification = "git+" + version
|
||||||
|
elif self._path:
|
||||||
|
specification = version
|
||||||
|
else:
|
||||||
|
specification = f"poetry=={version}"
|
||||||
|
|
||||||
|
subprocess.call(
|
||||||
|
[str(python), "-m", "pip", "install", specification],
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
stderr=subprocess.STDOUT,
|
||||||
|
)
|
||||||
|
|
||||||
|
def display_pre_message(self) -> None:
|
||||||
|
kwargs = {
|
||||||
|
"poetry": colorize("info", "Poetry"),
|
||||||
|
"poetry_home_bin": colorize("comment", self._bin_dir),
|
||||||
|
}
|
||||||
|
self._write(PRE_MESSAGE.format(**kwargs))
|
||||||
|
|
||||||
|
def display_post_message(self, version: str) -> None:
|
||||||
|
if WINDOWS:
|
||||||
|
return self.display_post_message_windows(version)
|
||||||
|
|
||||||
|
if SHELL == "fish":
|
||||||
|
return self.display_post_message_fish(version)
|
||||||
|
|
||||||
|
return self.display_post_message_unix(version)
|
||||||
|
|
||||||
|
def display_post_message_windows(self, version: str) -> None:
|
||||||
|
path = self.get_windows_path_var()
|
||||||
|
|
||||||
|
message = POST_MESSAGE_NOT_IN_PATH
|
||||||
|
if path and str(self._bin_dir) in path:
|
||||||
|
message = POST_MESSAGE
|
||||||
|
|
||||||
|
self._write(
|
||||||
|
message.format(
|
||||||
|
poetry=colorize("info", "Poetry"),
|
||||||
|
version=colorize("b", version),
|
||||||
|
poetry_home_bin=colorize("comment", self._bin_dir),
|
||||||
|
poetry_executable=colorize("b", self._bin_dir.joinpath("poetry")),
|
||||||
|
configure_message=POST_MESSAGE_CONFIGURE_WINDOWS.format(
|
||||||
|
poetry_home_bin=colorize("comment", self._bin_dir)
|
||||||
|
),
|
||||||
|
test_command=colorize("b", "poetry --version"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
def get_windows_path_var(self) -> Optional[str]:
|
||||||
|
import winreg
|
||||||
|
|
||||||
|
with winreg.ConnectRegistry(None, winreg.HKEY_CURRENT_USER) as root:
|
||||||
|
with winreg.OpenKey(root, "Environment", 0, winreg.KEY_ALL_ACCESS) as key:
|
||||||
|
path, _ = winreg.QueryValueEx(key, "PATH")
|
||||||
|
|
||||||
|
return path
|
||||||
|
|
||||||
|
def display_post_message_fish(self, version: str) -> None:
|
||||||
|
fish_user_paths = subprocess.check_output(
|
||||||
|
["fish", "-c", "echo $fish_user_paths"]
|
||||||
|
).decode("utf-8")
|
||||||
|
|
||||||
|
message = POST_MESSAGE_NOT_IN_PATH
|
||||||
|
if fish_user_paths and str(self._bin_dir) in fish_user_paths:
|
||||||
|
message = POST_MESSAGE
|
||||||
|
|
||||||
|
self._write(
|
||||||
|
message.format(
|
||||||
|
poetry=colorize("info", "Poetry"),
|
||||||
|
version=colorize("b", version),
|
||||||
|
poetry_home_bin=colorize("comment", self._bin_dir),
|
||||||
|
poetry_executable=colorize("b", self._bin_dir.joinpath("poetry")),
|
||||||
|
configure_message=POST_MESSAGE_CONFIGURE_FISH.format(
|
||||||
|
poetry_home_bin=colorize("comment", self._bin_dir)
|
||||||
|
),
|
||||||
|
test_command=colorize("b", "poetry --version"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
def display_post_message_unix(self, version: str) -> None:
|
||||||
|
paths = os.getenv("PATH", "").split(":")
|
||||||
|
|
||||||
|
message = POST_MESSAGE_NOT_IN_PATH
|
||||||
|
if paths and str(self._bin_dir) in paths:
|
||||||
|
message = POST_MESSAGE
|
||||||
|
|
||||||
|
self._write(
|
||||||
|
message.format(
|
||||||
|
poetry=colorize("info", "Poetry"),
|
||||||
|
version=colorize("b", version),
|
||||||
|
poetry_home_bin=colorize("comment", self._bin_dir),
|
||||||
|
poetry_executable=colorize("b", self._bin_dir.joinpath("poetry")),
|
||||||
|
configure_message=POST_MESSAGE_CONFIGURE_UNIX.format(
|
||||||
|
poetry_home_bin=colorize("comment", self._bin_dir)
|
||||||
|
),
|
||||||
|
test_command=colorize("b", "poetry --version"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
def ensure_directories(self) -> None:
|
||||||
|
self._data_dir.mkdir(parents=True, exist_ok=True)
|
||||||
|
self._bin_dir.mkdir(parents=True, exist_ok=True)
|
||||||
|
|
||||||
|
def get_version(self):
|
||||||
|
current_version = None
|
||||||
|
if self._data_dir.joinpath("VERSION").exists():
|
||||||
|
current_version = self._data_dir.joinpath("VERSION").read_text().strip()
|
||||||
|
|
||||||
|
self._write(colorize("info", "Retrieving Poetry metadata"))
|
||||||
|
|
||||||
|
metadata = json.loads(self._get(self.METADATA_URL).decode())
|
||||||
|
|
||||||
|
def _compare_versions(x, y):
|
||||||
|
mx = self.VERSION_REGEX.match(x)
|
||||||
|
my = self.VERSION_REGEX.match(y)
|
||||||
|
|
||||||
|
vx = tuple(int(p) for p in mx.groups()[:3]) + (mx.group(5),)
|
||||||
|
vy = tuple(int(p) for p in my.groups()[:3]) + (my.group(5),)
|
||||||
|
|
||||||
|
if vx < vy:
|
||||||
|
return -1
|
||||||
|
elif vx > vy:
|
||||||
|
return 1
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
self._write("")
|
||||||
|
releases = sorted(
|
||||||
|
metadata["releases"].keys(), key=cmp_to_key(_compare_versions)
|
||||||
|
)
|
||||||
|
|
||||||
|
if self._version and self._version not in releases:
|
||||||
|
self._write(
|
||||||
|
colorize("error", "Version {} does not exist.".format(self._version))
|
||||||
|
)
|
||||||
|
|
||||||
|
return None, None
|
||||||
|
|
||||||
|
version = self._version
|
||||||
|
if not version:
|
||||||
|
for release in reversed(releases):
|
||||||
|
m = self.VERSION_REGEX.match(release)
|
||||||
|
if m.group(5) and not self.allows_prereleases():
|
||||||
|
continue
|
||||||
|
|
||||||
|
version = release
|
||||||
|
|
||||||
|
break
|
||||||
|
|
||||||
|
if current_version == version and not self._force:
|
||||||
|
self._write(
|
||||||
|
"The latest version ({}) is already installed.".format(
|
||||||
|
colorize("b", version)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
return None, current_version
|
||||||
|
|
||||||
|
return version, current_version
|
||||||
|
|
||||||
|
def _write(self, line) -> None:
|
||||||
|
sys.stdout.write(line + "\n")
|
||||||
|
|
||||||
|
def _overwrite(self, line) -> None:
|
||||||
|
if not is_decorated():
|
||||||
|
return self._write(line)
|
||||||
|
|
||||||
|
self._cursor.move_up()
|
||||||
|
self._cursor.clear_line()
|
||||||
|
self._write(line)
|
||||||
|
|
||||||
|
def _get(self, url):
|
||||||
|
request = Request(url, headers={"User-Agent": "Python Poetry"})
|
||||||
|
|
||||||
|
with closing(urlopen(request)) as r:
|
||||||
|
return r.read()
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser(
|
||||||
|
description="Installs the latest (or given) version of poetry"
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"-p",
|
||||||
|
"--preview",
|
||||||
|
help="install preview version",
|
||||||
|
dest="preview",
|
||||||
|
action="store_true",
|
||||||
|
default=False,
|
||||||
|
)
|
||||||
|
parser.add_argument("--version", help="install named version", dest="version")
|
||||||
|
parser.add_argument(
|
||||||
|
"-f",
|
||||||
|
"--force",
|
||||||
|
help="install on top of existing version",
|
||||||
|
dest="force",
|
||||||
|
action="store_true",
|
||||||
|
default=False,
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"-y",
|
||||||
|
"--yes",
|
||||||
|
help="accept all prompts",
|
||||||
|
dest="accept_all",
|
||||||
|
action="store_true",
|
||||||
|
default=False,
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--uninstall",
|
||||||
|
help="uninstall poetry",
|
||||||
|
dest="uninstall",
|
||||||
|
action="store_true",
|
||||||
|
default=False,
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--path",
|
||||||
|
dest="path",
|
||||||
|
action="store",
|
||||||
|
help=(
|
||||||
|
"Install from a given path (file or directory) instead of "
|
||||||
|
"fetching the latest version of Poetry available online."
|
||||||
|
),
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--git",
|
||||||
|
dest="git",
|
||||||
|
action="store",
|
||||||
|
help=(
|
||||||
|
"Install from a git repository instead of fetching the latest version "
|
||||||
|
"of Poetry available online."
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
installer = Installer(
|
||||||
|
version=args.version or os.getenv("POETRY_VERSION"),
|
||||||
|
preview=args.preview or string_to_bool(os.getenv("POETRY_PREVIEW", "0")),
|
||||||
|
force=args.force,
|
||||||
|
accept_all=args.accept_all
|
||||||
|
or string_to_bool(os.getenv("POETRY_ACCEPT", "0"))
|
||||||
|
or not is_interactive(),
|
||||||
|
path=args.path,
|
||||||
|
git=args.git,
|
||||||
|
)
|
||||||
|
|
||||||
|
if args.uninstall or string_to_bool(os.getenv("POETRY_UNINSTALL", "0")):
|
||||||
|
return installer.uninstall()
|
||||||
|
|
||||||
|
return installer.run()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
sys.exit(main())
|
Loading…
Reference in a new issue