b2-cleanup 0.1.0__tar.gz

b2_cleanup-0.1.0/PKG-INFO

@@ -0,0 +1,7 @@
+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/README.md

@@ -0,0 +1,99 @@
+# b2-cleanup
+
+🧹 A Python CLI tool to clean up **unfinished large file uploads** in a [Backblaze B2](https://www.backblaze.com/b2/cloud-storage.html) bucket.
+
+Built using [`b2sdk`](https://github.com/Backblaze/b2-sdk-python), [`click`](https://click.palletsprojects.com/), and [`uv`](https://github.com/astral-sh/uv) for performance and reproducibility.
+
+---
+
+## 🔧 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
+- Clean CLI with logging support
+- Class-based and easily extensible
+
+---
+
+## 🚀 Installation
+
+### 1. Clone and create an isolated environment
+
+```bash
+git clone https://github.com/<your-username>/b2-cleanup.git
+cd b2-cleanup
+
+uv venv
+source .venv/bin/activate
+uv pip install -e .
+```
+
+> Requires [uv](https://github.com/astral-sh/uv) and Python 3.8+
+
+---
+
+## 🧪 Usage
+
+```bash
+b2-cleanup BUCKET_NAME [OPTIONS]
+```
+
+### 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
+```
+
+---
+
+## 🔐 Authentication
+
+You must be logged in with the `b2` CLI at least once:
+
+```bash
+b2 authorize-account
+```
+
+This stores credentials in `~/.b2_account_info`, which the tool reuses.
+
+---
+
+## 📁 Project Structure
+
+```
+b2-cleanup/
+├── cleanup_unfinished_b2_uploads.py   # Core CLI logic (class-based)
+├── pyproject.toml                     # Project metadata + dependencies
+├── .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) for reproducibility and speed
+
+---
+
+## 🛠️ 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 Your Name

b2_cleanup-0.1.0/b2_cleanup.egg-info/PKG-INFO

@@ -0,0 +1,7 @@
+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/b2_cleanup.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+README.md
+cleanup_unfinished_b2_uploads.py
+pyproject.toml
+b2_cleanup.egg-info/PKG-INFO
+b2_cleanup.egg-info/SOURCES.txt
+b2_cleanup.egg-info/dependency_links.txt
+b2_cleanup.egg-info/entry_points.txt
+b2_cleanup.egg-info/requires.txt
+b2_cleanup.egg-info/top_level.txt

b2_cleanup-0.1.0/b2_cleanup.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

b2_cleanup-0.1.0/b2_cleanup.egg-info/entry_points.txt

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

b2_cleanup-0.1.0/b2_cleanup.egg-info/requires.txt

@@ -0,0 +1,2 @@
+click>=8.0
+b2sdk>=2.8.1

b2_cleanup-0.1.0/b2_cleanup.egg-info/top_level.txt

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

b2_cleanup-0.1.0/cleanup_unfinished_b2_uploads.py

@@ -0,0 +1,106 @@
+import os
+import json
+import subprocess
+import logging
+import click
+from b2sdk.v2 import InMemoryAccountInfo, B2Api
+
+
+class B2CleanupTool:
+    def __init__(
+        self,
+        dry_run: bool = False,
+        override_key_id: str = None,
+        override_key: str = None,
+    ):
+        self.dry_run = dry_run
+        self.logger = logging.getLogger("B2Cleanup")
+        self.api = self._authorize(override_key_id, override_key)
+
+    def _authorize(self, override_key_id=None, override_key=None):
+        info = InMemoryAccountInfo()
+        api = B2Api(info)
+
+        if override_key_id and override_key:
+            self.logger.info("🔐 Using credentials from CLI override.")
+            api.authorize_account("production", override_key_id, override_key)
+            return api
+
+        key_id = os.getenv("B2_APPLICATION_KEY_ID")
+        app_key = os.getenv("B2_APPLICATION_KEY")
+
+        if key_id and app_key:
+            self.logger.info("🔐 Using credentials from environment variables.")
+            api.authorize_account("production", key_id, app_key)
+            return api
+
+        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,
+            )
+            creds = json.loads(result.stdout)
+            key_id = creds["applicationKeyId"]
+            app_key = creds["applicationKey"]
+            api.authorize_account("production", key_id, app_key)
+            self.logger.info("✅ Authorized with B2 CLI credentials.")
+            return api
+
+        except (subprocess.CalledProcessError, KeyError, json.JSONDecodeError) as e:
+            self.logger.error(
+                "❌ Failed to get B2 credentials from CLI or environment: %s", e
+            )
+            raise RuntimeError("Could not authorize with Backblaze B2.")
+
+    def cleanup_unfinished_uploads(self, bucket_name: str):
+        bucket = self.api.get_bucket_by_name(bucket_name)
+        unfinished = list(bucket.list_unfinished_large_files())
+        if not unfinished:
+            self.logger.info("✅ No unfinished large files found.")
+            return
+
+        self.logger.info("🗃️ Found %d unfinished uploads", len(unfinished))
+        for file_version in unfinished:
+            file_id = file_version.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()

b2_cleanup-0.1.0/pyproject.toml

@@ -0,0 +1,15 @@
+[project]
+name = "b2-cleanup"
+version = "0.1.0"
+description = "Cleanup unfinished Backblaze B2 large uploads"
+authors = [
+    { name = "Jeroen Verhoeven", email = "jeroen@joentje.org" }
+]
+dependencies = [
+    "click>=8.0",
+    "b2sdk>=2.8.1"
+]
+
+[project.scripts]
+b2-cleanup = "cleanup_unfinished_b2_uploads:cli"
+

b2_cleanup-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+