b2-cleanup 0.1.0__py3-none-any.whl → 0.1.4__py3-none-any.whl

b2_cleanup/__init__.py

@@ -0,0 +1,6 @@
+"""B2 Cleanup Tool - Clean up unfinished Backblaze B2 large uploads."""
+
+from .core import B2CleanupTool
+
+__version__ = "0.1.3"
+__all__ = ["B2CleanupTool"]

b2_cleanup/cli.py

@@ -0,0 +1,38 @@
+"""Command-line interface for B2 cleanup tool."""
+
+import logging
+import click
+from .core import B2CleanupTool
+
+
+@click.command()
+@click.argument("bucket")
+@click.option(
+    "--dry-run", is_flag=True, help="Only list unfinished uploads, do not cancel."
+)
+@click.option("--log-file", default="b2_cleanup.log", help="Path to log file.")
+@click.option("--key-id", help="Backblaze B2 applicationKeyId to override env/config.")
+@click.option("--key", help="Backblaze B2 applicationKey to override env/config.")
+def cli(bucket, dry_run, log_file, key_id, key):
+    """Clean up unfinished large file uploads in BUCKET."""
+    logger = logging.getLogger()
+    logger.setLevel(logging.INFO)
+    if logger.hasHandlers():
+        logger.handlers.clear()
+
+    file_handler = logging.FileHandler(log_file)
+    file_handler.setFormatter(
+        logging.Formatter("%(asctime)s [%(levelname)s] %(message)s")
+    )
+    logger.addHandler(file_handler)
+
+    stream_handler = logging.StreamHandler()
+    stream_handler.setFormatter(logging.Formatter("%(message)s"))
+    logger.addHandler(stream_handler)
+
+    tool = B2CleanupTool(dry_run=dry_run, override_key_id=key_id, override_key=key)
+    tool.cleanup_unfinished_uploads(bucket)
+
+
+if __name__ == "__main__":
+    cli()  # pragma: no cover

cleanup_unfinished_b2_uploads.py → b2_cleanup/core.py

@@ -1,18 +1,28 @@
+"""Core functionality for B2 cleanup tool."""
+
 import os
 import json
 import subprocess
 import logging
-import click
 from b2sdk.v2 import InMemoryAccountInfo, B2Api
 
 
 class B2CleanupTool:
+    """Tool to clean up unfinished large file uploads in B2 buckets."""
+
     def __init__(
         self,
         dry_run: bool = False,
         override_key_id: str = None,
         override_key: str = None,
     ):
+        """Initialize the B2 cleanup tool.
+
+        Args:
+            dry_run: If True, only list uploads but don't delete them
+            override_key_id: Optional B2 key ID to override env/config
+            override_key: Optional B2 application key to override env/config
+        """
         self.dry_run = dry_run
         self.logger = logging.getLogger("B2Cleanup")
         self.api = self._authorize(override_key_id, override_key)
@@ -36,12 +46,17 @@ class B2CleanupTool:
 
         try:
             self.logger.info("🔍 Trying to load credentials via `b2 account get`...")
-            result = subprocess.run(
-                ["b2", "account", "get"],
-                check=True,
-                capture_output=True,
-                text=True,
-            )
+            try:
+                result = subprocess.run(
+                    ["b2", "account", "get"],
+                    check=True,
+                    capture_output=True,
+                    text=True,
+                )
+            except FileNotFoundError:
+                self.logger.error("❌ Command 'b2' not found. Please install the B2 CLI or provide credentials.")
+                raise RuntimeError("B2 CLI not found. Please install it or provide credentials manually.")
+                
             creds = json.loads(result.stdout)
             key_id = creds["applicationKeyId"]
             app_key = creds["applicationKey"]
@@ -56,6 +71,11 @@ class B2CleanupTool:
             raise RuntimeError("Could not authorize with Backblaze B2.")
 
     def cleanup_unfinished_uploads(self, bucket_name: str):
+        """Find and clean up unfinished uploads in the specified bucket.
+
+        Args:
+            bucket_name: Name of the B2 bucket to clean up
+        """
         bucket = self.api.get_bucket_by_name(bucket_name)
         unfinished = list(bucket.list_unfinished_large_files())
         if not unfinished:
@@ -64,43 +84,11 @@ class B2CleanupTool:
 
         self.logger.info("🗃️ Found %d unfinished uploads", len(unfinished))
         for file_version in unfinished:
-            file_id = file_version.id_
+            # Use the correct attribute names for UnfinishedLargeFile objects
+            file_id = file_version.file_id
             file_name = file_version.file_name
             if self.dry_run:
                 self.logger.info(f"💡 Dry run: would cancel {file_id} ({file_name})")
             else:
                 self.logger.info(f"🗑️ Cancelling {file_id} ({file_name})")
