1
0
Fork 0
mirror of https://github.com/python-poetry/install.python-poetry.org.git synced 2024-11-08 23:13:38 -05:00

Use a symlink to the venv-installed poetry script

This commit is contained in:
Sébastien Eustace 2021-02-19 17:29:43 +01:00 committed by Arun Babu Neelicattu
parent 3921e4bea3
commit a9ee3a4327

View file

@ -19,7 +19,6 @@ import os
import re import re
import shutil import shutil
import site import site
import stat
import subprocess import subprocess
import sys import sys
import tempfile import tempfile
@ -30,7 +29,6 @@ from functools import cmp_to_key
from io import UnsupportedOperation from io import UnsupportedOperation
from pathlib import Path from pathlib import Path
from typing import Optional from typing import Optional
from typing import Tuple
from urllib.request import Request from urllib.request import Request
from urllib.request import urlopen from urllib.request import urlopen
@ -238,24 +236,6 @@ def temporary_directory(*args, **kwargs):
yield 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}! PRE_MESSAGE = """# Welcome to {poetry}!
This will download and install the latest version of {poetry}, This will download and install the latest version of {poetry},
@ -458,9 +438,9 @@ class Installer:
) )
) )
env_path, site_packages = self.make_env(version) env_path = self.make_env(version)
self.install_poetry(version, env_path) self.install_poetry(version, env_path)
self.make_bin(site_packages, version) self.make_bin(version)
self._overwrite( self._overwrite(
"Installing {} ({}): {}".format( "Installing {} ({}): {}".format(
@ -497,11 +477,12 @@ class Installer:
shutil.rmtree(str(self._data_dir)) shutil.rmtree(str(self._data_dir))
for script in ["poetry", "poetry.bat"]: for script in ["poetry", "poetry.bat"]:
self._bin_dir.joinpath(script).unlink(missing_ok=True) if self._bin_dir.joinpath(script).exists():
self._bin_dir.joinpath(script).unlink()
return 0 return 0
def make_env(self, version: str) -> Tuple[Path, Path]: def make_env(self, version: str) -> Path:
self._overwrite( self._overwrite(
"Installing {} ({}): {}".format( "Installing {} ({}): {}".format(
colorize("info", "Poetry"), colorize("info", "Poetry"),
@ -525,12 +506,9 @@ class Installer:
virtualenv.cli_run([str(env_path), "--clear"]) virtualenv.cli_run([str(env_path), "--clear"])
if WINDOWS: return env_path
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, version: str) -> None:
def make_bin(self, site_packages: Path, version: str) -> None:
self._overwrite( self._overwrite(
"Installing {} ({}): {}".format( "Installing {} ({}): {}".format(
colorize("info", "Poetry"), colorize("info", "Poetry"),
@ -541,68 +519,17 @@ class Installer:
self._bin_dir.mkdir(parents=True, exist_ok=True) self._bin_dir.mkdir(parents=True, exist_ok=True)
python_executable = sys.executable script = "poetry"
target_script = "venv/bin/poetry"
if WINDOWS: if WINDOWS:
with self._bin_dir.joinpath("poetry.bat").open("w") as f: script = "poetry.exe"
f.write( target_script = "venv/Scripts/poetry.exe"
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() if self._bin_dir.joinpath(script).exists():
# function at the poetry.console.application level but et he poetry.console one. self._bin_dir.joinpath(script).unlink()
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) self._bin_dir.joinpath(script).symlink_to(
if not current_version_re: self._data_dir.joinpath(target_script)
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: def install_poetry(self, version: str, env_path: Path) -> None: