Rotors

class Rotor:
    def __init__(self,name):
        self.setting={
            'I':'EKMFLGDQVZNTOWYHXUSPAIBRCJ',
            'II':'AJDKSIRUXBLHWTMCQGZNPYFVOE',
            'III':'BDFHJLCPRTXVZNYEIWGAKMUSQO',
            'IV': 'ESOVPZJAYQUIRHXLNFTGKDCMWB',
            'V':'VZBRGITYUPSDNHLXAWMJQOFECK',
            'Beta':'LEYJVCNIXWPBQMDRTAKZGFUHOS',
            'Gamma':'FSOKANUERHMBTIYCWLQPZXVGJD'
        }
        self.alphabet = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
        if name in self.setting:
            self.rotor = name
        else:
            print('Invalid Rotor!')
    def encode_right_to_left(self, letter):
        return(self.setting[self.rotor][self.alphabet.index(letter)])

    def encode_left_to_right(self,letter):
        return(self.alphabet[self.setting[self.rotor].index(letter)])