Add base vars and sudo check
This commit is contained in:
Benjamin Zimmerman
parent
c151fd6910
commit
054f5ad80c
8733 changed files with 137813 additions and 15 deletions
|
|
@ -0,0 +1 @@
|
|||
/home/runner/.cache/pip/pool/e3/b0/c4/4298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
|
|
@ -0,0 +1 @@
|
|||
/home/runner/.cache/pip/pool/49/7f/25/1bf09512d6075ebaac8cfbd7a1476311aaa51d4e25c6e82b8887527c29
|
||||
|
|
@ -0,0 +1,472 @@
|
|||
from __future__ import unicode_literals
|
||||
|
||||
import hashlib
|
||||
import os
|
||||
import re
|
||||
import shutil
|
||||
import site
|
||||
import stat
|
||||
import subprocess
|
||||
import sys
|
||||
import tarfile
|
||||
|
||||
from functools import cmp_to_key
|
||||
from gzip import GzipFile
|
||||
|
||||
from cleo import argument
|
||||
from cleo import option
|
||||
|
||||
from poetry.core.packages import Dependency
|
||||
from poetry.utils._compat import PY2
|
||||
from poetry.utils._compat import Path
|
||||
|
||||
from ..command import Command
|
||||
|
||||
|
||||
try:
|
||||
from urllib.error import HTTPError
|
||||
from urllib.request import urlopen
|
||||
except ImportError:
|
||||
from urllib2 import HTTPError
|
||||
from urllib2 import urlopen
|
||||
|
||||
|
||||
BIN = """# -*- coding: utf-8 -*-
|
||||
import glob
|
||||
import sys
|
||||
import os
|
||||
|
||||
lib = os.path.normpath(os.path.join(os.path.realpath(__file__), "../..", "lib"))
|
||||
vendors = os.path.join(lib, "poetry", "_vendor")
|
||||
current_vendors = os.path.join(
|
||||
vendors, "py{}".format(".".join(str(v) for v in sys.version_info[:2]))
|
||||
)
|
||||
sys.path.insert(0, lib)
|
||||
sys.path.insert(0, current_vendors)
|
||||
|
||||
if __name__ == "__main__":
|
||||
from poetry.console import main
|
||||
main()
|
||||
"""
|
||||
|
||||
BAT = '@echo off\r\n{python_executable} "{poetry_bin}" %*\r\n'
|
||||
|
||||
|
||||
class SelfUpdateCommand(Command):
|
||||
|
||||
name = "update"
|
||||
description = "Updates Poetry to the latest version."
|
||||
|
||||
arguments = [argument("version", "The version to update to.", optional=True)]
|
||||
options = [option("preview", None, "Install prereleases.")]
|
||||
|
||||
REPOSITORY_URL = "https://github.com/python-poetry/poetry"
|
||||
BASE_URL = REPOSITORY_URL + "/releases/download"
|
||||
|
||||
_data_dir = None
|
||||
_bin_dir = None
|
||||
_pool = None
|
||||
|
||||
@property
|
||||
def home(self):
|
||||
from poetry.utils._compat import Path
|
||||
|
||||
return Path(os.environ.get("POETRY_HOME", "~/.poetry")).expanduser()
|
||||
|
||||
@property
|
||||
def bin(self):
|
||||
return self.home / "bin"
|
||||
|
||||
@property
|
||||
def lib(self):
|
||||
return self.home / "lib"
|
||||
|
||||
@property
|
||||
def lib_backup(self):
|
||||
return self.home / "lib-backup"
|
||||
|
||||
@property
|
||||
def data_dir(self): # type: () -> Path
|
||||
if self._data_dir is not None:
|
||||
return self._data_dir
|
||||
|
||||
from poetry.locations import data_dir
|
||||
|
||||
self._data_dir = data_dir()
|
||||
|
||||
return self._data_dir
|
||||
|
||||
@property
|
||||
def bin_dir(self): # type: () -> Path
|
||||
if self._data_dir is not None:
|
||||
return self._data_dir
|
||||
|
||||
from poetry.utils._compat import WINDOWS
|
||||
|
||||
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")
|
||||
|
||||
self._bin_dir = Path(bin_dir)
|
||||
|
||||
return self._bin_dir
|
||||
|
||||
@property
|
||||
def pool(self):
|
||||
if self._pool is not None:
|
||||
return self._pool
|
||||
|
||||
from poetry.repositories.pool import Pool
|
||||
from poetry.repositories.pypi_repository import PyPiRepository
|
||||
|
||||
pool = Pool()
|
||||
pool.add_repository(PyPiRepository(config=self.poetry.config, fallback=False))
|
||||
|
||||
self._pool = pool
|
||||
|
||||
return self._pool
|
||||
|
||||
def handle(self):
|
||||
from poetry.__version__ import __version__
|
||||
from poetry.core.semver import Version
|
||||
from poetry.utils.env import EnvManager
|
||||
|
||||
new_update_method = False
|
||||
try:
|
||||
self._check_recommended_installation()
|
||||
except RuntimeError as e:
|
||||
env = EnvManager.get_system_env(naive=True)
|
||||
try:
|
||||
env.path.relative_to(self.data_dir)
|
||||
except ValueError:
|
||||
raise e
|
||||
|
||||
new_update_method = True
|
||||
|
||||
version = self.argument("version")
|
||||
if not version:
|
||||
version = ">=" + __version__
|
||||
|
||||
repo = self.pool.repositories[0]
|
||||
packages = repo.find_packages(
|
||||
Dependency("poetry", version, allows_prereleases=self.option("preview"))
|
||||
)
|
||||
if not packages:
|
||||
self.line("No release found for the specified version")
|
||||
return
|
||||
|
||||
packages.sort(
|
||||
key=cmp_to_key(
|
||||
lambda x, y: 0
|
||||
if x.version == y.version
|
||||
else int(x.version < y.version or -1)
|
||||
)
|
||||
)
|
||||
|
||||
release = None
|
||||
for package in packages:
|
||||
if package.is_prerelease():
|
||||
if self.option("preview"):
|
||||
release = package
|
||||
|
||||
break
|
||||
|
||||
continue
|
||||
|
||||
release = package
|
||||
|
||||
break
|
||||
|
||||
if release is None:
|
||||
self.line("No new release found")
|
||||
return
|
||||
|
||||
if release.version == Version.parse(__version__):
|
||||
self.line("You are using the latest version")
|
||||
return
|
||||
|
||||
if new_update_method:
|
||||
return self.update_with_new_method(release.version)
|
||||
|
||||
self.update(release)
|
||||
|
||||
def update(self, release):
|
||||
version = release.version
|
||||
self.line("Updating to <info>{}</info>".format(version))
|
||||
|
||||
if self.lib_backup.exists():
|
||||
shutil.rmtree(str(self.lib_backup))
|
||||
|
||||
# Backup the current installation
|
||||
if self.lib.exists():
|
||||
shutil.copytree(str(self.lib), str(self.lib_backup))
|
||||
shutil.rmtree(str(self.lib))
|
||||
|
||||
try:
|
||||
self._update(version)
|
||||
except Exception:
|
||||
if not self.lib_backup.exists():
|
||||
raise
|
||||
|
||||
shutil.copytree(str(self.lib_backup), str(self.lib))
|
||||
shutil.rmtree(str(self.lib_backup))
|
||||
|
||||
raise
|
||||
finally:
|
||||
if self.lib_backup.exists():
|
||||
shutil.rmtree(str(self.lib_backup))
|
||||
|
||||
self.make_bin()
|
||||
|
||||
self.line("")
|
||||
self.line("")
|
||||
self.line(
|
||||
"<info>Poetry</info> (<comment>{}</comment>) is installed now. Great!".format(
|
||||
version
|
||||
)
|
||||
)
|
||||
|
||||
def update_with_new_method(self, version):
|
||||
self.line("Updating <c1>Poetry</c1> to <c2>{}</c2>".format(version))
|
||||
self.line("")
|
||||
|
||||
self._update_with_new_method(version)
|
||||
self._make_bin()
|
||||
|
||||
self.line("")
|
||||
self.line(
|
||||
"<c1>Poetry</c1> (<c2>{}</c2>) is installed now. Great!".format(version)
|
||||
)
|
||||
|
||||
def _update(self, version):
|
||||
from poetry.utils.helpers import temporary_directory
|
||||
|
||||
release_name = self._get_release_name(version)
|
||||
|
||||
checksum = "{}.sha256sum".format(release_name)
|
||||
|
||||
base_url = self.BASE_URL
|
||||
|
||||
try:
|
||||
r = urlopen(base_url + "/{}/{}".format(version, checksum))
|
||||
except HTTPError as e:
|
||||
if e.code == 404:
|
||||
raise RuntimeError("Could not find {} file".format(checksum))
|
||||
|
||||
raise
|
||||
|
||||
checksum = r.read().decode().strip()
|
||||
|
||||
# We get the payload from the remote host
|
||||
name = "{}.tar.gz".format(release_name)
|
||||
try:
|
||||
r = urlopen(base_url + "/{}/{}".format(version, name))
|
||||
except HTTPError as e:
|
||||
if e.code == 404:
|
||||
raise RuntimeError("Could not find {} file".format(name))
|
||||
|
||||
raise
|
||||
|
||||
meta = r.info()
|
||||
size = int(meta["Content-Length"])
|
||||
current = 0
|
||||
block_size = 8192
|
||||
|
||||
bar = self.progress_bar(max=size)
|
||||
bar.set_format(" - Downloading <info>{}</> <comment>%percent%%</>".format(name))
|
||||
bar.start()
|
||||
|
||||
sha = hashlib.sha256()
|
||||
with temporary_directory(prefix="poetry-updater-") as dir_:
|
||||
tar = os.path.join(dir_, name)
|
||||
with open(tar, "wb") as f:
|
||||
while True:
|
||||
buffer = r.read(block_size)
|
||||
if not buffer:
|
||||
break
|
||||
|
||||
current += len(buffer)
|
||||
f.write(buffer)
|
||||
sha.update(buffer)
|
||||
|
||||
bar.set_progress(current)
|
||||
|
||||
bar.finish()
|
||||
|
||||
# Checking hashes
|
||||
if checksum != sha.hexdigest():
|
||||
raise RuntimeError(
|
||||
"Hashes for {} do not match: {} != {}".format(
|
||||
name, checksum, sha.hexdigest()
|
||||
)
|
||||
)
|
||||
|
||||
gz = GzipFile(tar, mode="rb")
|
||||
try:
|
||||
with tarfile.TarFile(tar, fileobj=gz, format=tarfile.PAX_FORMAT) as f:
|
||||
f.extractall(str(self.lib))
|
||||
finally:
|
||||
gz.close()
|
||||
|
||||
def _update_with_new_method(self, version):
|
||||
from poetry.config.config import Config
|
||||
from poetry.core.packages.dependency import Dependency
|
||||
from poetry.core.packages.project_package import ProjectPackage
|
||||
from poetry.installation.installer import Installer
|
||||
from poetry.packages.locker import NullLocker
|
||||
from poetry.repositories.installed_repository import InstalledRepository
|
||||
from poetry.utils.env import EnvManager
|
||||
|
||||
env = EnvManager.get_system_env(naive=True)
|
||||
installed = InstalledRepository.load(env)
|
||||
|
||||
root = ProjectPackage("poetry-updater", "0.0.0")
|
||||
root.python_versions = ".".join(str(c) for c in env.version_info[:3])
|
||||
root.add_dependency(Dependency("poetry", version.text))
|
||||
|
||||
installer = Installer(
|
||||
self.io,
|
||||
env,
|
||||
root,
|
||||
NullLocker(self.data_dir.joinpath("poetry.lock"), {}),
|
||||
self.pool,
|
||||
Config(),
|
||||
installed=installed,
|
||||
)
|
||||
installer.update(True)
|
||||
installer.run()
|
||||
|
||||
def _make_bin(self):
|
||||
from poetry.utils._compat import WINDOWS
|
||||
|
||||
self.line("")
|
||||
self.line("Updating the <c1>poetry</c1> script")
|
||||
|
||||
self.bin_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
script = "poetry"
|
||||
target_script = "venv/bin/poetry"
|
||||
if WINDOWS:
|
||||
script = "poetry.exe"
|
||||
target_script = "venv/Scripts/poetry.exe"
|
||||
|
||||
if self.bin_dir.joinpath(script).exists():
|
||||
self.bin_dir.joinpath(script).unlink()
|
||||
|
||||
if not PY2 and not WINDOWS:
|
||||
try:
|
||||
self.bin_dir.joinpath(script).symlink_to(
|
||||
self.data_dir.joinpath(target_script)
|
||||
)
|
||||
except OSError:
|
||||
# This can happen if the user
|
||||
# does not have the correct permission on Windows
|
||||
shutil.copy(
|
||||
self.data_dir.joinpath(target_script), self.bin_dir.joinpath(script)
|
||||
)
|
||||
else:
|
||||
shutil.copy(
|
||||
str(self.data_dir.joinpath(target_script)),
|
||||
str(self.bin_dir.joinpath(script)),
|
||||
)
|
||||
|
||||
def process(self, *args):
|
||||
return subprocess.check_output(list(args), stderr=subprocess.STDOUT)
|
||||
|
||||
def _check_recommended_installation(self):
|
||||
from poetry.utils._compat import Path
|
||||
|
||||
current = Path(__file__)
|
||||
try:
|
||||
current.relative_to(self.home)
|
||||
except ValueError:
|
||||
raise RuntimeError(
|
||||
"Poetry was not installed with the recommended installer. "
|
||||
"Cannot update automatically."
|
||||
)
|
||||
|
||||
def _get_release_name(self, version):
|
||||
platform = sys.platform
|
||||
if platform == "linux2":
|
||||
platform = "linux"
|
||||
|
||||
return "poetry-{}-{}".format(version, platform)
|
||||
|
||||
def _bin_path(self, base_path, bin):
|
||||
from poetry.utils._compat import WINDOWS
|
||||
|
||||
if WINDOWS:
|
||||
return (base_path / "Scripts" / bin).with_suffix(".exe")
|
||||
|
||||
return base_path / "bin" / bin
|
||||
|
||||
def make_bin(self):
|
||||
from poetry.utils._compat import WINDOWS
|
||||
|
||||
self.bin.mkdir(0o755, parents=True, exist_ok=True)
|
||||
|
||||
python_executable = self._which_python()
|
||||
|
||||
if WINDOWS:
|
||||
with self.bin.joinpath("poetry.bat").open("w", newline="") as f:
|
||||
f.write(
|
||||
BAT.format(
|
||||
python_executable=python_executable,
|
||||
poetry_bin=str(self.bin / "poetry").replace(
|
||||
os.environ["USERPROFILE"], "%USERPROFILE%"
|
||||
),
|
||||
)
|
||||
)
|
||||
|
||||
bin_content = BIN
|
||||
if not WINDOWS:
|
||||
bin_content = "#!/usr/bin/env {}\n".format(python_executable) + bin_content
|
||||
|
||||
self.bin.joinpath("poetry").write_text(bin_content, encoding="utf-8")
|
||||
|
||||
if not WINDOWS:
|
||||
# Making the file executable
|
||||
st = os.stat(str(self.bin.joinpath("poetry")))
|
||||
os.chmod(str(self.bin.joinpath("poetry")), st.st_mode | stat.S_IEXEC)
|
||||
|
||||
def _which_python(self):
|
||||
"""
|
||||
Decides which python executable we'll embed in the launcher script.
|
||||
"""
|
||||
from poetry.utils._compat import WINDOWS
|
||||
|
||||
allowed_executables = ["python", "python3"]
|
||||
if WINDOWS:
|
||||
allowed_executables += ["py.exe -3", "py.exe -2"]
|
||||
|
||||
# \d in regex ensures we can convert to int later
|
||||
version_matcher = re.compile(r"^Python (?P<major>\d+)\.(?P<minor>\d+)\..+$")
|
||||
fallback = None
|
||||
for executable in allowed_executables:
|
||||
try:
|
||||
raw_version = subprocess.check_output(
|
||||
executable + " --version", stderr=subprocess.STDOUT, shell=True
|
||||
).decode("utf-8")
|
||||
except subprocess.CalledProcessError:
|
||||
continue
|
||||
|
||||
match = version_matcher.match(raw_version.strip())
|
||||
if match and tuple(map(int, match.groups())) >= (3, 0):
|
||||
# favor the first py3 executable we can find.
|
||||
return executable
|
||||
|
||||
if fallback is None:
|
||||
# keep this one as the fallback; it was the first valid executable we found.
|
||||
fallback = executable
|
||||
|
||||
if fallback is None:
|
||||
# Avoid breaking existing scripts
|
||||
fallback = "python"
|
||||
|
||||
return fallback
|
||||
Loading…
Add table
Add a link
Reference in a new issue