Add venv site-packages support to package_copier module

Author: PyCompiler ARK++

Core/package_copier.py

@@ -27,6 +27,7 @@
 - Verify package integrity
 - Handle .dist-info and .egg-info metadata
 - Support for namespace packages
+- Auto-detect and copy to venv site-packages
 """
 
 import os
@@ -503,6 +504,96 @@ def get_statistics(self) -> Dict:
             "total_size_mb": round(total_size / (1024 * 1024), 2),
         }
 
+    @staticmethod
+    def find_venv_site_packages(venv_path: Optional[str] = None) -> Optional[str]:
+        """
+        Find the site-packages directory in a virtual environment.
+
+        Args:
+            venv_path: Path to virtual environment. If None, uses current venv.
+
+        Returns:
+            Path to site-packages directory or None.
+        """
+        if venv_path is None:
+            # Try to detect current venv
+            venv_path = os.environ.get("VIRTUAL_ENV")
+            if not venv_path:
+                logger.warning("No virtual environment detected")
+                return None
+
+        # Common site-packages locations
+        possible_paths = [
+            os.path.join(venv_path, "lib", "python3.9", "site-packages"),
+            os.path.join(venv_path, "lib", "python3.10", "site-packages"),
+            os.path.join(venv_path, "lib", "python3.11", "site-packages"),
+            os.path.join(venv_path, "lib", "python3.12", "site-packages"),
+            os.path.join(venv_path, "lib", "python3.13", "site-packages"),
+            os.path.join(venv_path, "Lib", "site-packages"),  # Windows
+        ]
+
+        for path in possible_paths:
+            if os.path.isdir(path):
+                logger.debug(f"Found venv site-packages: {path}")
+                return path
+
+        logger.warning(f"Could not find site-packages in venv: {venv_path}")
+        return None
+
+    def copy_to_venv(
+        self,
+        package_names: List[str],
+        venv_path: Optional[str] = None,
+        include_metadata: bool = True,
+        skip_errors: bool = True,
+    ) -> Tuple[List[str], Dict[str, str]]:
+        """
+        Copy packages to virtual environment site-packages.
+
+        Args:
+            package_names: List of package names to copy.
+            venv_path: Path to virtual environment. If None, uses current venv.
+            include_metadata: Whether to copy metadata.
+            skip_errors: Whether to continue on error.
+
+        Returns:
+            Tuple of (copied_packages, failed_packages).
+        """
+        site_packages = self.find_venv_site_packages(venv_path)
+        if not site_packages:
+            error_msg = "Could not find venv site-packages directory"
+            logger.error(error_msg)
+            return [], {name: error_msg for name in package_names}
+
+        logger.info(f"Copying packages to venv: {site_packages}")
+        return self.copy_packages(package_names, site_packages, include_metadata, skip_errors)
+
+    def copy_to_venv_with_dependencies(
+        self,
+        package_names: List[str],
+        venv_path: Optional[str] = None,
+        include_metadata: bool = True,
+    ) -> Tuple[List[str], Dict[str, str]]:
+        """
+        Copy packages and their dependencies to virtual environment.
+
+        Args:
+            package_names: List of package names to copy.
+            venv_path: Path to virtual environment. If None, uses current venv.
+            include_metadata: Whether to copy metadata.
+
+        Returns:
+            Tuple of (copied_packages, failed_packages).
+        """
+        site_packages = self.find_venv_site_packages(venv_path)
+        if not site_packages:
+            error_msg = "Could not find venv site-packages directory"
+            logger.error(error_msg)
+            return [], {name: error_msg for name in package_names}
+
+        logger.info(f"Copying packages with dependencies to venv: {site_packages}")
+        return self.copy_packages_with_dependencies(package_names, site_packages, include_metadata)
+
 
 def copy_package_simple(
     package_name: str, destination: str, source_dir: Optional[str] = None
@@ -566,6 +657,54 @@ def list_system_packages(source_dir: Optional[str] = None) -> List[str]:
         return []
 
 
+def copy_to_venv_simple(
+    package_names: List[str],
+    venv_path: Optional[str] = None,
+    source_dir: Optional[str] = None,
+) -> Tuple[List[str], Dict[str, str]]:
+    """
+    Simple function to copy packages to virtual environment.
+
+    Args:
+        package_names: List of package names to copy.
+        venv_path: Path to virtual environment. If None, uses current venv.
+        source_dir: Source directory (optional).
+
+    Returns:
+        Tuple of (copied_packages, failed_packages).
+    """
+    try:
+        copier = PackageCopier(source_dir=source_dir)
+        return copier.copy_to_venv(package_names, venv_path)
+    except Exception as e:
+        logger.error(f"Error copying to venv: {e}")
+        return [], {name: str(e) for name in package_names}
+
+
+def copy_to_venv_with_deps_simple(
+    package_names: List[str],
+    venv_path: Optional[str] = None,
+    source_dir: Optional[str] = None,
+) -> Tuple[List[str], Dict[str, str]]:
+    """
+    Simple function to copy packages with dependencies to virtual environment.
+
+    Args:
+        package_names: List of package names to copy.
+        venv_path: Path to virtual environment. If None, uses current venv.
+        source_dir: Source directory (optional).
+
+    Returns:
+        Tuple of (copied_packages, failed_packages).
+    """
+    try:
+        copier = PackageCopier(source_dir=source_dir)
+        return copier.copy_to_venv_with_dependencies(package_names, venv_path)
+    except Exception as e:
+        logger.error(f"Error copying to venv with dependencies: {e}")
+        return [], {name: str(e) for name in package_names}
+
+
 if __name__ == "__main__":
     # Example usage
     import argparse