Allow '-f -' to denote stdin

2017-11-01 21:34:56 +00:00 · 2017-11-01 21:34:56 +00:00 · f1807d34ea
commit f1807d34ea
parent 9caa006e5b
1 changed files with 7 additions and 2 deletions
--- a/dnseval.py
+++ b/dnseval.py
@ -256,8 +256,13 @@ def main():

    try:
        if fromfile:
-            with open(inputfilename, 'rt') as flist:
-                f = flist.read().splitlines()
+            if inputfilename == '-':
+                # read from stdin
+                with sys.stdin as flist:
+                    f = flist.read().splitlines()
+            else:
+                with open(inputfilename, 'rt') as flist:
+                    f = flist.read().splitlines()
        else:
            f = resolvers
        if len(f) == 0: