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
|
||||
|
||||
sock = socket.socket()
|
||||
|
||||
host = "10.93.0.246"
|
||||
port = 80
|
||||
|
||||
sock.connect((host, port))
|
||||
sock.send("test".encode())
|
||||
sock.close()
|
||||
import socket
|
||||
import sys
|
||||
import getopt
|
||||
import argparse
|
||||
|
||||
|
||||
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