104 lines
2.8 KiB
Python
104 lines
2.8 KiB
Python
import logging
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
import sys
|
|
from typing import Optional
|
|
|
|
from rich.console import Console
|
|
|
|
from safety.console import main_console
|
|
from safety.tool.constants import (
|
|
PYPI_PUBLIC_REPOSITORY_URL,
|
|
PYPI_ORGANIZATION_REPOSITORY_URL,
|
|
PYPI_PROJECT_REPOSITORY_URL,
|
|
)
|
|
from safety.tool.resolver import get_unwrapped_command
|
|
from safety.utils.pyapp_utils import get_path, get_env
|
|
|
|
if sys.version_info >= (3, 11):
|
|
import tomllib
|
|
else:
|
|
import tomli as tomllib
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Poetry:
|
|
@classmethod
|
|
def is_installed(cls) -> bool:
|
|
"""
|
|
Checks if the Poetry program is installed
|
|
|
|
Returns:
|
|
True if Poetry is installed on system, or false otherwise
|
|
"""
|
|
return shutil.which("poetry", path=get_path()) is not None
|
|
|
|
@classmethod
|
|
def is_poetry_project_file(cls, file: Path) -> bool:
|
|
try:
|
|
cfg = tomllib.loads(file.read_text())
|
|
|
|
# First check: tool.poetry section (most definitive)
|
|
if "tool" in cfg and "poetry" in cfg.get("tool", {}):
|
|
return True
|
|
|
|
# Extra check on build-system section
|
|
build_backend = cfg.get("build-system", {}).get("build-backend", "")
|
|
if build_backend and "poetry.core" in build_backend:
|
|
return True
|
|
|
|
return False
|
|
except (IOError, ValueError):
|
|
return False
|
|
|
|
@classmethod
|
|
def configure_pyproject(
|
|
cls,
|
|
file: Path,
|
|
org_slug: Optional[str],
|
|
project_id: Optional[str] = None,
|
|
console: Console = main_console,
|
|
) -> Optional[Path]:
|
|
"""
|
|
Configures index url for specified requirements file.
|
|
|
|
Args:
|
|
file (Path): Path to requirements.txt file.
|
|
org_slug (Optional[str]): Organization slug.
|
|
project_id (Optional[str]): Project ID.
|
|
console (Console): Console instance.
|
|
"""
|
|
if not cls.is_installed():
|
|
logger.error("Poetry is not installed.")
|
|
return None
|
|
|
|
repository_url = (
|
|
PYPI_PROJECT_REPOSITORY_URL.format(org_slug, project_id)
|
|
if project_id and org_slug
|
|
else (
|
|
PYPI_ORGANIZATION_REPOSITORY_URL.format(org_slug)
|
|
if org_slug
|
|
else PYPI_PUBLIC_REPOSITORY_URL
|
|
)
|
|
)
|
|
|
|
result = subprocess.run(
|
|
[
|
|
get_unwrapped_command(name="poetry"),
|
|
"source",
|
|
"add",
|
|
"safety",
|
|
repository_url,
|
|
],
|
|
capture_output=True,
|
|
env=get_env(),
|
|
)
|
|
|
|
if result.returncode != 0:
|
|
logger.error(f"Failed to configure {file} file")
|
|
return None
|
|
|
|
return file
|