Skip to content

Commit 30bd6cc

Browse files
committed
ENH: support multiple filenames on CLI
1 parent 80c530d commit 30bd6cc

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

src/latform/main.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def process_file(
9595

9696

9797
def main(
98-
filename: str | pathlib.Path,
98+
filename: pathlib.Path | str,
9999
verbose: int = 0,
100100
line_length: int = 100,
101101
max_line_length: int | None = 0,
@@ -125,8 +125,7 @@ def main(
125125

126126
if verbose >= 4:
127127
output_mod.LATFORM_OUTPUT_DEBUG = True
128-
logger = logging.getLogger("latform")
129-
logger.setLevel("DEBUG")
128+
logging.getLogger("latform").setLevel("DEBUG")
130129

131130
renames = load_renames(rename_file, raw_renames, renames)
132131

@@ -220,6 +219,7 @@ def _build_argparser() -> argparse.ArgumentParser:
220219
parser.add_argument(
221220
"filename",
222221
help="Filename to parse (use '-' for stdin/standard input)",
222+
nargs="+",
223223
)
224224

225225
parser.add_argument(
@@ -366,10 +366,15 @@ def cli_main(args: list[str] | None = None) -> None:
366366
log_level = kwargs.pop("log_level")
367367

368368
# Adjust the package-level logger level as requested:
369-
logger = logging.getLogger("latform")
370-
logger.setLevel(log_level)
369+
logging.getLogger("latform").setLevel(log_level)
371370
logging.basicConfig()
372-
return main(**kwargs)
371+
372+
filenames = kwargs.pop("filename")
373+
for filename in filenames:
374+
if len(filename) > 1:
375+
logger.info("Processing %s", filename)
376+
main(filename=filename, **kwargs)
377+
return
373378

374379

375380
if __name__ == "__main__":

0 commit comments

Comments
 (0)