From 67bc60b3ad8fe74c7c07b6495d8e2749622e13e1 Mon Sep 17 00:00:00 2001
From: Adam Ladachowski <adam.ladachowski@gmail.com>
Date: Sat, 24 Jan 2026 19:28:40 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=92=AC=20Commit=20message:=20Update=20202?=
 =?UTF-8?q?6-01-24=2019:28:40,=201=20files,=2026=20lines?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

📁 Files changed: 1
📝 Lines changed: 26

  • sft_get.py
---
 sft_get.py | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/sft_get.py b/sft_get.py
index 0f60bc5..79c472c 100644
--- a/sft_get.py
+++ b/sft_get.py
@@ -240,13 +240,13 @@ def save_metadata(
     sha256_hash: str,
     local_metadata: dict[str, Any],
     civitai_data: dict[str, Any] | None,
+    output_dir: Path,
 ) -> tuple[Path, Path]:
-    """Save metadata JSON and SHA256 hash to files alongside the model."""
+    """Save metadata JSON and SHA256 hash to the specified output directory."""
     base_name = get_base_name(file_path)
-    parent = file_path.parent
 
     # Save JSON metadata
-    json_path = parent / f"{base_name}-xm.json"
+    json_path = output_dir / f"{base_name}-xm.json"
     output = {
         "file": str(file_path),
         "sha256": sha256_hash,
@@ -258,7 +258,7 @@ def save_metadata(
     json_path.write_text(json.dumps(output, indent=2))
 
     # Save SHA256 hash
-    sha_path = parent / f"{base_name}-xm.sha256"
+    sha_path = output_dir / f"{base_name}-xm.sha256"
     sha_path.write_text(f"{sha256_hash}  {file_path.name}\n")
 
     return json_path, sha_path
@@ -293,9 +293,10 @@ def main() -> int:
         help="Output results as JSON",
     )
     parser.add_argument(
-        "--save",
-        action="store_true",
-        help="Save metadata JSON and SHA256 hash alongside the model file",
+        "--save-to",
+        type=Path,
+        metavar="DIR",
+        help="Save metadata JSON and SHA256 hash to the specified directory",
     )
 
     args = parser.parse_args()
@@ -336,9 +337,16 @@ def main() -> int:
             display_results(file_path, local_metadata, sha256_hash, civitai_data)
 
         # Save files if requested
-        if args.save:
+        if args.save_to:
+            output_dir: Path = args.save_to.resolve()
+            if not output_dir.exists():
+                console.print(f"[red]Error: Output directory not found: {output_dir}[/red]")
+                return 1
+            if not output_dir.is_dir():
+                console.print(f"[red]Error: Not a directory: {output_dir}[/red]")
+                return 1
             json_path, sha_path = save_metadata(
-                file_path, sha256_hash, local_metadata, civitai_data
+                file_path, sha256_hash, local_metadata, civitai_data, output_dir
             )
             console.print()
             console.print(f"[green]Saved:[/green] {json_path}")