Properly decode names from command-line

3 files changed, 9 insertions, 3 deletions

diff --git a/bin/dnsget b/bin/dnsget
index 401a7e6..7366caf 100755
--- a/bin/dnsget
+++ b/bin/dnsget
@@ -597,7 +597,9 @@ def main():
                     names.append(name)
             f.close()
         else:
-            if not args:
+            if args:
+                args = map(lambda x: x.decode(sys.getfilesystemencoding()), args)
+            else:
                 try:
                     args = analysis_structured['_meta._dnsviz.']['names']
                 except KeyError:
diff --git a/bin/dnsgrok b/bin/dnsgrok
index 3edcc85..8e4edd1 100755
--- a/bin/dnsgrok
+++ b/bin/dnsgrok
@@ -163,7 +163,9 @@ def main():
                     names.append(name)
             f.close()
         else:
-            if not args:
+            if args:
+                args = map(lambda x: x.decode(sys.getfilesystemencoding()), args)
+            else:
                 try:
                     args = analysis_structured['_meta._dnsviz.']['names']
                 except KeyError:
diff --git a/bin/dnsviz b/bin/dnsviz
index f3b058e..ce6db85 100755
--- a/bin/dnsviz
+++ b/bin/dnsviz
@@ -396,7 +396,9 @@ def main():
                     names.append(name)
             f.close()
         else:
-            if not args:
+            if args:
+                args = map(lambda x: x.decode(sys.getfilesystemencoding()), args)
+            else:
                 try:
                     args = analysis_structured['_meta._dnsviz.']['names']
                 except KeyError: