Source code for thoth.python.project

#!/usr/bin/env python3
# thoth-python
# Copyright(C) 2018, 2019, 2020 Fridolin Pokorny
# This program is free software: you can redistribute it and / or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY without even the implied warranty of
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <>.

"""Project abstraction and operations on project dependencies."""

import os
import logging
from itertools import chain
import tempfile

from typing import Optional, Dict, Any, List, Tuple

import attr
from thoth.common import cwd
from thoth.common import RuntimeEnvironment
from thoth.analyzer import run_command
from thoth.analyzer import CommandError

from .constraints import Constraints
from .digests_fetcher import DigestsFetcherBase
from .digests_fetcher import PythonDigestsFetcher
from .exceptions import FileLoadError
from .exceptions import InternalError
from .exceptions import NotFoundError
from .exceptions import UnableLockError
from .helpers import parse_requirements
from .package_version import PackageVersion
from .pipfile import Pipfile
from .pipfile import PipfileLock
from .pipfile import PipfileMeta
from .source import Source

_LOGGER = logging.getLogger(__name__)

[docs]@attr.s(slots=True) class Project: """A representation of a Python project.""" pipfile = attr.ib(type=Pipfile) pipfile_lock = attr.ib(type=Optional[PipfileLock]) runtime_environment = attr.ib(type=RuntimeEnvironment, default=attr.Factory(RuntimeEnvironment.from_dict)) constraints = attr.ib(type=Constraints, default=attr.Factory(Constraints)) _graph_db = attr.ib(default=None) _workdir = attr.ib(default=None) @property def workdir(self) -> str: """Access working directory of project.""" if self._workdir: return self._workdir self._workdir = tempfile.mkdtemp() return self._workdir
[docs] @classmethod def from_files( cls, pipfile_path: str, pipfile_lock_path: Optional[str] = None, *, constraints: Optional[Constraints] = None, runtime_environment: Optional[RuntimeEnvironment] = None, without_pipfile_lock: bool = False, ): """Create project from Pipfile and Pipfile.lock files.""" try: pipfile = Pipfile.from_file(pipfile_path) except Exception as exc: raise FileLoadError( "Failed to load Pipfile (path: " f"{os.getcwd() if not pipfile_path else os.path.abspath(pipfile_path)}: {str(exc)}" ) from exc pipfile_lock = None if not without_pipfile_lock: try: pipfile_lock = PipfileLock.from_file(pipfile_lock_path, pipfile=pipfile) except Exception as exc: raise FileLoadError( f"Failed to load Pipfile.lock " f"(path: {os.getcwd() if not pipfile_lock_path else os.path.abspath(pipfile_lock_path)}: {str(exc)}" ) from exc return cls( pipfile, pipfile_lock, runtime_environment=runtime_environment or RuntimeEnvironment.from_dict({}), constraints=constraints or Constraints(), )
[docs] @classmethod def from_dict( cls, pipfile: Dict[str, Any], pipfile_lock: Dict[str, Any], runtime_environment: Optional[RuntimeEnvironment] = None, constraints: Optional[Constraints] = None, ) -> "Project": """Construct project out of a dict representation.""" kwargs: Dict[str, Any] = {} if constraints: kwargs["constraints"] = constraints if runtime_environment: kwargs["runtime_environment"] = runtime_environment pip = Pipfile.from_dict(pipfile) return cls( pipfile=pip, pipfile_lock=PipfileLock.from_dict(pipfile_lock, pipfile=pip), **kwargs, )
[docs] @classmethod def from_strings( cls, pipfile_str: str, pipfile_lock_str: str = None, *, runtime_environment: Optional[RuntimeEnvironment] = None, constraints: Optional[Constraints] = None, ): """Create project from Pipfile and Pipfile.lock loaded into strings.""" pipfile = Pipfile.from_string(pipfile_str) pipfile_lock = None if pipfile_lock_str: pipfile_lock = PipfileLock.from_string(pipfile_lock_str, pipfile) return cls( pipfile, pipfile_lock, runtime_environment=runtime_environment or RuntimeEnvironment.from_dict({}), constraints=constraints or Constraints(), )
[docs] def to_files( self, pipfile_path: str = None, pipfile_lock_path: str = None, *, without_pipfile_lock: bool = False, keep_thoth_section: bool = False, ) -> None: """Write the current state of project into Pipfile and Pipfile.lock files.""" with open(pipfile_path or "Pipfile", "w") as pipfile_file: pipfile_file.write(self.pipfile.to_string(keep_thoth_section=keep_thoth_section)) if not without_pipfile_lock: with open(pipfile_lock_path or "Pipfile.lock", "w") as pipfile_lock_file: pipfile_lock_file.write(self.pipfile_lock.to_string()) # type: ignore
[docs] def construct_requirements_in(self) -> str: """Construct file for the current project.""" return self.pipfile.construct_requirements_in()
[docs] def construct_requirements_txt(self) -> str: """Construct requirements.txt file for the current project - pip-tools compatible.""" if self.pipfile_lock is not None: return self.pipfile_lock.construct_requirements_txt() else: raise TypeError("PipfileLock was not provided.")
[docs] def to_pip_compile_files( self, requirements_path: str = "", requirements_lock_path: str = "requirements.txt", without_lock: bool = False, ) -> None: """Write the current state of project into and requirements.txt files. Files created are compatible with pip/pip-tools. If no lock is specified, requirements.txt hold unpinned direct dependencies as in case of pip. """ if without_lock: with open(requirements_path, "w") as requirements_file: requirements_file.write(self.construct_requirements_in()) return with open(requirements_path, "w") as requirements_file: requirements_file.write(self.construct_requirements_in()) with open(requirements_lock_path, "w") as requirements_lock_file: requirements_lock_file.write(self.construct_requirements_txt())
[docs] @classmethod def from_pip_compile_files( cls, requirements_path: str = "", requirements_lock_path: Optional[str] = None, allow_without_lock: bool = False, runtime_environment: Optional[RuntimeEnvironment] = None, constraints: Optional[Constraints] = None, ) -> "Project": """Parse project from files compatible with pip/pip-tools.""" constraints = constraints or Constraints() sources_lock, package_versions_lock = None, None if allow_without_lock: if not os.path.exists(requirements_path): requirements_lock_path = requirements_lock_path or "requirements.txt" sources, package_versions = parse_requirements(requirements_lock_path) else: sources, package_versions = parse_requirements(requirements_path) if requirements_lock_path is not None: sources_lock, package_versions_lock = parse_requirements(requirements_lock_path) elif os.path.exists("requirements.txt"): sources_lock, package_versions_lock = parse_requirements("requirements.txt") else: requirements_lock_path = requirements_lock_path or "requirements.txt" sources, package_versions = parse_requirements(requirements_path) sources_lock, package_versions_lock = parse_requirements(requirements_lock_path) for package_version in package_versions_lock or []: if not package_version.is_locked(): raise ValueError( f"Package {} in lockfile does not " f"use locked version: {package_version.version!r}" ) if not package_version.hashes: _LOGGER.warning( "Parsed package %r in version %r from lockfile %r has no hashes assigned",, package_version.version, requirements_lock_path, ) sources = sources or sources_lock or [] # We know which index is assigned. if len(sources) == 1: for package_version in package_versions: package_version.index = sources[0] for package_version in package_versions_lock or []: package_version.index = sources[0] # XXX: do we want a default source? meta = PipfileMeta(sources={ s for s in sources}) pipfile = Pipfile.from_package_versions(package_versions, meta=meta) pipfile_lock = None if package_versions_lock: pipfile_lock = PipfileLock.from_package_versions(pipfile=pipfile, packages=package_versions_lock, meta=meta) if runtime_environment: return cls(pipfile, pipfile_lock, runtime_environment=runtime_environment, constraints=constraints) return cls(pipfile, pipfile_lock, constraints=constraints)
[docs] @classmethod def from_package_versions( cls, packages: List[PackageVersion], packages_locked: List[PackageVersion] = None, meta: Optional[PipfileMeta] = None, *, runtime_environment: Optional[RuntimeEnvironment] = None, constraints: Optional[Constraints] = None, ): """Create project from PackageVersion objects. If locked packages are omitted, the lock has to be explicitly performed to generate in-memory Pipfile.lock representation. """ constraints = constraints or Constraints() pipfile = Pipfile.from_package_versions(packages, meta=meta) pipfile_lock = None if packages_locked: pipfile_lock = PipfileLock.from_package_versions(pipfile, packages_locked, meta=pipfile.meta) if runtime_environment: instance = cls(pipfile, pipfile_lock, runtime_environment=runtime_environment, constraints=constraints) else: # Let default be expanded. instance = cls(pipfile, pipfile_lock, constraints=constraints) instance.sanitize_source_indexes() return instance
[docs] def set_allow_prereleases(self, allow_prereleases: bool = True) -> None: """Allow or disallow pre-releases for this project.""" self.pipfile.meta.pipenv["allow_prereleases"] = allow_prereleases # type: ignore
@property def prereleases_allowed(self) -> bool: """Check if pre-releases are allowed for this project.""" return self.pipfile.meta.pipenv.get("allow_prereleases", False) if self.pipfile.meta.pipenv else False
[docs] def set_python_version(self, python_version: str = None) -> None: """Set version of Python used in the project.""" if python_version is None: self.pipfile.meta.requires.pop("python_version") else: if self.pipfile.meta.requires is None: self.pipfile.meta.requires = {} self.pipfile.meta.requires["python_version"] = python_version
@property def python_version(self) -> Optional[str]: """Get Python version used in this project.""" if not self.pipfile.meta.requires: return None return self.pipfile.meta.requires.get("python_version", None)
[docs] def to_dict(self, *, keep_thoth_section: bool = False): """Create a dictionary representation of this project.""" return { "requirements": self.pipfile.to_dict(keep_thoth_section=keep_thoth_section), "requirements_locked": self.pipfile_lock.to_dict() if self.pipfile_lock else None, "runtime_environment": self.runtime_environment.to_dict(), "constraints": self.constraints.to_dict(), }
[docs] def get_configuration_check_report(self) -> Optional[Tuple[Optional[dict], List[dict]]]: """Get a report on project configuration for the given runtime environment.""" result = [] changes_in_config = False # We check Python version if there is used Pipfile, it should match runtime configuration. pipfile_python_version = self.pipfile.meta.requires.get("python_version") runtime_python_version = self.runtime_environment.python_version # Keep the current runtime environment untouched. recommended_runtime_configuration_entry = RuntimeEnvironment.from_dict(self.runtime_environment.to_dict()) if not pipfile_python_version and not runtime_python_version: result.append( { "type": "WARNING", "justification": "Please specify Python version in Pipfile using `pipenv --python <VERSION>` " "and in Thoth's configuration file to have reproducible deployment and " "recommendations targeting specific Python version", } ) elif pipfile_python_version and not runtime_python_version: changes_in_config = True recommended_runtime_configuration_entry.python_version = pipfile_python_version result.append( { "type": "WARNING", "justification": "Use Python version in Thoth's configuration file to have " "recommendations matching configuration in Pipfile, configured Python version " f"in Pipfile is {pipfile_python_version}", } ) # if pipfile_python_version and runtime_python_version: # This case is not related to adjustments in Thoth's configuration but rather in Pipfile - that is handled # in scoring. if not result: return None return recommended_runtime_configuration_entry.to_dict(without_none=True) if changes_in_config else None, result
[docs] def get_outdated_package_versions(self, devel: bool = True) -> dict: """Get outdated packages in the lock file. This function will check indexes configured and look for version of each package. If the given package package is not latest, it will add it to the resulting list with the newer version identifier found on package index. """ if not self.pipfile_lock: raise InternalError("Cannot check outdated packages on not-locked application stack") result = {} for package_version in self.iter_dependencies_locked(with_devel=devel): if package_version.index: latest = package_version.index.get_latest_package_version( if package_version.semantic_version != latest: result[] = (package_version, latest) else: found = False for index in self.pipfile_lock.meta.sources.values(): try: latest = index.get_latest_package_version( found = True except NotFoundError: continue if package_version.semantic_version != latest: result[] = (package_version, latest) if not found: raise NotFoundError( f"Package {package_version!r} was not found on any package index" f"configured: {self.pipfile_lock.meta.to_dict()}" ) return result
[docs] def pipenv_lock(self): """Perform pipenv lock on the current state of project.""" with cwd(self.workdir): self.pipfile.to_file() _LOGGER.debug("Running pipenv lock") try: result = run_command("pipenv lock", env={"PIPENV_IGNORE_VIRTUALENVS": "1"}) except CommandError as exc: _LOGGER.exception( "Unable to lock application stack (return code: %d):\n%s\n", exc.return_code, exc.stdout, exc.stderr ) raise UnableLockError("Failed to perform lock") from exc _LOGGER.debug("pipenv stdout:\n%s", result.stdout) _LOGGER.debug("pipenv stderr:\n%s", result.stderr) self.pipfile_lock = PipfileLock.from_file(pipfile=self.pipfile)
[docs] def is_direct_dependency(self, package_version: PackageVersion) -> bool: """Check whether the given package is a direct dependency.""" return in self.pipfile.packages
[docs] def get_locked_package_version(self, package_name: str) -> Optional[PackageVersion]: """Get locked version of the package.""" if self.pipfile_lock is None: raise ValueError("PipfileLock was not provided and is None.") package_version = self.pipfile_lock.dev_packages.get(package_name) if not package_version: package_version = self.pipfile_lock.packages.get(package_name) return package_version
[docs] def get_package_version(self, package_name: str) -> Optional[PackageVersion]: """Get locked version of the package.""" package_version = self.pipfile.dev_packages.get(package_name) if not package_version: package_version = self.pipfile.packages.get(package_name) return package_version
[docs] def exclude_package(self, package_version: PackageVersion) -> None: """Exclude the given package from application stack.""" if not package_version.is_locked(): raise InternalError("Cannot exclude package not pinned down to specific version: %r", package_version) section = self.pipfile.dev_packages if package_version.develop else self.pipfile.packages to_exclude_package = section.get( if to_exclude_package: package_version.negate_version() _LOGGER.debug(f"Excluding package {to_exclude_package} with package specified {package_version}") if package_version.index != to_exclude_package.index: _LOGGER.warning( f"Excluding package {to_exclude_package} with {package_version} but package has " f"different index configuration" ) if package_version.markers != to_exclude_package: _LOGGER.warning( f"Excluding package {to_exclude_package} with {package_version} but package has " f"different markers configured" ) # We do operations on the package we passed via parameters so the original package is # not adjusted if referenced elsewhere. if to_exclude_package.version != "*": package_version.version = f"{package_version.version},{to_exclude_package.version}" _LOGGER.debug("Package with excluded version %r", package_version) section[] = package_version else: # Adding a new requirement. package_version.negate_version() section[] = package_version _LOGGER.debug("Added excluded package to pipfile configuration: %r", package_version) self.pipenv_lock()
[docs] def iter_dependencies_locked(self, with_devel: bool = True): """Iterate through locked dependencies of this project.""" if not self.pipfile_lock: raise InternalError("Unable to chain locked dependencies - no Pipfile.lock provided") if with_devel: yield from chain(self.pipfile_lock.packages, self.pipfile_lock.dev_packages) else: yield from self.pipfile_lock.packages
[docs] def iter_dependencies(self, with_devel: bool = True): """Iterate through direct dependencies of this project (not locked dependencies).""" if with_devel: yield from chain(self.pipfile.packages, self.pipfile.dev_packages) else: yield from self.pipfile.packages
def _check_sources(self, whitelisted_sources: Optional[List]) -> List: """Check sources configuration in the Pipfile and report back any issues.""" report = [] for source in self.pipfile.meta.sources.values(): if whitelisted_sources and source.url not in whitelisted_sources: report.append( { "type": "ERROR", "id": "SOURCE-NOT-WHITELISTED", "justification": f"Provided index {!r} is not stated in the " f"whitelisted package sources listing", "source": source.to_dict(), } ) elif not source.verify_ssl: report.append( { "type": "WARNING", "id": "INSECURE-SOURCE", "justification": f"Source {!r} does not use SSL/TLS verification", "source": source.to_dict(), } ) return report def _index_scan(self, digests_fetcher: DigestsFetcherBase) -> Tuple[list, dict]: """Generate full report for packages given the sources configured.""" report = {} findings = [] if self.pipfile_lock is None: raise ValueError("PipfileLock was not provided and has value None.") for package_version in chain(self.pipfile_lock.packages, self.pipfile_lock.dev_packages): if in report: # TODO: can we have the same package in dev packages and packages? _LOGGER.warning(f"Package {} already present in the report") continue index_report = digests_fetcher.fetch_digests(, package_version.locked_version) findings.extend(self._check_scan(package_version, index_report)) report[] = index_report return findings, report def _check_scan(self, package_version: PackageVersion, index_report: dict) -> list: """Find and report errors found in the index scan.""" scan_report = [] # Prepare hashes for inspection. hashes = [] pipenv_style_hashes = [] for entry in index_report.values(): hashes.append([item["sha256"] for item in entry]) pipenv_style_hashes.append([f"sha256:{item['sha256']}" for item in entry]) if not package_version.index and len(index_report.keys()) > 1: # Are there indexes with different artifacts present? - suggest to specify index explicitly. if set(chain(*hashes)) != set(hashes[0]): sources_reported = ", ".join(index_report.keys()) scan_report.append( { "type": "WARNING", "id": "DIFFERENT-ARTIFACTS-ON-SOURCES", "justification": f"Configured sources ({sources_reported}) have different artifacts " f"available, specify explicitly source to be used", "indexes": list(index_report.keys()), "package_locked": package_version.to_pipfile_lock(), "package_name":, "package_version": package_version.version, "sources": index_report, } ) # Source for reports. source = None if package_version.index: source = package_version.index.to_dict() # type: ignore if package_version.index and index_report.get(package_version.index.url) and len(hashes) > 1: # Is installed from different source - which one? used_package_version_hashes = set(h[len("sha256:") :] for h in package_version.hashes) # type: ignore configured_index_hashes = set(h["sha256"] for h in index_report[package_version.index.url]) # Find other sources from which artifacts can be installed. other_sources = {} # type: Dict[str, List[Tuple]] for artifact_hash in package_version.hashes: artifact_hash = artifact_hash[ # type: ignore len("sha256:") : ] # Remove pipenv-specific hash formatting. # type: ignore for index_url, index_info in index_report.items(): if index_url == package_version.index.url: # Skip index that is assigned to the package, we are inspecting from the other sources. continue artifact_entry = [(i.get("name"), i["sha256"]) for i in index_info if i["sha256"] == artifact_hash] if not artifact_entry: continue if index_url not in other_sources: other_sources[index_url] = [] other_sources[index_url].extend(artifact_entry) if not set(used_package_version_hashes).issubset(set(configured_index_hashes)): # Source is different from the configured one. scan_report.append( { "type": "ERROR", "id": "ARTIFACT-DIFFERENT-SOURCE", "justification": f"Artifacts are installed from different " f'sources ({",".join(other_sources.keys())}) not respecting configuration', "source": source, # type: ignore "package_locked": package_version.to_pipfile_lock(), "package_name":, "package_version": package_version.version, "indexes": list(other_sources.keys()), "sources": other_sources, } ) elif other_sources: # Warn about possible installation from another source. scan_report.append( { "type": "INFO", "id": "ARTIFACT-POSSIBLE-DIFFERENT-SOURCE", "justification": f"Artifacts can be installed from different sources " f'({",".join(other_sources.keys())}) not respecting configuration ' f"that expects {!r}", "source": source, # type: ignore "package_locked": package_version.to_pipfile_lock(), "package_name":, "package_version": package_version.version, "indexes": list(other_sources.keys()), "sources": other_sources, } ) if package_version.index and not index_report.get(package_version.index.url): # Configured index does not provide the given package. scan_report.append( { "type": "ERROR", "id": "MISSING-PACKAGE", "justification": f"Source index {!r} explicitly " f"assigned to package {!r} but there was not found " f"any record for the given package", "source": source, # type: ignore "package_locked": package_version.to_pipfile_lock(), "package_name":, "package_version": package_version.version, "sources": index_report, } ) # Changed hashes? for digest in package_version.hashes: digest = digest[len("sha256:") :] # type: ignore for index_info in index_report.values(): if any(item["sha256"] == digest for item in index_info): break else: scan_report.append( { "type": "ERROR", "id": "INVALID-ARTIFACT-HASH", "justification": "Hash for the given artifact was not found", "source": source, # type: ignore "package_locked": package_version.to_pipfile_lock(), "package_name":, "package_version": package_version.version, "digest": digest, # type: ignore } ) return scan_report
[docs] def check_provenance(self, whitelisted_sources: list = None, digests_fetcher: DigestsFetcherBase = None) -> List: """Check provenance/origin of packages that are stated in the project.""" digests_fetcher = digests_fetcher or PythonDigestsFetcher(list(self.pipfile.meta.sources.values())) findings, _ = self._index_scan(digests_fetcher) findings.extend(self._check_sources(whitelisted_sources)) return findings
[docs] def sanitize_source_indexes(self): """Make sure all the indexes are correctly propagated to Pipfile and Pipfile.lock metadata.""" self.pipfile.sanitize_source_indexes() if self.pipfile_lock: self.pipfile_lock.sanitize_source_indexes()
[docs] def add_source( self, url: str, verify_ssl: bool = True, name: str = None, warehouse: bool = False, warehouse_api_url: str = None, ) -> Source: """Add a package source (index) to the project.""" if name: source = Source( name=name, url=url, verify_ssl=verify_ssl, warehouse=warehouse, warehouse_api_url=warehouse_api_url ) else: # Do not provide source index name so that attrs correctly compute default based on URL. source = Source(url=url, verify_ssl=verify_ssl, warehouse=warehouse, warehouse_api_url=warehouse_api_url) self.pipfile.meta.add_source(source) if self.pipfile_lock: self.pipfile_lock.meta.add_source(source) return source
[docs] def add_package( self, package_name: str, package_version: str = None, *, source: Source = None, develop: bool = False ): """Add the given package to project. This method will add packages only to Pipfile, locking has to be explicitly done once package is added. """ if source and not in self.pipfile.meta.sources: raise InternalError( f"Adding package {package_name} to project without having source index " f"{} registered in the project" ) pkg_vers = PackageVersion( name=package_name, version=package_version, develop=develop, index=source if source else None ) self.pipfile.add_package_version(pkg_vers)