python-package-folder 3.1.1__py3-none-any.whl → 3.1.3__py3-none-any.whl

python_package_folder/manager.py

@@ -88,6 +88,8 @@ class BuildManager:
         self.subfolder_config: SubfolderBuildConfig | None = None
         # Cache for package name lookups (expensive operation)
         self._packages_distributions_cache: dict[str, list[str]] | None = None
+        # Track files with modified imports and their original content
+        self._modified_import_files: dict[Path, str] = {}
 
         # Check if it's a valid Python package directory
         if not any(self.src_dir.glob("*.py")) and not (self.src_dir / "__init__.py").exists():
@@ -270,6 +272,13 @@ class BuildManager:
         for dep in external_deps:
             self._copy_dependency(dep)
 
+        # For subfolder builds, fix imports
+        if self._is_subfolder_build() and external_deps:
+            # Fix relative imports in copied dependency files (convert to absolute)
+            self._fix_relative_imports_in_copied_files(external_deps)
+            # Convert absolute imports of copied dependencies and local files to relative imports
+            self._convert_imports_to_relative(python_files, external_deps)
+
         # For subfolder builds, extract third-party dependencies and add to pyproject.toml
         if self._is_subfolder_build() and self.subfolder_config:
             # Re-analyze all Python files (including copied dependencies) to find third-party imports
@@ -629,6 +638,258 @@ class BuildManager:
 
         return sorted(list(third_party_packages))
 
