7
7
import functools
8
8
import os
9
9
import typing
10
- from pathlib import Path
11
10
from typing import Any , Callable
12
11
13
12
import click
25
24
from click_default_group import DefaultGroup
26
25
27
26
28
- def _existing_path_before_colon (_ctx : object , _value : object , path : str ) -> str :
29
- prefix , _ , _ = path .partition (":" )
30
- if not Path (prefix ).is_file ():
31
- msg = "{prefix!r} must be an exiting path"
32
- raise click .BadParameter (msg )
33
-
34
- return path
35
-
36
-
37
27
@click .group (context_settings = CONTEXT_SETTINGS , cls = DefaultGroup , default = "browse" )
38
28
@click .version_option (version = VERSION )
39
29
def main () -> None :
@@ -43,7 +33,7 @@ def main() -> None:
43
33
44
34
45
35
@main .command ()
46
- @click .argument ("filename" , callback = _existing_path_before_colon )
36
+ @click .argument ("filename" )
47
37
def tree (filename : str ) -> None :
48
38
"""
49
39
Display a tree.
@@ -68,7 +58,7 @@ def new_func(*args: Any, **kwargs: Any) -> Any:
68
58
69
59
70
60
@main .command ()
71
- @click .argument ("filename" , callback = _existing_path_before_colon )
61
+ @click .argument ("filename" )
72
62
@click .option (
73
63
"--iterm" , is_flag = True , help = "Display an iTerm plot (requires [iterm] extra)."
74
64
)
@@ -87,10 +77,7 @@ def plot(filename: str, iterm: bool) -> None:
87
77
88
78
import uproot_browser .dirs # pylint: disable=import-outside-toplevel
89
79
90
- fname = uproot_browser .dirs .filename (filename )
91
- selections = uproot_browser .dirs .selections (filename )
92
- my_tree = uproot .open (fname )
93
- * _ , item = uproot_browser .dirs .apply_selection (my_tree , selections )
80
+ item = uproot .open (filename )
94
81
95
82
if iterm :
96
83
uproot_browser .plot_mpl .plot (item )
@@ -109,18 +96,15 @@ def plot(filename: str, iterm: bool) -> None:
109
96
110
97
111
98
@main .command ()
112
- @click .argument ("filename" , callback = _existing_path_before_colon )
99
+ @click .argument ("filename" )
113
100
def browse (filename : str ) -> None :
114
101
"""
115
102
Display a TUI.
116
103
"""
117
- import uproot_browser .dirs # pylint: disable=import-outside-toplevel
118
104
import uproot_browser .tui .browser # pylint: disable=import-outside-toplevel
119
105
120
- fname = uproot_browser .dirs .filename (filename )
121
-
122
106
app = uproot_browser .tui .browser .Browser (
123
- path = Path ( fname ) ,
107
+ path = filename ,
124
108
)
125
109
126
110
app .run ()
0 commit comments