televault 0.1.0__py3-none-any.whl → 2.0.0__py3-none-any.whl

televault/compress.py

@@ -1,8 +1,6 @@
 """Compression utilities for TeleVault - zstd for speed and ratio."""
 
-import io
 from pathlib import Path
-from typing import BinaryIO
 
 import zstandard as zstd
 
@@ -15,17 +13,51 @@ DEFAULT_LEVEL = 3
 # File extensions that are already compressed (skip compression)
 INCOMPRESSIBLE_EXTENSIONS = {
     # Images
-    ".jpg", ".jpeg", ".png", ".gif", ".webp", ".heic", ".heif", ".avif",
+    ".jpg",
+    ".jpeg",
+    ".png",
+    ".gif",
+    ".webp",
+    ".heic",
+    ".heif",
+    ".avif",
     # Video
-    ".mp4", ".mkv", ".avi", ".mov", ".webm", ".m4v", ".wmv", ".flv",
+    ".mp4",
+    ".mkv",
+    ".avi",
+    ".mov",
+    ".webm",
+    ".m4v",
+    ".wmv",
+    ".flv",
     # Audio
-    ".mp3", ".aac", ".ogg", ".opus", ".flac", ".m4a", ".wma",
+    ".mp3",
+    ".aac",
+    ".ogg",
+    ".opus",
+    ".flac",
+    ".m4a",
+    ".wma",
     # Archives
-    ".zip", ".gz", ".bz2", ".xz", ".7z", ".rar", ".zst", ".lz4", ".lzma",
+    ".zip",
+    ".gz",
+    ".bz2",
+    ".xz",
+    ".7z",
+    ".rar",
+    ".zst",
+    ".lz4",
+    ".lzma",
     # Documents (already compressed)
-    ".pdf", ".docx", ".xlsx", ".pptx", ".odt",
+    ".pdf",
+    ".docx",
+    ".xlsx",
+    ".pptx",
+    ".odt",
     # Other
-    ".woff", ".woff2", ".br",
+    ".woff",
+    ".woff2",
+    ".br",
 }
 
 
@@ -49,53 +81,57 @@ def decompress_data(data: bytes, max_output_size: int = 0) -> bytes:
     return dctx.decompress(data, max_output_size=max_output_size)
 
 
-def compress_file(input_path: str | Path, output_path: str | Path, level: int = DEFAULT_LEVEL) -> float:
+def compress_file(
+    input_path: str | Path,
+    output_path: str | Path,
+    level: int = DEFAULT_LEVEL,
+) -> float:
     """
     Compress a file using zstd.
-    
+
     Returns compression ratio (compressed_size / original_size).
     """
     cctx = zstd.ZstdCompressor(level=level)
-    
+
     with open(input_path, "rb") as fin, open(output_path, "wb") as fout:
         cctx.copy_stream(fin, fout)
-    
+
     original_size = Path(input_path).stat().st_size
     compressed_size = Path(output_path).stat().st_size
-    
+
     return compressed_size / original_size if original_size > 0 else 1.0
 
 
 def decompress_file(input_path: str | Path, output_path: str | Path) -> None:
     """Decompress a zstd file."""
     dctx = zstd.ZstdDecompressor()
-    
+
     with open(input_path, "rb") as fin, open(output_path, "wb") as fout:
         dctx.copy_stream(fin, fout)
 
 
 class StreamingCompressor:
     """Streaming compressor for pipeline integration."""
-    
+
     def __init__(self, level: int = DEFAULT_LEVEL):
         self.cctx = zstd.ZstdCompressor(level=level)
         self.compressor = self.cctx.compressobj()
         self.total_in = 0
         self.total_out = 0
-    
+
     def compress(self, data: bytes) -> bytes:
         """Compress a chunk of data."""
         self.total_in += len(data)
         compressed = self.compressor.compress(data)
         self.total_out += len(compressed)
         return compressed
-    
+
     def flush(self) -> bytes:
         """Flush remaining data and finalize compression."""
         final = self.compressor.flush()
         self.total_out += len(final)
         return final
-    
+
     @property
     def ratio(self) -> float:
         """Current compression ratio."""
@@ -106,11 +142,11 @@ class StreamingCompressor:
 
 class StreamingDecompressor:
     """Streaming decompressor for pipeline integration."""
-    
+
     def __init__(self):
         self.dctx = zstd.ZstdDecompressor()
         self.decompressor = self.dctx.decompressobj()
-    
+
     def decompress(self, data: bytes) -> bytes:
         """Decompress a chunk of data."""
         return self.decompressor.decompress(data)
@@ -119,15 +155,15 @@ class StreamingDecompressor:
 def estimate_compressed_size(original_size: int, filename: str) -> int:
     """
     Estimate compressed size based on file type.
-    
+
     Returns estimated size in bytes.
     """
     if not should_compress(filename):
         return original_size
-    
+
     # Typical compression ratios by type
     suffix = Path(filename).suffix.lower()
-    
+
     if suffix in {".txt", ".log", ".csv", ".json", ".xml", ".html", ".md"}:
         return int(original_size * 0.2)  # Text compresses well
     elif suffix in {".sql", ".py", ".js", ".ts", ".go", ".rs", ".c", ".cpp", ".h"}:

televault/config.py

@@ -1,10 +1,9 @@
 """Configuration management for TeleVault."""
 
 import json
-from pathlib import Path
-from dataclasses import dataclass, field, asdict
-from typing import Optional
 import os
+from dataclasses import asdict, dataclass
+from pathlib import Path
 
 
 def get_config_dir() -> Path:
@@ -13,7 +12,7 @@ def get_config_dir() -> Path:
         base = Path(os.environ.get("APPDATA", "~"))
     else:  # Unix
         base = Path(os.environ.get("XDG_CONFIG_HOME", "~/.config"))
-    
+
     config_dir = base.expanduser() / "televault"
     config_dir.mkdir(parents=True, exist_ok=True)
     return config_dir
@@ -25,7 +24,7 @@ def get_data_dir() -> Path:
         base = Path(os.environ.get("LOCALAPPDATA", "~"))
     else:
         base = Path(os.environ.get("XDG_DATA_HOME", "~/.local/share"))
-    
+
     data_dir = base.expanduser() / "televault"
     data_dir.mkdir(parents=True, exist_ok=True)
     return data_dir
@@ -34,44 +33,44 @@ def get_data_dir() -> Path:
 @dataclass
 class Config:
     """TeleVault configuration."""
-    
+
     # Telegram settings
-    channel_id: Optional[int] = None
-    
+    channel_id: int | None = None
+
     # Chunking
     chunk_size: int = 100 * 1024 * 1024  # 100MB
-    
+
     # Processing options
     compression: bool = True
     encryption: bool = True
-    
+
     # Concurrency
     parallel_uploads: int = 3
     parallel_downloads: int = 5
-    
+
     # Retry settings
     max_retries: int = 3
     retry_delay: float = 1.0
-    
+
     def save(self) -> None:
         """Save config to file."""
         config_path = get_config_dir() / "config.json"
         with open(config_path, "w") as f:
             json.dump(asdict(self), f, indent=2)
-    
+
     @classmethod
     def load(cls) -> "Config":
         """Load config from file."""
         config_path = get_config_dir() / "config.json"
-        
+
         if not config_path.exists():
             return cls()
-        
+
         with open(config_path) as f:
             data = json.load(f)
-        
+
         return cls(**data)
-    
+
     @classmethod
     def load_or_create(cls) -> "Config":
         """Load config or create default."""