-                self.api.cancel_large_file(file_id)
-
-
-@click.command()
-@click.argument("bucket")
-@click.option(
-    "--dry-run", is_flag=True, help="Only list unfinished uploads, do not cancel."
-)
-@click.option("--log-file", default="b2_cleanup.log", help="Path to log file.")
-@click.option("--key-id", help="Backblaze B2 applicationKeyId to override env/config.")
-@click.option("--key", help="Backblaze B2 applicationKey to override env/config.")
-def cli(bucket, dry_run, log_file, key_id, key):
-    """Clean up unfinished large file uploads in BUCKET."""
-    logger = logging.getLogger()
-    logger.setLevel(logging.INFO)
-    if logger.hasHandlers():
-        logger.handlers.clear()
-
-    file_handler = logging.FileHandler(log_file)
-    file_handler.setFormatter(
-        logging.Formatter("%(asctime)s [%(levelname)s] %(message)s")
-    )
-    logger.addHandler(file_handler)
-
-    stream_handler = logging.StreamHandler()
-    stream_handler.setFormatter(logging.Formatter("%(message)s"))
-    logger.addHandler(stream_handler)
-
-    tool = B2CleanupTool(dry_run=dry_run, override_key_id=key_id, override_key=key)
-    tool.cleanup_unfinished_uploads(bucket)
-
-
-if __name__ == "__main__":
-    cli()
+                self.api.cancel_large_file(file_id)

b2_cleanup-0.1.4.dist-info/METADATA