+    def _fix_relative_imports_in_copied_files(
+        self, external_deps: list[ExternalDependency]
+    ) -> None:
+        """
+        Fix relative imports in copied dependency files.
+
+        When files are copied into the subfolder, their relative imports (like
+        `from ._shared.shared_dataclasses import ...`) break because the file
+        structure has changed. Convert these to absolute imports based on the
+        target location.
+
+        Args:
+            external_deps: List of external dependencies that were copied
+        """
+        import ast
+        import re
+
+        # Find all Python files in copied dependencies
+        copied_files: list[Path] = []
+        for dep in external_deps:
+            if dep.target_path.is_file() and dep.target_path.suffix == ".py":
+                copied_files.append(dep.target_path)
+            elif dep.target_path.is_dir():
+                copied_files.extend(dep.target_path.rglob("*.py"))
+
+        for file_path in copied_files:
+            try:
+                content = file_path.read_text(encoding="utf-8")
+                original_content = content
+                lines = content.split("\n")
+                modified = False
+
+                try:
+                    tree = ast.parse(content, filename=str(file_path))
+                except SyntaxError:
+                    continue
+
+                lines_to_modify: dict[int, str] = {}
+
+                for node in ast.walk(tree):
+                    if isinstance(node, ast.ImportFrom):
+                        if node.module is None:
+                            continue
+
+                        # Check if this is a relative import (level > 0)
+                        if node.level == 0:
+                            continue
+
+                        line_num = node.lineno - 1
+                        if line_num < 0 or line_num >= len(lines):
+                            continue
+
+                        original_line = lines[line_num]
+
+                        # Convert relative import to absolute based on target location
+                        # When a file is copied to the package root, relative imports need to be absolute
+                        # For example: from ._shared.shared_dataclasses -> from _shared.shared_dataclasses
+                        if node.module:
+                            # Remove the leading dots and convert to absolute
+                            # If it was `from ._shared.shared_dataclasses`, it becomes `from _shared.shared_dataclasses`
+                            absolute_module = node.module
+                            new_line = re.sub(
+                                rf"^(\s*)from\s+\.+{re.escape(node.module)}\s+import",
+                                rf"\1from {absolute_module} import",
+                                original_line,
+                            )
+                        else:
+                            # from . import X -> from . import X (keep as relative, but at package root level)
+                            # Actually, if we're at package root, this should work as-is
+                            # But if the file was in a subdirectory, we need to adjust
+                            # For now, keep it as relative import
+                            continue
+
+                        if new_line != original_line:
+                            lines_to_modify[line_num] = new_line
+                            modified = True
+
+                if modified:
+                    for line_num, new_line in lines_to_modify.items():
+                        lines[line_num] = new_line
+
+                    new_content = "\n".join(lines)
+                    if file_path not in self._modified_import_files:
+                        self._modified_import_files[file_path] = original_content
+
+                    file_path.write_text(new_content, encoding="utf-8")
+                    print(f"Fixed relative imports in copied file: {file_path}")
+
+            except Exception as e:
+                print(
+                    f"Warning: Could not fix imports in copied file {file_path}: {e}",
+                    file=sys.stderr,
+                )
+
+    def _convert_imports_to_relative(
+        self, python_files: list[Path], external_deps: list[ExternalDependency]
+    ) -> None:
+        """
+        Convert absolute imports to relative imports for subfolder builds.
+
+        For subfolder builds, when external dependencies are copied into the subfolder,
+        imports need to be converted from absolute to relative so they work correctly
+        when the package is installed. This includes:
+        1. Imports of copied dependencies (e.g., `from _shared.image_utils` -> `from ._shared.image_utils`)
+        2. Imports of local files within the subfolder (e.g., `from detect_empty_drawings_utils` -> `from .detect_empty_drawings_utils`)
+
+        Args:
+            python_files: List of Python files in the source directory
+            external_deps: List of external dependencies that were copied
+        """
+        import ast
+        import re
+
+        # Build a set of import names that were copied
+        copied_import_names: set[str] = set()
+        for dep in external_deps:
+            root_module = dep.import_name.split(".")[0]
+            copied_import_names.add(root_module)
+            copied_import_names.add(dep.import_name)
+
+        # Build a set of local file names in the subfolder (excluding copied dependencies)
+        local_file_names: set[str] = set()
+        for file_path in python_files:
+            # Skip files that are part of copied dependencies
+            is_copied_file = any(file_path.is_relative_to(dep.target_path) for dep in external_deps)
+            if is_copied_file:
+                continue
+            if not file_path.is_relative_to(self.src_dir):
+                continue
+            # Get the module name (filename without .py extension)
+            if file_path.suffix == ".py":
+                module_name = file_path.stem
+                if module_name != "__init__":
+                    local_file_names.add(module_name)
+
+        # Only modify files that are in the original subfolder (not the copied dependencies)
+        for file_path in python_files:
+            # Skip files that are part of copied dependencies
+            is_copied_file = any(file_path.is_relative_to(dep.target_path) for dep in external_deps)
+            if is_copied_file:
+                continue
+
+            # Skip if file is not in src_dir (shouldn't happen, but safety check)
+            if not file_path.is_relative_to(self.src_dir):
+                continue
+
+            try:
+                content = file_path.read_text(encoding="utf-8")
+                original_content = content
+                lines = content.split("\n")
+                modified = False
+
+                # Parse the file with AST to find imports accurately
+                try:
+                    tree = ast.parse(content, filename=str(file_path))
+                except SyntaxError:
+                    # Skip files with syntax errors
+                    continue
+
+                # Track which lines need to be modified
+                lines_to_modify: dict[int, str] = {}
+
+                for node in ast.walk(tree):
+                    if isinstance(node, ast.ImportFrom):
+                        if node.module is None:
+                            continue
+
+                        # Check if this import matches a copied dependency or a local file
+                        root_module = node.module.split(".")[0]
+                        is_copied_dependency = (
+                            root_module in copied_import_names or node.module in copied_import_names
+                        )
+                        is_local_file = root_module in local_file_names
+
+                        if not is_copied_dependency and not is_local_file:
+                            continue
+
+                        # Get the line content
+                        line_num = node.lineno - 1  # Convert to 0-based index
+                        if line_num < 0 or line_num >= len(lines):
+                            continue
+
+                        original_line = lines[line_num]
+
+                        # Skip if already a relative import
+                        if original_line.strip().startswith("from ."):
+                            continue
+
+                        # Convert absolute import to relative import
+                        # from _shared.image_utils import ... -> from ._shared.image_utils import ...
+                        new_line = re.sub(
+                            rf"^(\s*)from\s+{re.escape(node.module)}\s+import",
+                            rf"\1from .{node.module} import",
+                            original_line,
+                        )
+
+                        if new_line != original_line:
+                            lines_to_modify[line_num] = new_line
+                            modified = True
+
+                    elif isinstance(node, ast.Import):
+                        # Handle "import X" statements
+                        for alias in node.names:
+                            root_module = alias.name.split(".")[0]
+                            is_copied_dependency = root_module in copied_import_names
+                            is_local_file = root_module in local_file_names
+
+                            if not is_copied_dependency and not is_local_file:
+                                continue
+
+                            line_num = node.lineno - 1
+                            if line_num < 0 or line_num >= len(lines):
+                                continue
+
+                            original_line = lines[line_num]
+
+                            # Skip if already a relative import
+                            if original_line.strip().startswith("import ."):
+                                continue
+
+                            # Convert "import _shared" to "from . import _shared"
+                            # This is more complex, so we'll use a regex replacement
+                            new_line = re.sub(
+                                rf"^(\s*)import\s+{re.escape(alias.name)}\b",
+                                rf"\1from . import {alias.name}",
+                                original_line,
+                            )
+
+                            if new_line != original_line:
+                                lines_to_modify[line_num] = new_line
+                                modified = True
+
+                # Apply modifications
+                if modified:
+                    for line_num, new_line in lines_to_modify.items():
+                        lines[line_num] = new_line
+
+                    new_content = "\n".join(lines)
+                    # Store original content for restoration
+                    if file_path not in self._modified_import_files:
+                        self._modified_import_files[file_path] = original_content
+
+                    # Write modified content
+                    file_path.write_text(new_content, encoding="utf-8")
+                    print(f"Converted imports to relative in: {file_path}")
+
+            except Exception as e:
+                print(
+                    f"Warning: Could not modify imports in {file_path}: {e}",
+                    file=sys.stderr,
+                )
+
     def _report_ambiguous_imports(self, python_files: list[Path]) -> None:
         """
         Report any ambiguous imports that couldn't be resolved.
@@ -708,6 +969,19 @@ class BuildManager:
         self.copied_files.clear()
         self.copied_dirs.clear()
 
+        # Restore files with modified imports
+        for file_path, original_content in self._modified_import_files.items():
+            if file_path.exists():
+                try:
+                    file_path.write_text(original_content, encoding="utf-8")
+                    print(f"Restored original imports in: {file_path}")
+                except Exception as e:
+                    print(
+                        f"Warning: Could not restore imports in {file_path}: {e}",
+                        file=sys.stderr,
+                    )
+        self._modified_import_files.clear()
+
         # Remove all .egg-info directories in src_dir and project_root
         self._cleanup_egg_info_dirs()
 

{python_package_folder-3.1.1.dist-info → python_package_folder-3.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-package-folder
-Version: 3.1.1
+Version: 3.1.3
 Summary: Python package to automatically package and build a folder, fetching all relevant dependencies.
 Project-URL: Repository, https://github.com/alelom/python-package-folder
 Author-email: Alessio Lombardi <work@alelom.com>

{python_package_folder-3.1.1.dist-info → python_package_folder-3.1.3.dist-info}/RECORD

@@ -2,7 +2,7 @@ python_package_folder/__init__.py,sha256=DQt-uldOEKfh0MUqCvKdeNKOnpuOvpb7blYvXMy
 python_package_folder/__main__.py,sha256=a-__-VLhYw-J7S7CsHdhtEvQr3RiAZxiYDvKhKTgMX4,291
 python_package_folder/analyzer.py,sha256=cmTNUDCWBIh3XZ_mShlQVG1P9NN_oe3FUBTirVtYfTQ,16709
 python_package_folder/finder.py,sha256=RPidZ7LKCFuQ_KgCFIZdHWPXsZIDor3M4C0hKeYW7EI,11799
-python_package_folder/manager.py,sha256=kyULmS948uW3yWUcabc2EVK_Nic3wrWMj8qCs18jj2Q,43526
+python_package_folder/manager.py,sha256=Z9RPg0ZQ7jZhmEXfCzX9OrD_oiA5p2Pnm5Y9tgW3ObQ,55970
 python_package_folder/publisher.py,sha256=TSjdOvxvnWLbJCnduTK_xZBRfvsrq9kpEH-sfebeWkU,13507
 python_package_folder/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 python_package_folder/python_package_folder.py,sha256=RPsqRcIy_LjzzTHdp4qdtFJ4-4xhtR_0YLIC0RlUxFo,8841
@@ -10,8 +10,8 @@ python_package_folder/subfolder_build.py,sha256=oH_KKLJIMByUZCl8y3AyohUO6Om0OvsI
 python_package_folder/types.py,sha256=3yeSRR5p_3PDKEAaehW_RJ7NwJHexOIeA08bGaT1iSY,2368
 python_package_folder/utils.py,sha256=lIkWsFKeAYAJ9TDUM99T4pUBHJVbUvCdUgkWQN-LUho,3111
 python_package_folder/version.py,sha256=kIDP6S9trEfs9gj7lBYGxrWm4RPssRla24UtlO9Jkh4,9111
-python_package_folder-3.1.1.dist-info/METADATA,sha256=ccaosiXpsDgxQyy6UYHLYj34BKnrosnLbvFBjh18kNY,33282
-python_package_folder-3.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-python_package_folder-3.1.1.dist-info/entry_points.txt,sha256=ttu4wAhoYSHGhWQNercLz9IVTTpXxhVlRA9vSTvaLe0,91
-python_package_folder-3.1.1.dist-info/licenses/LICENSE,sha256=vNgRJh8YiecqZoZld7TtwPI5I72HIymKD9g32fiJjCE,1073
-python_package_folder-3.1.1.dist-info/RECORD,,
+python_package_folder-3.1.3.dist-info/METADATA,sha256=dkCtq6nLmfiygTwYXzgVomMejX4AfQTM1U3LYdev8zQ,33282
+python_package_folder-3.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+python_package_folder-3.1.3.dist-info/entry_points.txt,sha256=ttu4wAhoYSHGhWQNercLz9IVTTpXxhVlRA9vSTvaLe0,91
+python_package_folder-3.1.3.dist-info/licenses/LICENSE,sha256=vNgRJh8YiecqZoZld7TtwPI5I72HIymKD9g32fiJjCE,1073
+python_package_folder-3.1.3.dist-info/RECORD,,