Improve CLI typing

nickw444 · Dec 24, 2018 · 91071b5 · 91071b5
1 parent 4f9359f
commit 91071b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nessclient/cli/send_command.py b/nessclient/cli/send_command.py
@@ -7,9 +7,9 @@
 
 @click.command(help="Send a command")
 @click.option('--host', required=True)
-@click.option('--port', required=True)
+@click.option('--port', required=True, type=int)
 @click.argument('command')
-def send_command(host, port, command):
+def send_command(host: str, port: int, command: str):
     loop = asyncio.get_event_loop()
     client = Client(host=host, port=port, loop=loop)