also enable reading dk-handle (userid) from stdin

[public/dnssec-tools.git] / dsu_upload

diff --git a/dsu_upload b/dsu_upload
index 778d145482dde04400b789ea203e771f041d6218..b6b271b58a8e02073b86a8d77a3c6bdcb9f9d89c 100755 (executable)
--- a/dsu_upload
+++ b/dsu_upload
@@ -15,13 +15,20 @@ import sys
 import urllib.request
 import urllib.parse
 
-if len(sys.argv) != 3: 
-       print("dsu_upload <dkhm-handle> <ds-rr-file>");
+postdata = []
+if len(sys.argv) == 3: 
+    postdata.append(("userid",sys.argv[1].upper()))
+    keyfile = sys.argv[2]
+elif len(sys.argv) == 2:
+    print ("Username:"),
+    userid = sys.stdin.readline().strip()
+    print ("Username entered was " + userid)
+    postdata.append(("userid",userid))
+    keyfile = sys.argv[1]
+else:
+       print("dsu_upload [dkhm-handle] <ds-rr-file>");
        print("be ready to provide the password on stdin (best piped from gpg :)")
        sys.exit();
-postdata = []
-postdata.append(("userid",sys.argv[1].upper()))
-keyfile = sys.argv[2]
 
 print ("Password:"),
 password = sys.stdin.readline().strip()