use args for python test script

This commit is contained in:
Florian Hoss 2022-06-12 14:21:18 +02:00
parent 2e474d8e2a
commit 56a74f9121

View file

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