provide-foundation 0.0.0.dev2__py3-none-any.whl → 0.0.0.dev3__py3-none-any.whl

provide/foundation/archive/operations.py

@@ -1,15 +1,14 @@
 """Archive operation chains and helpers."""
 
-import tempfile
 from pathlib import Path
-from typing import Callable
+import tempfile
 
 from attrs import define, field
 
 from provide.foundation.archive.base import ArchiveError
-from provide.foundation.archive.tar import TarArchive
-from provide.foundation.archive.gzip import GzipCompressor
 from provide.foundation.archive.bzip2 import Bzip2Compressor
+from provide.foundation.archive.gzip import GzipCompressor
+from provide.foundation.archive.tar import TarArchive
 from provide.foundation.archive.zip import ZipArchive
 from provide.foundation.file import ensure_parent_dir
 from provide.foundation.file.safe import safe_delete
@@ -22,30 +21,30 @@ logger = get_logger(__name__)
 class OperationChain:
     """
     Chain multiple archive operations together.
-    
+
     Enables complex operations like tar.gz, tar.bz2, etc.
     Operations are executed in order for creation, reversed for extraction.
     """
-    
+
     operations: list[str] = field(factory=list)
-    
+
     def execute(self, source: Path, output: Path) -> Path:
         """
         Execute operation chain on source.
-        
+
         Args:
             source: Source file or directory
             output: Final output path
-            
+
         Returns:
             Path to final output
-            
+
         Raises:
             ArchiveError: If any operation fails
         """
         current = source
         temp_files = []
-        
+
         try:
             for i, op in enumerate(self.operations):
                 # Determine output for this operation
@@ -59,54 +58,54 @@ class OperationChain:
                     temp.close()
                     next_output = Path(temp.name)
                     temp_files.append(next_output)
-                
+
                 # Execute operation
                 current = self._execute_operation(op, current, next_output)
                 logger.debug(f"Executed operation '{op}': {current}")
-            
+
             return current
-            
+
         except Exception as e:
             raise ArchiveError(f"Operation chain failed: {e}") from e
         finally:
             # Clean up temp files using Foundation's safe file operations
             for temp in temp_files:
                 safe_delete(temp, missing_ok=True)
-    
+
     def reverse(self, source: Path, output: Path) -> Path:
         """
         Reverse operation chain (extract/decompress).
-        
+
         Args:
             source: Source archive
             output: Final output path
-            
+
         Returns:
             Path to final output
-            
+
         Raises:
             ArchiveError: If any operation fails
         """
         # Reverse the operations and invert them
         reverse_map = {
-            'tar': 'untar',
-            'untar': 'tar',
-            'gzip': 'gunzip', 
-            'gunzip': 'gzip',
-            'bzip2': 'bunzip2',
-            'bunzip2': 'bzip2',
-            'zip': 'unzip',
-            'unzip': 'zip',
+            "tar": "untar",
+            "untar": "tar",
+            "gzip": "gunzip",
+            "gunzip": "gzip",
+            "bzip2": "bunzip2",
+            "bunzip2": "bzip2",
+            "zip": "unzip",
+            "unzip": "zip",
         }
-        
+
         reversed_ops = []
         for op in reversed(self.operations):
             reversed_op = reverse_map.get(op.lower(), op)
             reversed_ops.append(reversed_op)
-        
+
         reversed_chain = OperationChain(operations=reversed_ops)
         return reversed_chain.execute(source, output)
-    
+
     def _execute_operation(self, operation: str, source: Path, output: Path) -> Path:
         """Execute a single operation."""
         match operation.lower():
@@ -136,7 +135,7 @@ class OperationChain:
                 return zip_archive.extract(source, output)
             case _:
                 raise ArchiveError(f"Unknown operation: {operation}")
