import argparse
import sys, getpass

parser = argparse.ArgumentParser()
parser.add_argument('connect', help = '[username@host], [host]')
parser.add_argument('-E', help = '-E logfile')
parser.add_argument('-p', type = int, help = '-p port')
parser.add_argument('-l', help = '-l login_name')
parser.add_argument('-L', help = '-L adress')
parser.add_argument('-W', help = '-W host:port')
parser.add_argument('--connection', help = 'connect', action = 'store_true')
parser.add_argument('--noconnection', help = 'noconnect', action = 'store_true')

if len(sys.argv) == 1:
	print('---------------------------------------------------------------\n')
	parser.print_help()
	print('\n---------------------------------------------------------------')
	sys.exit(0)

args = parser.parse_args()

password = host = password = ""
port = 22

if args.connection and args.noconnection:
	print('---------------------------------------------------------------\n')
	print('Use only one, --connection or --noconnection')
	parser.print_help()
	print('\n---------------------------------------------------------------')
	sys.exit(0)


if args.p:
	port = args.p

if args.connect:
	if '@' in args.connect:
		username = args.connect.split('@')[0]
		host = args.connect.split('@')[1]
		password = getpass.getpass('Password: ')
	else:
		if not args.L and not args.W:
			host = args.connect
			username = (input('Username: '))
		password = getpass.getpass('Password: ')

if args.l:
	username = args.l

if args.L:
	host = args.L

if args.W:
	if ':' in args.W:
		host = args.W.split(':')[0]
		port = int(args.W.split(':')[1])


if not args.E:
		if password == 'raspberry' and username == 'pi' and host == 'localhost' and port == 22:
			print('Connecting to %s on port %d with username %s...' % (host, port, username))
			print('Connection successful!\n\n')
		else:
			print('Error connecting to %s on port %d with user %s: Unknown user' % (host, port, username))
else:
	if args.connect:
		f = open(args.E, 'w')
		if password == 'raspberry' and username == 'pi' and host == 'localhost' and port == 22:
			f.write('Connecting to %s with username %s...\n' % (host, username))
			f.write('Connection successful!\n\n\n')
		else:
			f.write('Error connecting to %s with user %s: Unknown user\n' % (host, username))
		f.close()