From da4835cb64c4eb2463dd91b0a240528cd6336f27 Mon Sep 17 00:00:00 2001
From: Thierry Laporte <37160474+rirethy@users.noreply.github.com>
Date: Thu, 27 Sep 2018 11:36:07 +0200
Subject: [PATCH] Add "port" parameter

Add management for a new parameter : "port".
Allows to modify the defalt port, and not use the default one : 9999

This is useful to send a command to a plug that is not on the local network
---
 tplink_smartplug.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/tplink_smartplug.py b/tplink_smartplug.py
index 4012a55..131ce0f 100755
--- a/tplink_smartplug.py
+++ b/tplink_smartplug.py
@@ -33,6 +33,19 @@ def validHostname(hostname):
 		parser.error("Invalid hostname.")
 	return hostname
 
+# Check if port is valid
+def validPort(port):
+        try:
+            port = int(port)
+        except ValueError:
+            parser.error("Invalid port number.")
+
+        if ((port <= 1024) or (port >65535)) :
+            parser.error("Invalid port number.")
+
+        return port
+
+
 # Predefined Smart Plug Commands
 # For a full list of commands, consult tplink_commands.txt
 commands = {'info'     : '{"system":{"get_sysinfo":{}}}',
@@ -72,6 +85,7 @@ def decrypt(string):
 # Parse commandline arguments
 parser = argparse.ArgumentParser(description="TP-Link Wi-Fi Smart Plug Client v" + str(version))
 parser.add_argument("-t", "--target", metavar="<hostname>", required=True, help="Target hostname or IP address", type=validHostname)
+parser.add_argument("-p", "--port", metavar="<port>", default=9999, required=False, help="Target port", type=validPort)
 group = parser.add_mutually_exclusive_group(required=True)
 group.add_argument("-c", "--command", metavar="<command>", help="Preset command to send. Choices are: "+", ".join(commands), choices=commands)
 group.add_argument("-j", "--json", metavar="<JSON string>", help="Full JSON string of command to send")
@@ -80,7 +94,7 @@ args = parser.parse_args()
 
 # Set target IP, port and command to send
 ip = args.target
-port = 9999
+port = args.port
 if args.command is None:
 	cmd = args.json
 else: