fix: arguments can take a colon (#170)

Signed-off-by: Henry Schreiner <henryschreineriii@gmail.com>

Author: henryiii

pyproject.toml

@@ -158,6 +158,7 @@ ignore = [
     "E722",
     "RUF001",  # Unicode chars
     "PLR",
+    "ISC001",  # Conflicts with formatter
 ]
 typing-modules = ["uproot_browser._compat.typing"]
 unfixable = [

src/uproot_browser/__main__.py

@@ -25,6 +25,15 @@
     from click_default_group import DefaultGroup
 
 
+def _existing_path_before_colon(_ctx: object, _value: object, path: str) -> str:
+    prefix, _, _ = path.partition(":")
+    if not Path(prefix).is_file():
+        msg = "{prefix!r} must be an exiting path"
+        raise click.BadParameter(msg)
+
+    return path
+
+
 @click.group(context_settings=CONTEXT_SETTINGS, cls=DefaultGroup, default="browse")
 @click.version_option(version=VERSION)
 def main() -> None:
@@ -34,7 +43,7 @@ def main() -> None:
 
 
 @main.command()
-@click.argument("filename", type=click.Path(exists=True))
+@click.argument("filename", callback=_existing_path_before_colon)
 def tree(filename: str) -> None:
     """
     Display a tree.
@@ -59,7 +68,7 @@ def new_func(*args: Any, **kwargs: Any) -> Any:
 
 
 @main.command()
-@click.argument("filename", type=click.Path(exists=True))
+@click.argument("filename", callback=_existing_path_before_colon)
 @click.option(
     "--iterm", is_flag=True, help="Display an iTerm plot (requires [iterm] extra)."
 )
@@ -100,7 +109,7 @@ def plot(filename: str, iterm: bool) -> None:
 
 
 @main.command()
-@click.argument("filename", type=click.Path(exists=True))
+@click.argument("filename", callback=_existing_path_before_colon)
 def browse(filename: str) -> None:
     """
     Display a TUI.