import argparse
import colorful
import getpass
import sys
import telnetlib

color_palette = {
    'ip' : '#5888B8',
    'fail' : '#D33F3F',
    'success' : '#32CD32'
}
fail_count = 0
failed_servers = []
host = ''
port = ''
tn = telnetlib.Telnet()

colorful.update_palette(color_palette)
parser = argparse.ArgumentParser(description='Telnet to a an addresses')
parser.add_argument('telnet', metavar='Telnet Address', type=str, nargs='+',
                   help='Address to telnet to')
args = parser.parse_args()

host = args.telnet[0]
port = args.telnet[1]

try:
    tn.open(host, port)
    print(f'telnet to {colorful.ip}{host}{colorful.reset} on port {colorful.ip}{port}{colorful.reset}: {colorful.success}success{colorful.reset}')
except:
    print(f'telnet to {colorful.ip}{host}{colorful.reset} on port {colorful.ip}{port}{colorful.reset}: {colorful.fail}failed{colorful.reset}')
    fail_count += 1
    failed_servers.append(host)
tn.close