mirrors/tplink-smartplug
2
0
Fork 0
mirror of https://github.com/softScheck/tplink-smartplug synced 2026-01-11 23:38:46 +01:00
Code Issues Releases Wiki Activity

Merge pull request #72 from fuegas/master

Browse Source 
Add timeout argument
This commit is contained in:
softScheck GmbH
2020-06-04 17:17:20 +02:00
committed by GitHub
parent 081280d46d 58d8959fc2
commit dcf978b970
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tplink_smartplug.py
View File

@@ -112,6 +112,7 @@ parser = argparse.ArgumentParser(description="TP-Link Wi-Fi Smart Plug Client v"
parser.add_argument("-t", "--target", metavar="<hostname>", required=True, help="Target hostname or IP address", type=validHostname) 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) parser.add_argument("-p", "--port", metavar="<port>", default=9999, required=False, help="Target port", type=validPort)
parser.add_argument("-q", "--quiet", dest='quiet', action='store_true', help="Only show result") parser.add_argument("-q", "--quiet", dest='quiet', action='store_true', help="Only show result")
parser.add_argument("--timeout", default=10, required=False, help="Timeout to establish connection")
group = parser.add_mutually_exclusive_group(required=True) 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("-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") group.add_argument("-j", "--json", metavar="<JSON string>", help="Full JSON string of command to send")
@@ -131,7 +132,9 @@ else:
# Send command and receive reply # Send command and receive reply
try: try:
sock_tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock_tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock_tcp.settimeout(int(args.timeout))
sock_tcp.connect((ip, port)) sock_tcp.connect((ip, port))
sock_tcp.settimeout(None)
sock_tcp.send(encrypt(cmd)) sock_tcp.send(encrypt(cmd))
data = sock_tcp.recv(2048) data = sock_tcp.recv(2048)
sock_tcp.close() sock_tcp.close()
@@ -145,4 +148,4 @@ try:
print("Received: ", decrypted) print("Received: ", decrypted)
except socket.error: except socket.error:
quit("Cound not connect to host " + ip + ":" + str(port)) quit("Could not connect to host " + ip + ":" + str(port))