Test a TCP socket connection in Python

test_tcp_socket.py

import socket
import sys
"""
Usage: python3 test_tcp_socket.py [host [port]]
Where host can be a string or IPv4 address. Port defaults to 80.
"""
host = sys.argv[1] if len(sys.argv)>1 and sys.argv[1] \
        else "example.com"
port = int(sys.argv[2]) if len(sys.argv)>2 and sys.argv[2] else 80
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
  s.settimeout(3)
  s.connect((host, port))
  print(f"Successfully connected to {host}:{port}")
except Exception as e:
  print(f"Failed! {e}")
s.close()