@@ -0,0 +1,196 @@
+Metadata-Version: 2.4
+Name: b2-cleanup
+Version: 0.1.4
+Summary: CLI tool to clean up unfinished Backblaze B2 large uploads
+Project-URL: Homepage, https://github.com/your-username/b2-cleanup
+Project-URL: Issues, https://github.com/your-username/b2-cleanup/issues
+Project-URL: Changelog, https://github.com/your-username/b2-cleanup/blob/main/CHANGELOG.md
+Author-email: Jeroen Verhoeven <jeroen@joentje.org>
+License: MIT
+Keywords: b2,backblaze,cleanup,cli,cloud-storage
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Utilities
+Requires-Python: >=3.8
+Requires-Dist: b2sdk>=1.20.0
+Requires-Dist: click>=8.0
+Provides-Extra: dev
+Requires-Dist: pytest-cov>=4.0.0; extra == 'dev'
+Requires-Dist: pytest-mock>=3.10.0; extra == 'dev'
+Requires-Dist: pytest>=7.0.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# B2 Cleanup
+
+A CLI tool and Python library to clean up unfinished Backblaze B2 large file uploads.
+
+[![PyPI version](https://badge.fury.io/py/b2-cleanup.svg)](https://badge.fury.io/py/b2-cleanup)
+[![Python Versions](https://img.shields.io/pypi/pyversions/b2-cleanup.svg)](https://pypi.org/project/b2-cleanup/)
+
+## 📋 Overview
+
+When uploading large files to Backblaze B2, interrupted uploads can leave behind unfinished file parts that consume storage and incur costs. This tool helps you identify and clean up these unfinished uploads.
+
+---
+
+## 🔧 Features
+
+- Lists all unfinished large file uploads in a given B2 bucket
+- Optionally cancels them (dry-run support included)
+- Uses the official `b2sdk` for native Backblaze API access
+- Supports authentication via env vars, CLI override, or the `b2` CLI
+- Clean CLI with logging support
+- Class-based and easily extensible
+
+---
+
+## 🚀 Installation
+
+```bash
+pip install b2-cleanup
+```
+
+---
+
+## 🧪 Usage
+
+```bash
+# Basic usage (requires B2 CLI to be installed and authorized)
+b2-cleanup your-bucket-name
+```
+
+```bash
+# Use with explicit credentials
+b2-cleanup your-bucket-name --key-id YOUR_KEY_ID --key YOUR_APPLICATION_KEY
+```
+
+```bash
+# Dry run to preview what would be deleted
+b2-cleanup your-bucket-name --dry-run
+```
+
+### Example (dry run):
+
+```bash
+b2-cleanup my-bucket --dry-run
+```
+
+### Example (delete for real, with logging):
+
+```bash
+b2-cleanup my-bucket --log-file cleanup_$(date +%F).log
+```
+
+### Example (override credentials):
+
+```bash
+b2-cleanup my-bucket --key-id my-key-id --key my-app-key
+```
+
+### Example (Python usage):
+
+```python
+from b2_cleanup import B2CleanupTool
+
+# Using environment variables or B2 CLI for auth
+tool = B2CleanupTool(dry_run=True)
+
+# Using explicit credentials
+tool = B2CleanupTool(
+    dry_run=False,
+    override_key_id="your-key-id",
+    override_key="your-application-key"
+)
+
+# Clean up unfinished uploads
+tool.cleanup_unfinished_uploads("your-bucket-name")
+```
+
+---
+
+## 🔐 Authentication
+
+This tool supports three ways to authenticate with B2, in priority order:
+
+1. **Explicit CLI arguments**:
+   ```bash
+   b2-cleanup bucket-name --key-id YOUR_KEY_ID --key YOUR_APPLICATION_KEY
+   ```
+
+2. **Environment variables**:
+   ```bash
+   export B2_APPLICATION_KEY_ID=abc123
+   export B2_APPLICATION_KEY=supersecretkey
+   b2-cleanup bucket-name
+   ```
+
+3. **The `b2` CLI** (must be previously authorized):
+   ```bash
+   b2 account authorize
+   # Then the tool will read credentials via:
+   b2 account get
+   ```
+
+---
+
+## 📁 Project Structure
+
+```
+b2-cleanup/
+├── b2_cleanup/
+│   ├── __init__.py     # Package exports
+│   ├── core.py         # Core functionality 
+│   └── cli.py          # CLI implementation
+├── tests/
+│   ├── __init__.py
+│   ├── test_core.py
+│   └── test_cli.py
+├── pyproject.toml      # Project metadata + dependencies
+├── CHANGELOG.md        # Version history
+├── .gitignore
+└── README.md
+```
+
+---
+
+## 📦 Packaging Notes
+
+- The CLI entry point is `b2-cleanup` via `pyproject.toml`
+- Install in editable mode (`uv pip install -e .`) for fast development
+- Dependencies are managed via [`uv`](https://github.com/astral-sh/uv)
+- Testing dependencies: `uv pip install -e ".[dev]"`
+
+---
+
+## 🧪 Testing
+
+```bash
+# Install dev dependencies
+pip install b2-cleanup[dev]
+
+# Run tests
+pytest
+
+# With coverage
+pytest --cov=b2_cleanup
+```
+
+## 🛠️ Roadmap
+
+- [ ] Filter uploads by file age
+- [ ] Support multiple buckets
+- [ ] Output metrics (count, size, cost saved)
+- [ ] Optional integration with S3-compatible B2 APIs
+
+---
+
+## 📝 License
+
+MIT License © 2025 Jeroen Verhoeven
+

b2_cleanup-0.1.4.dist-info/RECORD

@@ -0,0 +1,7 @@
+b2_cleanup/__init__.py,sha256=2AjGt1V_1mxLhPRvAah_m6T7yWa-TnxdfKZpDoS1pVM,155
+b2_cleanup/cli.py,sha256=gSBPzlP-rDDuiGqXvpgPD8VH_CNFlvel_TzPyHAPafQ,1273
+b2_cleanup/core.py,sha256=usnBc6ugEVOX7w2b8b35jAUTYSwSi2kzZdAPq2JfR4U,3687
+b2_cleanup-0.1.4.dist-info/METADATA,sha256=6k0DcgOv-BzO7o3YT1G5v27clFdPZbC5ZjyFVPGBWlw,4853
+b2_cleanup-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+b2_cleanup-0.1.4.dist-info/entry_points.txt,sha256=zrE0HFNfpIaNCY44wha_k1u4dNAbUt_YrBrRVU2j2_4,50
+b2_cleanup-0.1.4.dist-info/RECORD,,

{b2_cleanup-0.1.0.dist-info → b2_cleanup-0.1.4.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.1)
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any
-

b2_cleanup-0.1.4.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+b2-cleanup = b2_cleanup.cli:cli

b2_cleanup-0.1.0.dist-info/METADATA

@@ -1,7 +0,0 @@
-Metadata-Version: 2.4
-Name: b2-cleanup
-Version: 0.1.0
-Summary: Cleanup unfinished Backblaze B2 large uploads
-Author-email: Jeroen Verhoeven <jeroen@joentje.org>
-Requires-Dist: click>=8.0
-Requires-Dist: b2sdk>=2.8.1

b2_cleanup-0.1.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-cleanup_unfinished_b2_uploads.py,sha256=_wShubS02qNxiRdJmuNSoLzcjd7epcB3Q_fU1CDpH_Q,3877
-b2_cleanup-0.1.0.dist-info/METADATA,sha256=kHW0eHyzVtwFoYT9WNbG5g2HKJ6bESQ7YnXD4wLN1XY,215
-b2_cleanup-0.1.0.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-b2_cleanup-0.1.0.dist-info/entry_points.txt,sha256=hRMR8HC39r-P1TrTPYbH3hDQkBVvDb508tOUS8a0CcI,65
-b2_cleanup-0.1.0.dist-info/top_level.txt,sha256=GmKi68dORi-iv2h2sT5Uf7Q3udlX1mElQQg50cqqF6o,30
-b2_cleanup-0.1.0.dist-info/RECORD,,

b2_cleanup-0.1.0.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-b2-cleanup = cleanup_unfinished_b2_uploads:cli

b2_cleanup-0.1.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-cleanup_unfinished_b2_uploads