alphabet = ['A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W','X','Y','Z']
rot = [
	[[15, 4, 25, 20, 14, 7, 23, 18, 2, 21, 5, 12, 19, 1, 6, 11, 17, 8, 13, 16, 9, 22, 0, 24, 3, 10],0],
	[[25, 14, 20, 4, 18, 24, 3, 10, 5, 22, 15, 2, 8, 16, 23, 7, 12, 21, 1, 11, 6, 13, 9, 17, 0, 19],0],
	[[4, 7, 17, 21, 23, 6, 0, 14, 1, 16, 20, 18, 8, 12, 25, 5, 11, 24, 13, 22, 10, 19, 15, 3, 9, 2],0],
	[[8, 12, 4, 19, 2, 6, 5, 17, 0, 24, 18, 16, 1, 25, 23, 22, 11, 7, 10, 3, 21, 20, 15, 14, 9, 13],0],
	[[16, 22, 4, 17, 19, 25, 20, 8, 14, 0, 18, 3, 5, 6, 7, 9, 10, 15, 24, 23, 2, 21, 1, 13, 12, 11],0]
		 ]
rotors = [
	[[15, 4, 25, 20, 14, 7, 23, 18, 2, 21, 5, 12, 19, 1, 6, 11, 17, 8, 13, 16, 9, 22, 0, 24, 3, 10],0],
	[[25, 14, 20, 4, 18, 24, 3, 10, 5, 22, 15, 2, 8, 16, 23, 7, 12, 21, 1, 11, 6, 13, 9, 17, 0, 19],0],
	[[4, 7, 17, 21, 23, 6, 0, 14, 1, 16, 20, 18, 8, 12, 25, 5, 11, 24, 13, 22, 10, 19, 15, 3, 9, 2],0],
	[[8, 12, 4, 19, 2, 6, 5, 17, 0, 24, 18, 16, 1, 25, 23, 22, 11, 7, 10, 3, 21, 20, 15, 14, 9, 13],0],
	[[16, 22, 4, 17, 19, 25, 20, 8, 14, 0, 18, 3, 5, 6, 7, 9, 10, 15, 24, 23, 2, 21, 1, 13, 12, 11],0]
		 ]
reflector = [
	('A', 'E'), ('B', 'J'), ('C', 'M'), ('D', 'Z'), ('H', 'X'), ('I', 'V'),
	('L', 'F'), ('Q', 'O'), ('R', 'N'), ('T', 'S'), ('U', 'P'),
	('W', 'K'), ('Y', 'G'),
			]
#Creates the rotors for the enigma machine

def createPlugboard():
	global Plugboard
	Plugboard = []
	print('You are now requested to enter 10 pairs of letters,\neach pair must consist of unique letters and no letter can appear twice!')
	while len(Plugboard) != 10:
		temp = set(alphabet)
		for i in Plugboard:
			i = set(i)
			temp = temp - i
		print('The unchoosen letters are:',' '.join(temp))
		flag = True
		print('Pair -',len(Plugboard)+1)
		a , b = (input('Letter 1- ').upper() , input('Letter 2- ').upper())
		for i in Plugboard:
			if a in i or b in i:
				flag = False
				print('Letter allready choosen')
				#checks to see if letter allready in plugboard
		if a not in alphabet or b not in alphabet:
			flag = False
			print('Please enter a single letter')
			#Checks to see if letter is an individual charector and an actual letter
		elif a == b:
			flag = False
			print('Please enter different letters')
		if flag:
			Plugboard.append((a,b))
			#Finally adds the pair to the plugboard
		if Plugboard:
			print('Current plugboard is:')
			for i in Plugboard: print('	',i[0],'and',i[1])
		else:
			print('Current plugboard is:\nEmpty')

def selectRotors():
	global RotorInUse
	RotorInUse = []
	print('You will now select the order you would like the rotors in,\nthere are five rotors (1,2,3,4,5).')
	availableRotor = ['1','2','3','4','5']
	while len(RotorInUse) < 3:
		choice = input()
		#user chooses the rotors
		if choice not in availableRotor:
			print('Please choose an available rotor!')
			#checks to see if rotor in use
		else:
			RotorInUse.append(int(choice))
			#if not in use adds selected rotor
		print('Your Current Rotor order is:',RotorInUse)
	for i in range(len(RotorInUse)):
		RotorInUse[i] -=1

def RotorSettings():
	global rotorPositions , RotorInUse , rotors , rot
	print(rot)
	print('You will now have to set the rotor positions')
	print(rotors)
	rotors = rot
	print(rotors)
	#resets the rotors to default position
	rotorPositions = []
	desiredSetting = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20', '21', '22', '23', '24', '25']
	while len(rotorPositions) <3:
		flag = True
		setting = input('Please enter the desired position 0-25 for rotor',RotorInUse[len(rotorPositions)])
		if not(setting not in desiredSetting or setting in rotorPositions):
			rotorPositions.append(int(setting))
	for i in range(len(RotorInUse)):
		for j in range(rotorPositions[i]):
			rotors[RotorInUse[i]][0].append(rotors[RotorInUse[i]][0].pop(0))
			rotors[RotorInUse[i]][1] += 1

