Merge pull request #1 from 42B/choices

Use choices keyword argument to restrict `TYPE` to server or client
2018-10-24 19:57:23 +00:00 · 2018-10-24 19:57:23 +00:00 · 47d1a88d96
parent d5112ea739 6d95d129a9
commit 47d1a88d96
1 changed files with 1 additions and 1 deletions
--- a/i2plib/tunnel.py
+++ b/i2plib/tunnel.py
@ -151,7 +151,7 @@ class ServerTunnel(I2PTunnel):

 if __name__ == '__main__':
    parser = argparse.ArgumentParser()
-    parser.add_argument('type', metavar="TYPE", 
+    parser.add_argument('type', metavar="TYPE", choices=('server', 'client'),
                        help="Tunnel type (server or client)")
    parser.add_argument('address', metavar="ADDRESS", 
                        help="Local address (e.g. 127.0.0.1:8000)")