sync, truncate: Add --progress/--no-progress support

Expertcoderz · Expertcoderz · commit 196a476174ca · 2025-04-10T01:07:19.000Z
Progress bars for coreutils are an extension specific to
python-userland.
diff --git a/pyproject.toml b/pyproject.toml
@@ -6,6 +6,9 @@ build-backend = "setuptools.build_meta"
 name = "userland"
 version = "0.0.2"
 requires-python = ">=3.13"
+dependencies = [
+	"tqdm>=4.67.1"
+]
 authors = [
 	{ name="Expertcoderz", email="expertcoderzx@gmail.com"}
 ]
diff --git a/userland/utilities/sync.py b/userland/utilities/sync.py
@@ -1,21 +1,39 @@
 import os
 import sys
 
+from tqdm import tqdm
+
 from .. import core
 
 
 parser = core.create_parser(
-    usage=("%prog [FILE]...",),
+    usage=("%prog [OPTION] [FILE]...",),
     description="Sync the filesystem or write each FILE's blocks to disk.",
 )
 
+parser.add_option(
+    "--progress",
+    dest="progress",
+    action="store_true",
+    help="show a progress bar when syncing files",
+)
+parser.add_option(
+    "--no-progress",
+    dest="progress",
+    action="store_false",
+    default=False,
+    help="do not show a progress bar (default)",
+)
+
 
 @core.command(parser)
-def python_userland_sync(_, args):
+def python_userland_sync(opts, args):
     if args:
         failed = False
 
-        for name in args:
+        for name in (
+            tqdm(args, ascii=True, desc="Syncing files") if opts.progress else args
+        ):
             try:
                 with open(name, "rb+") as io:
                     os.fsync(io)
diff --git a/userland/utilities/truncate.py b/userland/utilities/truncate.py
@@ -2,6 +2,8 @@
 from pathlib import Path
 from typing import Callable
 
+from tqdm import tqdm
+
 from .. import core
 
 
@@ -13,6 +15,20 @@
     description="Shrink or extend each FILE to SIZE.",
 )
 
+parser.add_option(
+    "--progress",
+    dest="progress",
+    action="store_true",
+    help="show a progress bar when truncating files",
+)
+parser.add_option(
+    "--no-progress",
+    dest="progress",
+    action="store_false",
+    default=False,
+    help="do not show a progress bar (default)",
+)
+
 parser.add_option("-c", "--no-create", action="store_true", help="do not create files")
 parser.add_option("-s", "--size", help="set or adjust file size by SIZE bytes")
 parser.add_option(
@@ -79,7 +95,9 @@ def python_userland_truncate(opts, args):
         print(e, file=sys.stderr)
         return 1
 
-    for file in map(Path, args):
+    for file in map(
+        Path, tqdm(args, ascii=True, desc="Truncating files") if opts.progress else args
+    ):
         if not file.exists() and opts.no_create:
             continue
 
