port-scanner.py

import argparse
import socket
import sys

from tqdm import trange


def main(argv):
    parser = argparse.ArgumentParser(description="A port scanner")
    parser.add_argument("host", type=str)
    parser.add_argument("-t", "--timeout", type=float, default=0.1)

    args = parser.parse_args(argv[1:])

    open_ports = []

    for port in trange(0, 65536):
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.settimeout(args.timeout)
        if sock.connect_ex((args.host, port)) == 0:
            open_ports.append(port)
        sock.close()

    print("Open Ports:", *open_ports)


if __name__ == "__main__":
    main(sys.argv)