From 5d7c724bc8191b1bf7bbf9b689852871426fd340 Mon Sep 17 00:00:00 2001 From: LordBaryhobal Date: Fri, 5 Jun 2026 10:44:20 +0200 Subject: [PATCH] fix(cli): add types files argument --- midas/cli/main.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/midas/cli/main.py b/midas/cli/main.py index a9833bb..07dfd07 100644 --- a/midas/cli/main.py +++ b/midas/cli/main.py @@ -34,8 +34,9 @@ def midas(): @midas.command() @click.option("-l", "--highlight", type=click.File("w")) +@click.option("-t", "--types", type=click.File("r"), multiple=True) @click.argument("file", type=click.File("r")) -def compile(highlight: Optional[TextIO], file: TextIO): +def compile(highlight: Optional[TextIO], file: TextIO, types: tuple[TextIO]): logging.basicConfig(level=logging.DEBUG) source: str = file.read() tree: ast.Module = ast.parse(source, filename=file.name) @@ -43,7 +44,12 @@ def compile(highlight: Optional[TextIO], file: TextIO): stmts: list[p.Stmt] = parser.parse_module(tree) resolver = Resolver() resolver.resolve(*stmts) - checker = Checker(resolver.locals, file_path=Path(file.name).resolve()) + types_paths: list[Path] = [Path(t.name).resolve() for t in types] + checker = Checker( + resolver.locals, + source_path=Path(file.name).resolve(), + types_paths=types_paths, + ) diagnostics: list[Diagnostic] = checker.check(stmts) for diagnostic in diagnostics: print(diagnostic)