b2-cleanup 0.1.0__py3-none-any.whl

b2_cleanup-0.1.0.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,6 @@
+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/WHEEL

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

b2_cleanup-0.1.0.dist-info/entry_points.txt

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

b2_cleanup-0.1.0.dist-info/top_level.txt

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

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()