diff --git a/tplink_smartplug.py b/tplink_smartplug.py index 4012a55..c1e8734 100755 --- a/tplink_smartplug.py +++ b/tplink_smartplug.py @@ -72,6 +72,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="", required=True, help="Target hostname or IP address", type=validHostname) +parser.add_argument("-v", "--verbose", dest='verbose', action='store_true', help="Display verbose output") group = parser.add_mutually_exclusive_group(required=True) group.add_argument("-c", "--command", metavar="", help="Preset command to send. Choices are: "+", ".join(commands), choices=commands) group.add_argument("-j", "--json", metavar="", help="Full JSON string of command to send") @@ -96,8 +97,14 @@ try: data = sock_tcp.recv(2048) sock_tcp.close() - print "Sent: ", cmd - print "Received: ", decrypt(data[4:]) + decrypted = decrypt(data[4:]) + + if args.verbose: + print "Sent: ", cmd + print "Received: ", decrypted + else: + print decrypted + except socket.error: quit("Cound not connect to host " + ip + ":" + str(port))