updates

Backend/venv/lib/python3.12/site-packages/safety/tool/poetry/main.py

@@ -0,0 +1,103 @@
+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