Merge pull request #35 from nschloe/more-input-files

More input files

Author: nschloe

README.md

@@ -18,7 +18,7 @@ comments from a given file and corrects [some common
 anti-patterns](http://mirrors.ctan.org/info/l2tabu/english/l2tabuen.pdf). For example,
 with
 ```
-blacktex in.tex out.tex
+blacktex in.tex > out.tex
 ```
 the input file
 ```latex
@@ -37,9 +37,9 @@ and \(y = 2^n g\) with \(n = 1,\dots,10\), we have \(\frac{\Gamma}{2} = 8\).
 ```
 You can use
 ```
-blacktex -i in.tex
+blacktex -i in0.tex in1.tex ...
 ```
-to modify a file in-place. See `blacktex -h` for all options.
+to modify files in-place. See `blacktex -h` for all options.
 
 ### Installation
 

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = blacktex
-version = 0.4.0
+version = 0.5.0
 author = Nico Schlömer
 author_email = [email protected]
 description = Cleans up your LaTeX files

src/blacktex/cli.py

@@ -8,36 +8,32 @@ def main(argv=None):
     parser = _get_parser()
     args = parser.parse_args(argv)
 
-    out = blacktex.clean(args.infile.read(), args.keep_comments, args.keep_dollar_math)
+    stdout = []
+    for fl in args.infiles:
+        content = fl.read()
+        out = blacktex.clean(content, args.keep_comments, args.keep_dollar_math)
+        if args.in_place:
+            with open(fl.name, "w") as f:
+                f.write(out)
+        else:
+            stdout.append(out)
 
-    if args.in_place:
-        with open(args.infile.name, "w") as f:
-            f.write(out)
-    else:
-        args.outfile.write(out)
+    if not args.in_place:
+        return "\n".join(stdout)
 
 
 def _get_parser():
     parser = argparse.ArgumentParser(description="Clean up LaTeX files.")
 
     parser.add_argument(
-        "infile",
-        nargs="?",
+        "infiles",
+        nargs="+",
         type=argparse.FileType("r"),
-        default=sys.stdin,
-        help="input LaTeX file (default: stdin)",
+        help="input LaTeX file",
     )
 
     parser.add_argument(
-        "outfile",
-        nargs="?",
-        type=argparse.FileType("w"),
-        default=sys.stdout,
-        help="output LaTeX file (default: stdout)",
-    )
-
-    parser.add_argument(
-        "-i", "--in-place", action="store_true", help="modify infile in place"
+        "-i", "--in-place", action="store_true", help="modify all files in place"
     )
 
     parser.add_argument(

test/test_cli.py

@@ -8,12 +8,9 @@ def test_cli():
     with tempfile.TemporaryDirectory() as tmpdir:
         tmpdir = Path(tmpdir)
         infile = tmpdir / "in.tex"
-        outfile = tmpdir / "out.tex"
         with open(infile, "w") as f:
             f.write("a+b=c")
 
-        blacktex.cli.main([str(infile), str(outfile)])
+        stdout = blacktex.cli.main([str(infile)])
 
-        with open(outfile) as f:
-            line = f.read()
-            assert line == "a+b = c"
+        assert stdout == "a+b = c"