Skip to content

P
python-poetry
Commit def178f6 by Zechy Wong Committed by GitHub
Options

installer: support offline instalation using --file option

parent f45ec453
Showing with 60 additions and 24 deletions
get-poetry.py
...@@ -334,6 +334,7 @@ class Installer: ...@@ -334,6 +334,7 @@ class Installer:
preview=False, preview=False,
force=False, force=False,
accept_all=False, accept_all=False,
file=None,
base_url=BASE_URL, base_url=BASE_URL,
): ):
self._version = version self._version = version
...@@ -341,6 +342,7 @@ class Installer: ...@@ -341,6 +342,7 @@ class Installer:
self._force = force self._force = force
self._modify_path = True self._modify_path = True
self._accept_all = accept_all self._accept_all = accept_all
self._offline_file = file
self._base_url = base_url self._base_url = base_url
def allows_prereleases(self): def allows_prereleases(self):
...@@ -357,7 +359,9 @@ class Installer: ...@@ -357,7 +359,9 @@ class Installer:
self.ensure_home() self.ensure_home()
try: try:
self.install(version, upgrade=current_version is not None) self.install(
version, upgrade=current_version is not None, file=self._offline_file
)
except subprocess.CalledProcessError as e: except subprocess.CalledProcessError as e:
print(colorize("error", "An error has occured: {}".format(str(e)))) print(colorize("error", "An error has occured: {}".format(str(e))))
print(e.output.decode()) print(e.output.decode())
...@@ -378,6 +382,34 @@ class Installer: ...@@ -378,6 +382,34 @@ class Installer:
self.remove_from_path() self.remove_from_path()
def get_version(self): def get_version(self):
current_version = None
if os.path.exists(POETRY_LIB):
with open(
os.path.join(POETRY_LIB, "poetry", "__version__.py"), encoding="utf-8"
) as f:
version_content = f.read()
current_version_re = re.match(
'(?ms).*__version__ = "(.+)".*', version_content
)
if not current_version_re:
print(
colorize(
"warning",
"Unable to get the current Poetry version. Assuming None",
)
)
else:
current_version = current_version_re.group(1)
# Skip retrieving online release versions if install file is specified
if self._offline_file is not None:
if current_version is not None and not self._force:
print("There is a version of Poetry already installed.")
return None, current_version
return "from an offline file", current_version
print(colorize("info", "Retrieving Poetry metadata")) print(colorize("info", "Retrieving Poetry metadata"))
metadata = json.loads(self._get(self.METADATA_URL).decode()) metadata = json.loads(self._get(self.METADATA_URL).decode())
...@@ -417,26 +449,6 @@ class Installer: ...@@ -417,26 +449,6 @@ class Installer:
break break
current_version = None
if os.path.exists(POETRY_LIB):
with open(
os.path.join(POETRY_LIB, "poetry", "__version__.py"), encoding="utf-8"
) as f:
version_content = f.read()
current_version_re = re.match(
'(?ms).*__version__ = "(.+)".*', version_content
)
if not current_version_re:
print(
colorize(
"warning",
"Unable to get the current Poetry version. Assuming None",
)
)
else:
current_version = current_version_re.group(1)
if current_version == version and not self._force: if current_version == version and not self._force:
print("Latest version already installed.") print("Latest version already installed.")
return None, current_version return None, current_version
...@@ -484,10 +496,13 @@ class Installer: ...@@ -484,10 +496,13 @@ class Installer:
shutil.rmtree(POETRY_HOME) shutil.rmtree(POETRY_HOME)
def install(self, version, upgrade=False): def install(self, version, upgrade=False, file=None):
""" """
Installs Poetry in $POETRY_HOME. Installs Poetry in $POETRY_HOME.
""" """
if file is not None:
print("Attempting to install from file: " + colorize("info", file))
else:
print("Installing version: " + colorize("info", version)) print("Installing version: " + colorize("info", version))
self.make_lib(version) self.make_lib(version)
...@@ -524,6 +539,14 @@ class Installer: ...@@ -524,6 +539,14 @@ class Installer:
shutil.rmtree(POETRY_LIB_BACKUP) shutil.rmtree(POETRY_LIB_BACKUP)
def _make_lib(self, version): def _make_lib(self, version):
# Check if an offline installer file has been specified
if self._offline_file is not None:
try:
self.extract_lib(self._offline_file)
return
except Exception:
raise RuntimeError("Could not install from offline file.")
# We get the payload from the remote host # We get the payload from the remote host
platform = sys.platform platform = sys.platform
if platform == "linux2": if platform == "linux2":
...@@ -583,9 +606,12 @@ class Installer: ...@@ -583,9 +606,12 @@ class Installer:
) )
) )
gz = GzipFile(tar, mode="rb") self.extract_lib(tar)
def extract_lib(self, filename):
gz = GzipFile(filename, mode="rb")
try: try:
with tarfile.TarFile(tar, fileobj=gz, format=tarfile.PAX_FORMAT) as f: with tarfile.TarFile(filename, fileobj=gz, format=tarfile.PAX_FORMAT) as f:
f.extractall(POETRY_LIB) f.extractall(POETRY_LIB)
finally: finally:
gz.close() gz.close()
...@@ -931,10 +957,19 @@ def main(): ...@@ -931,10 +957,19 @@ def main():
parser.add_argument( parser.add_argument(
"--uninstall", dest="uninstall", action="store_true", default=False "--uninstall", dest="uninstall", action="store_true", default=False
) )
parser.add_argument(
"--file",
dest="file",
action="store",
help="Install from a local file instead of fetching the latest version "
"of Poetry available online.",
)
args = parser.parse_args() args = parser.parse_args()
base_url = Installer.BASE_URL base_url = Installer.BASE_URL
if args.file is None:
try: try:
urlopen(Installer.REPOSITORY_URL) urlopen(Installer.REPOSITORY_URL)
except HTTPError as e: except HTTPError as e:
...@@ -950,6 +985,7 @@ def main(): ...@@ -950,6 +985,7 @@ def main():
accept_all=args.accept_all accept_all=args.accept_all
or string_to_bool(os.getenv("POETRY_ACCEPT", "0")) or string_to_bool(os.getenv("POETRY_ACCEPT", "0"))
or not is_interactive(), or not is_interactive(),
file=args.file,
base_url=base_url, base_url=base_url,
) )
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment