X-Git-Url: https://git.phdru.name/?p=dotfiles.git;a=blobdiff_plain;f=bin%2Fidna.py;h=44843ac223fa4932ba872d64b6eea889fbb2b352;hp=fd2c8e5984d2783554863a93a646f5efe8079241;hb=HEAD;hpb=f0344023aa21ede3b47d77559af54b97c37f328b

diff --git a/bin/idna.py b/bin/idna.py
index fd2c8e5..44843ac 100755
--- a/bin/idna.py
+++ b/bin/idna.py
@@ -1,6 +1,8 @@
-#! /usr/bin/env python
+#! /usr/bin/env python3
 
 import sys
+from m_lib.defenc import default_encoding
+
 l = len(sys.argv)
 
 if (l == 2) and (sys.argv[1] != '-r'):
@@ -12,9 +14,11 @@ elif (l == 3) and (sys.argv[1] == '-r'):
 else:
     sys.exit('Usage: %s [-r] name.domain' % sys.argv[0])
 
-from m_lib.defenc import default_encoding
-
 if reverse:
-    print(address.decode("idna").encode(default_encoding))
+    if not isinstance(address, bytes):
+        address = address.encode('ascii')
+    print(address.decode("idna"))
 else:
-    print(address.decode(default_encoding).encode("idna"))
+    if isinstance(address, bytes):
+        address = address.decode(default_encoding)
+    print(address.encode("idna").decode('ascii'))