use args for python test script
This commit is contained in:
parent
2e474d8e2a
commit
56a74f9121
1 changed files with 35 additions and 10 deletions
|
@ -1,11 +1,36 @@
|
||||||
import socket
|
import socket
|
||||||
|
import sys
|
||||||
sock = socket.socket()
|
import getopt
|
||||||
|
import argparse
|
||||||
host = "10.93.0.246"
|
|
||||||
port = 80
|
|
||||||
|
|
||||||
sock.connect((host, port))
|
|
||||||
sock.send("test".encode())
|
|
||||||
sock.close()
|
|
||||||
|
|
||||||
|
|
||||||
|
def main(argv):
|
||||||
|
parser = argparse.ArgumentParser(
|
||||||
|
description='Send socket message to ip and port.')
|
||||||
|
parser.add_argument('-i', '--ip',
|
||||||
|
default='localhost',
|
||||||
|
dest='ip',
|
||||||
|
help='Provide destination ip. Defaults to localhost',
|
||||||
|
type=str
|
||||||
|
)
|
||||||
|
parser.add_argument('-p', '--port',
|
||||||
|
default=80,
|
||||||
|
dest='port',
|
||||||
|
help='Provide destination port. Defaults to 80',
|
||||||
|
type=int
|
||||||
|
)
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
print(f'Sending message to {args.ip}:{args.port}')
|
||||||
|
sock = socket.socket()
|
||||||
|
try:
|
||||||
|
sock.connect((args.ip, args.port))
|
||||||
|
sock.send("test".encode())
|
||||||
|
print("Message sent.")
|
||||||
|
sock.close()
|
||||||
|
except:
|
||||||
|
print("Cannot send message...")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main(sys.argv[1:])
|
||||||
|
|
Reference in a new issue