-    
+
     def _get_suffix_for_operation(self, operation: str) -> str:
         """Get file suffix for operation."""
         suffixes = {
@@ -151,34 +150,34 @@ class OperationChain:
 class ArchiveOperations:
     """
     Helper class for common archive operation patterns.
-    
+
     Provides convenient methods for common archive formats.
     """
-    
+
     @staticmethod
     def create_tar_gz(source: Path, output: Path, deterministic: bool = True) -> Path:
         """
         Create .tar.gz archive in one step.
-        
+
         Args:
             source: Source file or directory
             output: Output path (should end with .tar.gz)
             deterministic: Create reproducible archive
-            
+
         Returns:
             Path to created archive
-            
+
         Raises:
             ArchiveError: If creation fails
         """
         ensure_parent_dir(output)
-        
+
         # Create temp tar file
-        temp_tar = output.with_suffix('.tar')
+        temp_tar = output.with_suffix(".tar")
         try:
             tar = TarArchive(deterministic=deterministic)
             tar.create(source, temp_tar)
-            
+
             # Compress to final output
             gzip = GzipCompressor()
             return gzip.compress_file(temp_tar, output)
@@ -186,32 +185,32 @@ class ArchiveOperations:
             # Clean up temp file
             if temp_tar.exists():
                 temp_tar.unlink()
-    
+
     @staticmethod
     def extract_tar_gz(archive: Path, output: Path) -> Path:
         """
         Extract .tar.gz archive in one step.
-        
+
         Args:
             archive: Archive path
             output: Output directory
-            
+
         Returns:
             Path to extraction directory
-            
+
         Raises:
             ArchiveError: If extraction fails
         """
         output.mkdir(parents=True, exist_ok=True)
-        
+
         # Decompress to temp file
-        with tempfile.NamedTemporaryFile(suffix='.tar', delete=False) as temp:
+        with tempfile.NamedTemporaryFile(suffix=".tar", delete=False) as temp:
             temp_tar = Path(temp.name)
-        
+
         try:
             gzip = GzipCompressor()
             gzip.decompress_file(archive, temp_tar)
-            
+
             # Extract tar
             tar = TarArchive()
             return tar.extract(temp_tar, output)
@@ -219,31 +218,31 @@ class ArchiveOperations:
             # Clean up temp file
             if temp_tar.exists():
                 temp_tar.unlink()
-    
+
     @staticmethod
     def create_tar_bz2(source: Path, output: Path, deterministic: bool = True) -> Path:
         """
         Create .tar.bz2 archive in one step.
-        
+
         Args:
             source: Source file or directory
             output: Output path (should end with .tar.bz2)
             deterministic: Create reproducible archive
-            
+
         Returns:
             Path to created archive
-            
+
         Raises:
             ArchiveError: If creation fails
         """
         ensure_parent_dir(output)
-        
+
         # Create temp tar file
-        temp_tar = output.with_suffix('.tar')
+        temp_tar = output.with_suffix(".tar")
         try:
             tar = TarArchive(deterministic=deterministic)
             tar.create(source, temp_tar)
-            
+
             # Compress to final output
             bz2 = Bzip2Compressor()
             return bz2.compress_file(temp_tar, output)
@@ -251,32 +250,32 @@ class ArchiveOperations:
             # Clean up temp file
             if temp_tar.exists():
                 temp_tar.unlink()
-    
+
     @staticmethod
     def extract_tar_bz2(archive: Path, output: Path) -> Path:
         """
         Extract .tar.bz2 archive in one step.
-        
+
         Args:
             archive: Archive path
             output: Output directory
-            
+
         Returns:
             Path to extraction directory
-            
+
         Raises:
             ArchiveError: If extraction fails
         """
         output.mkdir(parents=True, exist_ok=True)
-        
+
         # Decompress to temp file
-        with tempfile.NamedTemporaryFile(suffix='.tar', delete=False) as temp:
+        with tempfile.NamedTemporaryFile(suffix=".tar", delete=False) as temp:
             temp_tar = Path(temp.name)
-        
+
         try:
             bz2 = Bzip2Compressor()
             bz2.decompress_file(archive, temp_tar)
-            
+
             # Extract tar
             tar = TarArchive()
             return tar.extract(temp_tar, output)
@@ -284,51 +283,51 @@ class ArchiveOperations:
             # Clean up temp file
             if temp_tar.exists():
                 temp_tar.unlink()
-    
+
     @staticmethod
     def detect_format(file: Path) -> list[str]:
         """
         Detect archive format and return operation chain.
-        
+
         Args:
             file: File path to analyze
-            
+
         Returns:
             List of operations needed to extract
-            
+
         Raises:
             ArchiveError: If format cannot be detected
         """
         name = file.name.lower()
-        
+
         # Check by extension
-        if name.endswith('.tar.gz') or name.endswith('.tgz'):
-            return ['gunzip', 'untar']
-        elif name.endswith('.tar.bz2') or name.endswith('.tbz2'):
-            return ['bunzip2', 'untar']
-        elif name.endswith('.tar'):
-            return ['untar']
-        elif name.endswith('.gz'):
-            return ['gunzip']
-        elif name.endswith('.bz2'):
-            return ['bunzip2']
-        elif name.endswith('.zip'):
-            return ['unzip']
-        
+        if name.endswith(".tar.gz") or name.endswith(".tgz"):
+            return ["gunzip", "untar"]
+        elif name.endswith(".tar.bz2") or name.endswith(".tbz2"):
+            return ["bunzip2", "untar"]
+        elif name.endswith(".tar"):
+            return ["untar"]
+        elif name.endswith(".gz"):
+            return ["gunzip"]
+        elif name.endswith(".bz2"):
+            return ["bunzip2"]
+        elif name.endswith(".zip"):
+            return ["unzip"]
+
         # Check by magic numbers
         try:
-            with open(file, 'rb') as f:
+            with open(file, "rb") as f:
                 magic = f.read(4)
-                
-            if magic[:2] == b'\x1f\x8b':  # gzip
-                return ['gunzip']
-            elif magic[:3] == b'BZh':  # bzip2
-                return ['bunzip2']
-            elif magic[:4] == b'PK\x03\x04':  # zip
-                return ['unzip']
-            elif magic[:3] == b'ustar':  # tar (at offset 257)
-                return ['untar']
+
+            if magic[:2] == b"\x1f\x8b":  # gzip
+                return ["gunzip"]
+            elif magic[:3] == b"BZh":  # bzip2
+                return ["bunzip2"]
+            elif magic[:4] == b"PK\x03\x04":  # zip
+                return ["unzip"]
+            elif magic[:3] == b"ustar":  # tar (at offset 257)
+                return ["untar"]
         except Exception:
             pass
-        
-        raise ArchiveError(f"Cannot detect format of {file}")
+
+        raise ArchiveError(f"Cannot detect format of {file}")

provide/foundation/archive/tar.py

@@ -1,11 +1,11 @@
 """TAR archive implementation."""
 
-import tarfile
 from pathlib import Path
+import tarfile
 
 from attrs import define, field
 
-from provide.foundation.archive.base import BaseArchive, ArchiveError
+from provide.foundation.archive.base import ArchiveError, BaseArchive
 from provide.foundation.file import ensure_parent_dir
 from provide.foundation.logger import get_logger
 
@@ -16,32 +16,32 @@ logger = get_logger(__name__)
 class TarArchive(BaseArchive):
     """
     TAR archive implementation.
-    
+
     Creates and extracts TAR archives with optional metadata preservation
     and deterministic output for reproducible builds.
     """
-    
+
     deterministic: bool = field(default=True)
     preserve_metadata: bool = field(default=True)
     preserve_permissions: bool = field(default=True)
-    
+
     def create(self, source: Path, output: Path) -> Path:
         """
         Create TAR archive from source.
-        
+
         Args:
             source: Source file or directory to archive
             output: Output TAR file path
-            
+
         Returns:
             Path to created archive
-            
+
         Raises:
             ArchiveError: If archive creation fails
         """
         try:
             ensure_parent_dir(output)
-            
+
             with tarfile.open(output, "w") as tar:
                 if source.is_dir():
                     # Add all files in directory
@@ -52,52 +52,52 @@ class TarArchive(BaseArchive):
                 else:
                     # Add single file
                     self._add_file(tar, source, source.name)
-            
+
             logger.debug(f"Created TAR archive: {output}")
             return output
-            
+
         except Exception as e:
             raise ArchiveError(f"Failed to create TAR archive: {e}") from e
 
     def extract(self, archive: Path, output: Path) -> Path:
         """
         Extract TAR archive to output directory.
-        
+
         Args:
             archive: TAR archive file path
             output: Output directory path
-            
+
         Returns:
             Path to extraction directory
-            
+
         Raises:
             ArchiveError: If extraction fails
         """
         try:
             output.mkdir(parents=True, exist_ok=True)
-            
+
             with tarfile.open(archive, "r") as tar:
                 # Security check - prevent path traversal
                 for member in tar.getmembers():
                     if member.name.startswith("/") or ".." in member.name:
                         raise ArchiveError(f"Unsafe path in archive: {member.name}")
-                
+
                 # Extract all
                 tar.extractall(output)
-            
+
             logger.debug(f"Extracted TAR archive to: {output}")
             return output
-            
+
         except Exception as e:
             raise ArchiveError(f"Failed to extract TAR archive: {e}") from e
 
     def validate(self, archive: Path) -> bool:
         """
         Validate TAR archive integrity.
-        
+
         Args:
             archive: TAR archive file path
-            
+
         Returns:
             True if archive is valid, False otherwise
         """
@@ -110,17 +110,17 @@ class TarArchive(BaseArchive):
             return True
         except Exception:
             return False
-    
+
     def list_contents(self, archive: Path) -> list[str]:
         """
         List contents of TAR archive.
-        
+
         Args:
             archive: TAR archive file path
-            
+
         Returns:
             List of file paths in archive
-            
+
         Raises:
             ArchiveError: If listing fails
         """
@@ -133,18 +133,20 @@ class TarArchive(BaseArchive):
             return sorted(contents)
         except Exception as e:
             raise ArchiveError(f"Failed to list TAR contents: {e}") from e
-    
-    def _add_file(self, tar: tarfile.TarFile, file_path: Path, arcname: str | Path) -> None:
+
+    def _add_file(
+        self, tar: tarfile.TarFile, file_path: Path, arcname: str | Path
+    ) -> None:
         """
         Add single file to TAR archive.
-        
+
         Args:
             tar: Open TarFile object
             file_path: Path to file to add
             arcname: Name in archive
         """
         tarinfo = tar.gettarinfo(str(file_path), str(arcname))
-        
+
         if self.deterministic:
             # Set consistent metadata for reproducible archives
             tarinfo.uid = 0
@@ -152,13 +154,13 @@ class TarArchive(BaseArchive):
             tarinfo.uname = ""
             tarinfo.gname = ""
             tarinfo.mtime = 0
-        
+
         if not self.preserve_permissions:
             # Normalize permissions
             if tarinfo.isfile():
                 tarinfo.mode = 0o644
             elif tarinfo.isdir():
                 tarinfo.mode = 0o755
-        
+
         with open(file_path, "rb") as f:
-            tar.addfile(tarinfo, f)
+            tar.addfile(tarinfo, f)