gravelrpc: SSL in rpctool

webgravel · Aug 29, 2013 · 2389043 · 2389043
1 parent d27160e
commit 2389043
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/rpctool.py b/rpctool.py
@@ -4,9 +4,15 @@
 from gravelrpc import FD
 
 parser = argparse.ArgumentParser()
+parser.add_argument('--ssl', action='store_true')
+parser.add_argument('--ssl-key', default='example.pem')
 parser.add_argument('name')
 parser.add_argument('expression')
 args = parser.parse_args()
 
-client = gravelrpc.Client(args.name)
+if args.ssl:
+    host, port = args.name.split(':')
+    client = gravelrpc.SSLClient((host, int(port)), args.ssl_key)
+else:
+    client = gravelrpc.Client(args.name)
 print eval('client.' + args.expression)