mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-24 21:04:40 +01:00
896 lines
31 KiB
Python
896 lines
31 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Bootstrap Pype repositories."""
|
|
import functools
|
|
import logging as log
|
|
import os
|
|
import re
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
from pathlib import Path
|
|
from typing import Union, Callable, List, Tuple
|
|
from zipfile import ZipFile, BadZipFile
|
|
|
|
from appdirs import user_data_dir
|
|
from speedcopy import copyfile
|
|
|
|
from pype.lib import PypeSettingsRegistry
|
|
from pype.version import __version__
|
|
from .tools import load_environments
|
|
|
|
|
|
@functools.total_ordering
|
|
class PypeVersion:
|
|
"""Class for storing information about Pype version.
|
|
|
|
Attributes:
|
|
major (int): [1].2.3-variant-client
|
|
minor (int): 1.[2].3-variant-client
|
|
subversion (int): 1.2.[3]-variant-client
|
|
variant (str): 1.2.3-[variant]-client
|
|
client (str): 1.2.3-variant-[client]
|
|
path (str): path to Pype
|
|
|
|
"""
|
|
major = 0
|
|
minor = 0
|
|
subversion = 0
|
|
variant = "production"
|
|
client = None
|
|
path = None
|
|
|
|
_version_regex = re.compile(
|
|
r"(?P<major>\d+)\.(?P<minor>\d+)\.(?P<sub>\d+)(-?((?P<variant>staging)|(?P<client>.+))(-(?P<cli>.+))?)?") # noqa: E501
|
|
|
|
@property
|
|
def version(self):
|
|
"""return formatted version string."""
|
|
return self._compose_version()
|
|
|
|
@version.setter
|
|
def version(self, val):
|
|
decomposed = self._decompose_version(val)
|
|
self.major = decomposed[0]
|
|
self.minor = decomposed[1]
|
|
self.subversion = decomposed[2]
|
|
self.variant = decomposed[3]
|
|
self.client = decomposed[4]
|
|
|
|
def __init__(self, major: int = None, minor: int = None,
|
|
subversion: int = None, version: str = None,
|
|
variant: str = "production", client: str = None,
|
|
path: Path = None):
|
|
self.path = path
|
|
|
|
if major is None or minor is None or subversion is None:
|
|
if version is None:
|
|
raise ValueError("Need version specified in some way.")
|
|
if version:
|
|
values = self._decompose_version(version)
|
|
self.major = values[0]
|
|
self.minor = values[1]
|
|
self.subversion = values[2]
|
|
self.variant = values[3]
|
|
self.client = values[4]
|
|
else:
|
|
self.major = major
|
|
self.minor = minor
|
|
self.subversion = subversion
|
|
# variant is set only if it is "staging", otherwise "production" is
|
|
# implied and no need to mention it in version string.
|
|
if variant == "staging":
|
|
self.variant = variant
|
|
self.client = client
|
|
|
|
def _compose_version(self):
|
|
version = "{}.{}.{}".format(self.major, self.minor, self.subversion)
|
|
if self.variant == "staging":
|
|
version = "{}-{}".format(version, self.variant)
|
|
|
|
if self.client:
|
|
version = "{}-{}".format(version, self.client)
|
|
|
|
return version
|
|
|
|
@classmethod
|
|
def _decompose_version(cls, version_string: str) -> tuple:
|
|
m = re.search(cls._version_regex, version_string)
|
|
if not m:
|
|
raise ValueError(
|
|
"Cannot parse version string: {}".format(version_string))
|
|
|
|
variant = None
|
|
if m.group("variant") == "staging":
|
|
variant = "staging"
|
|
|
|
client = m.group("client") or m.group("cli")
|
|
|
|
return (int(m.group("major")), int(m.group("minor")),
|
|
int(m.group("sub")), variant, client)
|
|
|
|
def __eq__(self, other):
|
|
if not isinstance(other, self.__class__):
|
|
return False
|
|
return self.version == other.version
|
|
|
|
def __str__(self):
|
|
return self.version
|
|
|
|
def __repr__(self):
|
|
return "{}, {}: {}".format(
|
|
self.__class__.__name__, self.version, self.path)
|
|
|
|
def __hash__(self):
|
|
return hash(self.version)
|
|
|
|
def __lt__(self, other):
|
|
if self.major < other.major:
|
|
return True
|
|
|
|
if self.major <= other.major and self.minor < other.minor:
|
|
return True
|
|
if self.major <= other.major and self.minor <= other.minor and \
|
|
self.subversion < other.subversion:
|
|
return True
|
|
|
|
if self.major == other.major and self.minor == other.minor and \
|
|
self.subversion == other.subversion and \
|
|
self.variant == "staging":
|
|
return True
|
|
|
|
return False
|
|
|
|
def is_staging(self) -> bool:
|
|
"""Test if current version is staging one."""
|
|
return self.variant == "staging"
|
|
|
|
def get_main_version(self) -> str:
|
|
"""Return main version component.
|
|
|
|
This returns x.x.x part of version from possibly more complex one
|
|
like x.x.x-foo-bar.
|
|
|
|
Returns:
|
|
str: main version component
|
|
|
|
"""
|
|
return "{}.{}.{}".format(self.major, self.minor, self.subversion)
|
|
|
|
@staticmethod
|
|
def version_in_str(string: str) -> Tuple:
|
|
"""Find Pype version in given string.
|
|
|
|
Args:
|
|
string (str): string to search.
|
|
|
|
Returns:
|
|
tuple: True/False and PypeVersion if found.
|
|
|
|
"""
|
|
try:
|
|
result = PypeVersion._decompose_version(string)
|
|
except ValueError:
|
|
return False, None
|
|
return True, PypeVersion(major=result[0],
|
|
minor=result[1],
|
|
subversion=result[2],
|
|
variant=result[3],
|
|
client=result[4])
|
|
|
|
|
|
class BootstrapRepos:
|
|
"""Class for bootstrapping local Pype installation.
|
|
|
|
Attributes:
|
|
data_dir (Path): local Pype installation directory.
|
|
live_repo_dir (Path): path to repos directory if running live,
|
|
otherwise `None`.
|
|
registry (PypeSettingsRegistry): Pype registry object.
|
|
zip_filter (list): List of files to exclude from zip
|
|
pype_filter (list): list of top level directories not to include in
|
|
zip in Pype repository.
|
|
|
|
"""
|
|
|
|
def __init__(self, progress_callback: Callable = None, message=None):
|
|
"""Constructor.
|
|
|
|
Args:
|
|
progress_callback (callable): Optional callback method to report
|
|
progress.
|
|
message (QtCore.Signal, optional): Signal to report messages back.
|
|
|
|
"""
|
|
# vendor and app used to construct user data dir
|
|
self._vendor = "pypeclub"
|
|
self._app = "pype"
|
|
self._log = log.getLogger(str(__class__))
|
|
self.data_dir = Path(user_data_dir(self._app, self._vendor))
|
|
self.registry = PypeSettingsRegistry()
|
|
self.zip_filter = [".pyc", "__pycache__"]
|
|
self.pype_filter = [
|
|
"build", "docs", "tests", "repos", "tools", "venv"
|
|
]
|
|
self._message = message
|
|
|
|
# dummy progress reporter
|
|
def empty_progress(x: int):
|
|
return x
|
|
|
|
if not progress_callback:
|
|
progress_callback = empty_progress
|
|
self._progress_callback = progress_callback
|
|
|
|
if getattr(sys, "frozen", False):
|
|
self.live_repo_dir = Path(sys.executable).parent / "repos"
|
|
else:
|
|
self.live_repo_dir = Path(Path(__file__).parent / ".." / "repos")
|
|
|
|
@staticmethod
|
|
def get_version_path_from_list(version: str, version_list: list) -> Path:
|
|
"""Get path for specific version in list of Pype versions.
|
|
|
|
Args:
|
|
version (str): Version string to look for (1.2.4-staging)
|
|
version_list (list of PypeVersion): list of version to search.
|
|
|
|
Returns:
|
|
Path: Path to given version.
|
|
|
|
"""
|
|
for v in version_list:
|
|
if str(v) == version:
|
|
return v.path
|
|
|
|
@staticmethod
|
|
def get_local_live_version() -> str:
|
|
"""Get version of local Pype."""
|
|
return __version__
|
|
|
|
@staticmethod
|
|
def get_version(repo_dir: Path) -> Union[str, None]:
|
|
"""Get version of Pype in given directory.
|
|
|
|
Args:
|
|
repo_dir (Path): Path to Pype repo.
|
|
|
|
Returns:
|
|
str: version string.
|
|
None: if Pype is not found.
|
|
|
|
"""
|
|
# try to find version
|
|
version_file = Path(repo_dir) / "pype" / "version.py"
|
|
if not version_file.exists():
|
|
return None
|
|
|
|
version = {}
|
|
with version_file.open("r") as fp:
|
|
exec(fp.read(), version)
|
|
|
|
return version['__version__']
|
|
|
|
def install_live_repos(self, repo_dir: Path = None) -> Union[Path, None]:
|
|
"""Copy zip created from Pype repositories to user data dir.
|
|
|
|
This detect Pype version either in local "live" Pype repository
|
|
or in user provided path. Then it will zip it in temporary directory
|
|
and finally it will move it to destination which is user data
|
|
directory. Existing files will be replaced.
|
|
|
|
Args:
|
|
repo_dir (Path, optional): Path to Pype repository.
|
|
|
|
Returns:
|
|
Path: path of installed repository file.
|
|
|
|
"""
|
|
# if repo dir is not set, we detect local "live" Pype repository
|
|
# version and use it as a source. Otherwise repo_dir is user
|
|
# entered location.
|
|
if not repo_dir:
|
|
version = self.get_local_live_version()
|
|
repo_dir = self.live_repo_dir
|
|
else:
|
|
version = self.get_version(repo_dir)
|
|
|
|
# create destination directory
|
|
if not self.data_dir.exists():
|
|
self.data_dir.mkdir(parents=True)
|
|
|
|
# create zip inside temporary directory.
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
temp_zip = \
|
|
Path(temp_dir) / f"pype-v{version}.zip"
|
|
self._log.info(f"creating zip: {temp_zip}")
|
|
|
|
self._create_pype_zip(temp_zip, repo_dir)
|
|
if not os.path.exists(temp_zip):
|
|
self._log.error("make archive failed.")
|
|
return None
|
|
|
|
destination = self.data_dir / temp_zip.name
|
|
|
|
if destination.exists():
|
|
self._log.warning(
|
|
f"Destination file {destination} exists, removing.")
|
|
try:
|
|
destination.unlink()
|
|
except Exception as e:
|
|
self._log.error(e)
|
|
return None
|
|
try:
|
|
shutil.move(temp_zip.as_posix(), self.data_dir.as_posix())
|
|
except shutil.Error as e:
|
|
self._log.error(e)
|
|
return None
|
|
return destination
|
|
|
|
def _create_pype_zip(
|
|
self,
|
|
zip_path: Path, include_dir: Path,
|
|
include_pype: bool = True) -> None:
|
|
"""Pack repositories and Pype into zip.
|
|
|
|
We are using :mod:`zipfile` instead :meth:`shutil.make_archive`
|
|
because we need to decide what file and directories to include in zip
|
|
and what not. They are determined by :attr:`zip_filter` on file level
|
|
and :attr:`pype_filter` on top level directory in Pype repository.
|
|
|
|
Args:
|
|
zip_path (str): path to zip file.
|
|
include_dir (Path): repo directories to include.
|
|
include_pype (bool): add Pype module itself.
|
|
|
|
"""
|
|
include_dir = include_dir.resolve()
|
|
|
|
def _filter_dir(path: Path, path_filter: List) -> List[Path]:
|
|
"""Recursively crawl over path and filter."""
|
|
result = []
|
|
for item in path.iterdir():
|
|
if item.name in path_filter:
|
|
continue
|
|
if item.name.startswith('.'):
|
|
continue
|
|
if item.is_dir():
|
|
result.extend(_filter_dir(item, path_filter))
|
|
else:
|
|
result.append(item)
|
|
return result
|
|
|
|
pype_list = []
|
|
# get filtered list of files in repositories (repos directory)
|
|
repo_list = _filter_dir(include_dir, self.zip_filter)
|
|
# count them
|
|
repo_files = len(repo_list)
|
|
|
|
# there must be some files, otherwise `include_dir` path is wrong
|
|
assert repo_files != 0, f"No repositories to include in {include_dir}"
|
|
pype_inc = 0
|
|
if include_pype:
|
|
# get filtered list of file in Pype repository
|
|
pype_list = _filter_dir(include_dir.parent, self.zip_filter)
|
|
pype_files = len(pype_list)
|
|
repo_inc = 48.0 / float(repo_files)
|
|
pype_inc = 48.0 / float(pype_files)
|
|
else:
|
|
repo_inc = 98.0 / float(repo_files)
|
|
progress = 0
|
|
|
|
with ZipFile(zip_path, "w") as zip_file:
|
|
file: Path
|
|
for file in repo_list:
|
|
progress += repo_inc
|
|
self._progress_callback(int(progress))
|
|
|
|
# archive name is relative to repos dir
|
|
arc_name = file.relative_to(include_dir)
|
|
zip_file.write(file, arc_name)
|
|
|
|
# add pype itself
|
|
if include_pype:
|
|
pype_root = include_dir.parent.resolve()
|
|
# generate list of filtered paths
|
|
dir_filter = [pype_root / f for f in self.pype_filter]
|
|
|
|
file: Path
|
|
for file in pype_list:
|
|
progress += pype_inc
|
|
self._progress_callback(int(progress))
|
|
|
|
# if file resides in filtered path, skip it
|
|
is_inside = None
|
|
df: Path
|
|
for df in dir_filter:
|
|
try:
|
|
is_inside = file.resolve().relative_to(df)
|
|
except ValueError:
|
|
pass
|
|
|
|
if is_inside:
|
|
continue
|
|
|
|
processed_path = file
|
|
self._log.debug(f"processing {processed_path}")
|
|
self._print(f"- processing {processed_path}", False)
|
|
|
|
zip_file.write(file,
|
|
"pype" / file.relative_to(pype_root))
|
|
|
|
# test if zip is ok
|
|
zip_file.testzip()
|
|
self._progress_callback(100)
|
|
|
|
@staticmethod
|
|
def add_paths_from_archive(archive: Path) -> None:
|
|
"""Add first-level directories as paths to :mod:`sys.path`.
|
|
|
|
This will enable Python to import modules is second-level directories
|
|
in zip file.
|
|
|
|
Adding to both `sys.path` and `PYTHONPATH`, skipping duplicates.
|
|
|
|
Args:
|
|
archive (Path): path to archive.
|
|
|
|
"""
|
|
with ZipFile(archive, "r") as zip_file:
|
|
name_list = zip_file.namelist()
|
|
|
|
roots = []
|
|
for item in name_list:
|
|
root = item.split("/")[0]
|
|
if root not in roots:
|
|
roots.append(root)
|
|
sys.path.insert(0, f"{archive}{os.path.sep}{root}")
|
|
|
|
pythonpath = os.getenv("PYTHONPATH", "")
|
|
paths = pythonpath.split(os.pathsep)
|
|
paths += roots
|
|
|
|
os.environ["PYTHONPATH"] = os.pathsep.join(paths)
|
|
|
|
@staticmethod
|
|
def add_paths_from_directory(directory: Path) -> None:
|
|
"""Add first level directories as paths to :mod:`sys.path`.
|
|
|
|
This works the same as :meth:`add_paths_from_archive` but in
|
|
specified directory.
|
|
|
|
Adding to both `sys.path` and `PYTHONPATH`, skipping duplicates.
|
|
|
|
Args:
|
|
directory (Path): path to directory.
|
|
|
|
"""
|
|
roots = []
|
|
for item in directory.iterdir():
|
|
if item.is_dir():
|
|
root = item.as_posix()
|
|
if root not in roots:
|
|
roots.append(root)
|
|
sys.path.insert(0, root)
|
|
|
|
pythonpath = os.getenv("PYTHONPATH", "")
|
|
paths = pythonpath.split(os.pathsep)
|
|
paths += roots
|
|
|
|
os.environ["PYTHONPATH"] = os.pathsep.join(paths)
|
|
|
|
def find_pype(
|
|
self,
|
|
pype_path: Path = None,
|
|
include_zips: bool = False) -> Union[List[PypeVersion], None]:
|
|
"""Get ordered dict of detected Pype version.
|
|
|
|
Resolution order for Pype is following:
|
|
|
|
1) First we test for ``PYPE_PATH`` environment variable
|
|
2) We try to find ``pypePath`` in registry setting
|
|
3) We use user data directory
|
|
|
|
Args:
|
|
pype_path (Path, optional): Try to find Pype on the given path.
|
|
include_zips (bool, optional): If set True it will try to find
|
|
Pype in zip files in given directory.
|
|
|
|
Returns:
|
|
dict of Path: Dictionary of detected Pype version.
|
|
Key is version, value is path to zip file.
|
|
|
|
None: if Pype is not found.
|
|
|
|
"""
|
|
dir_to_search = self.data_dir
|
|
|
|
# if we have pype_path specified, search only there.
|
|
if pype_path:
|
|
dir_to_search = pype_path
|
|
else:
|
|
if os.getenv("PYPE_PATH"):
|
|
if Path(os.getenv("PYPE_PATH")).exists():
|
|
dir_to_search = Path(os.getenv("PYPE_PATH"))
|
|
else:
|
|
try:
|
|
registry_dir = Path(
|
|
str(self.registry.get_item("pypePath")))
|
|
if registry_dir.exists():
|
|
dir_to_search = registry_dir
|
|
|
|
except ValueError:
|
|
# nothing found in registry, we'll use data dir
|
|
pass
|
|
|
|
# pype installation dir doesn't exists
|
|
if not dir_to_search.exists():
|
|
return None
|
|
|
|
_pype_versions = []
|
|
# iterate over directory in first level and find all that might
|
|
# contain Pype.
|
|
for file in dir_to_search.iterdir():
|
|
|
|
# if file, strip extension, in case of dir not.
|
|
name = file.name if file.is_dir() else file.stem
|
|
result = PypeVersion.version_in_str(name)
|
|
|
|
if result[0]:
|
|
detected_version = result[1]
|
|
|
|
if file.is_dir():
|
|
# if item is directory that might (based on it's name)
|
|
# contain Pype version, check if it really does contain
|
|
# Pype and that their versions matches.
|
|
try:
|
|
# add one 'pype' level as inside dir there should
|
|
# be many other repositories.
|
|
version_str = BootstrapRepos.get_version(
|
|
file / "pype")
|
|
version_check = PypeVersion(version=version_str)
|
|
except ValueError:
|
|
self._log.error(
|
|
f"cannot determine version from {file}")
|
|
continue
|
|
|
|
version_main = version_check.get_main_version()
|
|
detected_main = detected_version.get_main_version()
|
|
if version_main != detected_main:
|
|
self._log.error(
|
|
(f"dir version ({detected_version}) and "
|
|
f"its content version ({version_check}) "
|
|
"doesn't match. Skipping."))
|
|
continue
|
|
|
|
if file.is_file():
|
|
|
|
if not include_zips:
|
|
continue
|
|
|
|
# skip non-zip files
|
|
if file.suffix.lower() != ".zip":
|
|
continue
|
|
|
|
# open zip file, look inside and parse version from Pype
|
|
# inside it. If there is none, or it is different from
|
|
# version specified in file name, skip it.
|
|
try:
|
|
with ZipFile(file, "r") as zip_file:
|
|
with zip_file.open(
|
|
"pype/pype/version.py") as version_file:
|
|
zip_version = {}
|
|
exec(version_file.read(), zip_version)
|
|
version_check = PypeVersion(
|
|
version=zip_version["__version__"])
|
|
|
|
version_main = version_check.get_main_version()
|
|
detected_main = detected_version.get_main_version()
|
|
|
|
if version_main != detected_main:
|
|
self._log.error(
|
|
(f"zip version ({detected_version}) "
|
|
f"and its content version "
|
|
f"({version_check}) "
|
|
"doesn't match. Skipping."))
|
|
continue
|
|
except BadZipFile:
|
|
self._log.error(f"{file} is not zip file")
|
|
continue
|
|
except KeyError:
|
|
self._log.error("Zip not containing Pype")
|
|
continue
|
|
|
|
detected_version.path = file
|
|
_pype_versions.append(detected_version)
|
|
|
|
return sorted(_pype_versions)
|
|
|
|
@staticmethod
|
|
def _get_pype_from_mongo(mongo_url: str) -> Union[Path, None]:
|
|
"""Get path from Mongo database.
|
|
|
|
This sets environment variable ``PYPE_MONGO`` for
|
|
:mod:`pype.settings` to be able to read data from database.
|
|
It will then retrieve environment variables and among them
|
|
must be ``PYPE_PATH``.
|
|
|
|
Args:
|
|
mongo_url (str): mongodb connection url
|
|
|
|
Returns:
|
|
Path: if path from ``PYPE_PATH`` is found.
|
|
None: if not.
|
|
|
|
"""
|
|
os.environ["PYPE_MONGO"] = mongo_url
|
|
env = load_environments()
|
|
if not env.get("PYPE_PATH"):
|
|
return None
|
|
return Path(env.get("PYPE_PATH"))
|
|
|
|
def process_entered_location(self, location: str) -> Union[Path, None]:
|
|
"""Process user entered location string.
|
|
|
|
It decides if location string is mongodb url or path.
|
|
If it is mongodb url, it will connect and load ``PYPE_PATH`` from
|
|
there and use it as path to Pype. In it is _not_ mongodb url, it
|
|
is assumed we have a path, this is tested and zip file is
|
|
produced and installed using :meth:`install_live_repos`.
|
|
|
|
Args:
|
|
location (str): User entered location.
|
|
|
|
Returns:
|
|
Path: to Pype zip produced from this location.
|
|
None: Zipping failed.
|
|
|
|
"""
|
|
pype_path = None
|
|
# try to get pype path from mongo.
|
|
if location.startswith("mongodb"):
|
|
pype_path = self._get_pype_from_mongo(location)
|
|
if not pype_path:
|
|
self._log.error("cannot find PYPE_PATH in settings.")
|
|
return None
|
|
|
|
# if not successful, consider location to be fs path.
|
|
if not pype_path:
|
|
pype_path = Path(location)
|
|
|
|
# test if this path does exist.
|
|
if not pype_path.exists():
|
|
self._log.error(f"{pype_path} doesn't exists.")
|
|
return None
|
|
|
|
# test if entered path isn't user data dir
|
|
if self.data_dir == pype_path:
|
|
self._log.error("cannot point to user data dir")
|
|
return None
|
|
|
|
# find pype zip files in location. There can be
|
|
# either "live" Pype repository, or multiple zip files or even
|
|
# multiple pype version directories. This process looks into zip
|
|
# files and directories and tries to parse `version.py` file.
|
|
versions = self.find_pype(pype_path)
|
|
if versions:
|
|
self._log.info(f"found Pype in [ {pype_path} ]")
|
|
self._log.info(f"latest version found is [ {versions[-1]} ]")
|
|
|
|
destination = self.data_dir / versions[-1].path.name
|
|
|
|
# test if destination file already exist, if so lets delete it.
|
|
# we consider path on location as authoritative place.
|
|
if destination.exists():
|
|
try:
|
|
destination.unlink()
|
|
except OSError:
|
|
self._log.error(
|
|
f"cannot remove already existing {destination}",
|
|
exc_info=True)
|
|
return None
|
|
|
|
# create destination parent directories even if they don't exist.
|
|
if not destination.parent.exists():
|
|
destination.parent.mkdir(parents=True)
|
|
|
|
# latest version found is directory
|
|
if versions[-1].path.is_dir():
|
|
# zip it, copy it and extract it
|
|
# create zip inside temporary directory.
|
|
self._log.info("Creating zip from directory ...")
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
temp_zip = \
|
|
Path(temp_dir) / f"pype-v{versions[-1]}.zip"
|
|
self._log.info(f"creating zip: {temp_zip}")
|
|
|
|
self._create_pype_zip(temp_zip, versions[-1].path)
|
|
if not os.path.exists(temp_zip):
|
|
self._log.error("make archive failed.")
|
|
return None
|
|
|
|
destination = self.data_dir / temp_zip.name
|
|
|
|
elif versions[-1].path.is_file():
|
|
# in this place, it must be zip file as `find_pype()` is
|
|
# checking just that.
|
|
assert versions[-1].path.suffix.lower() == ".zip", (
|
|
"Invalid file format"
|
|
)
|
|
try:
|
|
self._log.info("Copying zip to destination ...")
|
|
copyfile(versions[-1].path.as_posix(), destination.as_posix())
|
|
except OSError:
|
|
self._log.error(
|
|
"cannot copy detected version to user data directory",
|
|
exc_info=True)
|
|
return None
|
|
|
|
# extract zip there
|
|
self._log.info("extracting zip to destination ...")
|
|
with ZipFile(versions[-1].path, "r") as zip_ref:
|
|
zip_ref.extractall(destination)
|
|
|
|
return destination
|
|
|
|
# if we got here, it means that location is "live" Pype repository.
|
|
# we'll create zip from it and move it to user data dir.
|
|
repo_file = self.install_live_repos(pype_path)
|
|
if not repo_file.exists():
|
|
self._log.error(f"installing zip {repo_file} failed.")
|
|
return None
|
|
|
|
destination = self.data_dir / repo_file.stem
|
|
if destination.exists():
|
|
try:
|
|
destination.unlink()
|
|
except OSError:
|
|
self._log.error(
|
|
f"cannot remove already existing {destination}",
|
|
exc_info=True)
|
|
return None
|
|
|
|
destination.mkdir(parents=True)
|
|
|
|
# extract zip there
|
|
self._log.info("extracting zip to destination ...")
|
|
with ZipFile(versions[-1].path, "r") as zip_ref:
|
|
zip_ref.extractall(destination)
|
|
|
|
return destination
|
|
|
|
def _print(self, message, error=False):
|
|
if self._message:
|
|
self._message.emit(message, error)
|
|
|
|
def extract_pype(self, version: PypeVersion):
|
|
if not version.path:
|
|
raise ValueError(
|
|
f"version {version} is not associated with any file")
|
|
|
|
destination = self.data_dir / version.path.stem
|
|
if destination.exists():
|
|
try:
|
|
destination.unlink()
|
|
except OSError as e:
|
|
msg = f"!!! Cannot remove already existing {destination}"
|
|
self._log.error(msg)
|
|
self._log.error(e.strerror)
|
|
self._print(msg, True)
|
|
self._print(e.strerror, True)
|
|
return
|
|
|
|
destination.mkdir(parents=True)
|
|
|
|
# extract zip there
|
|
self._print("Extracting zip to destination ...")
|
|
with ZipFile(version.path, "r") as zip_ref:
|
|
zip_ref.extractall(destination)
|
|
|
|
self._print(f"Installed as {version.path.stem}")
|
|
|
|
def install_version(self, pype_version: PypeVersion, force: bool = False):
|
|
"""Install Pype version to user data directory.
|
|
|
|
Args:
|
|
pype_version (PypeVersion): Pype version to install.
|
|
force (bool, optional): Force overwrite existing version.
|
|
|
|
Returns:
|
|
Path: Path to installed Pype.
|
|
|
|
Raises:
|
|
PypeVersionExists: If not forced and this version already exist
|
|
in user data directory.
|
|
PypeVersionInvalid: If version to install is invalid.
|
|
PypeVersionIOError: If copying or zipping fail.
|
|
|
|
"""
|
|
|
|
# test if version is located (in user data dir)
|
|
is_inside = False
|
|
try:
|
|
is_inside = pype_version.path.resolve().relative_to(
|
|
self.data_dir)
|
|
except ValueError:
|
|
# if relative path cannot be calculated, Pype version is not
|
|
# inside user data dir
|
|
pass
|
|
|
|
if is_inside:
|
|
raise PypeVersionExists("Pype already inside user data dir")
|
|
|
|
# determine destination directory name
|
|
# for zip file strip suffix
|
|
destination = self.data_dir / pype_version.path.stem
|
|
|
|
# test if destination file already exist, if so lets delete it.
|
|
# we consider path on location as authoritative place.
|
|
if destination.exists() and force:
|
|
try:
|
|
destination.unlink()
|
|
except OSError:
|
|
self._log.error(
|
|
f"cannot remove already existing {destination}",
|
|
exc_info=True)
|
|
return None
|
|
else:
|
|
raise PypeVersionExists(f"{destination} already exist.")
|
|
|
|
# create destination parent directories even if they don't exist.
|
|
if not destination.exists():
|
|
destination.mkdir(parents=True)
|
|
|
|
# version is directory
|
|
if pype_version.path.is_dir():
|
|
# create zip inside temporary directory.
|
|
self._log.info("Creating zip from directory ...")
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
temp_zip = \
|
|
Path(temp_dir) / f"pype-v{pype_version}.zip"
|
|
self._log.info(f"creating zip: {temp_zip}")
|
|
|
|
self._create_pype_zip(temp_zip, pype_version.path)
|
|
if not os.path.exists(temp_zip):
|
|
self._log.error("make archive failed.")
|
|
raise PypeVersionIOError("Zip creation failed.")
|
|
|
|
# set zip as version source
|
|
pype_version.path = temp_zip
|
|
|
|
elif pype_version.path.is_file():
|
|
# check if file is zip (by extension)
|
|
if pype_version.path.suffix.lower() != ".zip":
|
|
raise PypeVersionInvalid("Invalid file format")
|
|
|
|
try:
|
|
# copy file to destination
|
|
self._log.info("Copying zip to destination ...")
|
|
copyfile(pype_version.path.as_posix(), destination.as_posix())
|
|
except OSError as e:
|
|
self._log.error(
|
|
"cannot copy version to user data directory",
|
|
exc_info=True)
|
|
raise PypeVersionIOError(
|
|
"can't copy version to destination") from e
|
|
|
|
# extract zip there
|
|
self._log.info("extracting zip to destination ...")
|
|
with ZipFile(pype_version.path, "r") as zip:
|
|
zip.extractall(destination)
|
|
|
|
return destination
|
|
|
|
|
|
class PypeVersionExists(Exception):
|
|
"""Exception for handling existing Pype version."""
|
|
pass
|
|
|
|
|
|
class PypeVersionInvalid(Exception):
|
|
"""Exception for handling invalid Pype version."""
|
|
pass
|
|
|
|
|
|
class PypeVersionIOError(Exception):
|
|
"""Exception for handling IO errors in Pype version."""
|
|
pass
|