def ThroughRotor(letter):
	global rotors , RotorInUse
	r1 = rotors[RotorInUse[0]][0]
	r2 = rotors[RotorInUse[1]][0]
	r3 = rotors[RotorInUse[2]][0]
	#asigns easy variable names
	letter = alphabet.index(letter)
	letter = r3[r2[r1[letter]]]
	#first time through the rotors
	letter = alphabet[letter]
	#back into letter
	letter = reflect(letter)
	#through the reflector, switching letter
	letter = alphabet.index(letter)
	#back into a number
	letter = r3.index(letter)
	letter = r2.index(letter)
	letter = r1.index(letter)
	#through the rotors one final time
	letter = alphabet[letter]
	rotate()
	#adds one to the rotor possition
	return letter
	#returns the letter

def plugs(letter):
	global Plugboard
	p = Plugboard
	for i in range(len(p)):
		if letter in p[i]:
			letter = p[i][(1-p[i].index(letter))]
	return letter

def reflect(letter):
	global reflector
	r = reflector
	for i in range(len(r)):
		if letter in r[i]:
			letter = r[i][(1-r[i].index(letter))]
	return letter
	
	
def encode():
	string = input('Please enter the string you would like to encode or decode\n- ')
	encoded = ''
	for i in string:
		if i.upper() in alphabet:
			temp = plugs(i.upper())
			temp = ThroughRotor(temp)
			temp = plugs(temp)
			encoded += temp
		else:
			encoded += i
	return encoded
			
		
def rotate():
	global rotors , RotorInUse , rotorPositions
	r1 = RotorInUse[0]
	r2 = RotorInUse[1]
	r3 = RotorInUse[2]
	rotors[r1][1] += 1
	rotors[r1][0].append(rotors[r1][0].pop(0))
	if rotors[r1][1] == 26:
		rotors[r1][1] = 0
		rotors[r2][1] += 1
		rotors[r2][0].append(rotors[r2][0].pop(0))
		if rotors[r2][1] == 26:
			rotors[r2][1] = 0
			rotors[r3][1] += 1
			rotors[r3][0].append(rotors[r3][0].pop(0))
			if rotors[r3][1] == 26:
				rotors[r3][1] = 0
	rotorPositions[0] = rotors[r1][1]
	rotorPositions[1] = rotors[r2][1]
	rotorPositions[2] = rotors[r3][1]
def RotorMenu():
	global RotorInUse
	while True:
		choice = input('Please choose one of the following options;\n	1. New Rotors in machine\n	2. New Rotor positions\n	3. Exit\n- ')
		if choice =='2':
			if RotorInUse:	
				RotorSettings()			
			else:
				print('Please select which rotors you want first')
		elif choice == '1':
			selectRotors()
		elif choice == '3':
			break

def PlugboardMenu():			
	global Plugboard
	while True:
		choice = input('Please choose one of the following options;\n	1. Create completely new Plugboard pairs\n	2. Change a plugboard Pair\n	3. Exit\n- ')
		if choice =='1':
			createPlugboard()
		elif choice == '2':
			pass
			'''if Plugboard:
				ChangePair()
			else:
				print('Please Complete the plugboard before you attempt to change it!')'''
		elif choice == '3':
			break
	
def ChangePair():
	global Plugboard
	flag = True
	input('These are the pairs that you currently have set up!\n Please press enter to continue - ')
	for i in Plugboard:
		print(i[0],',',i[1])
	while flag:
		choice = input('Please enter a letter in the pair you wish to change,\n and then confirm that the selection is correct.\n If you wish to exit, simply type "exit"\n -').upper()
		if choice == 'exit':
			return
		for i in Plugboard:
			if choice in i:
				print(i)
				cont = input('Please confirm that the above pair is the one you wish to change,\nIf it is not then enter "n"\nIf it is correct enter "y"\n -').lower()
				if cont == 'y':
					flag = False
					choice = Plugboard.index(i)
	flag = True
	while flag:
		Plugboard.pop(choice)
		print('Please enter your new pair;')
		a , b = input('Letter 1: ').upper() , input('Letter 2: ').upper()
		if a not in alphabet or b not in alphabet:
			print('Please enter a single letter')
		else:
			for i in Plugboard:
				if a in i or b in i:
					print('Please enter letters that are not in use')
				else:
					Plugboard.append((a,b))
	
	
	
RotorInUse = [0,1,2]
Plugboard = [('Q', 'W'), ('E', 'R'), ('T', 'Y'), ('U', 'I'), ('O', 'P'), ('A', 'S'), ('D', 'F'), ('G', 'H'), ('J', 'K'), ('L', 'Z')]
rotorPositions = [0,0,0]
				
while True:
	print('\nPlease choose one of the following options;\n	1. Encode / Decode Message\n	2. Change the rotor settings\n	3. Change the plugboard\n	4. Complete Fresh Setup')
	print('Here is a list of the itmes you currently have set up and their settings:\nRotor Positions:',rotorPositions,'\nRotors in machine:',[i+1 for i in RotorInUse],'\nPlugboard:',Plugboard)
	choice = input('- ')
	
	if choice == '1':
		if RotorInUse and rotorPositions and Plugboard:	
			print(encode())
			input('Press enter to continue!')
		else:
			print('Please set up machine!')
	elif choice == '2':
		RotorMenu()
	elif choice == '3':
		PlugboardMenu()
	elif choice == '4':
		createPlugboard()
		selectRotors()
		RotorSettings()