vignere cipher python

# program to implement ceasar cipher encryption
import string

# method that removes symbols, letters, numbers from the message
def remove_ichars(message):
    other_chars = string.punctuation + string.digits + " "
    n_message = "".join([i for i in message if i not in other_chars])
    return n_message


# this method takes string and shiftvalue as input and returns encrypted string as output
def encrypt(original_msg, key):
    index = 0
    original_msg = original_msg.upper()
    original_msg = remove_ichars(original_msg)
    key = key.upper()
    allchars = string.ascii_uppercase
    encrypted_msg = ""

    for letter in original_msg:
        if letter in allchars:
            position = ( allchars.index(letter) + allchars.index(key[index]) ) % len(allchars)
            encrypted_msg += allchars[position]
            index += 1
            # reset index is greater than length of key
            if index >= len(key):
                index %= len(key)
        else:
            encrypted_msg += letter
            index = 0
    return encrypted_msg


# method that uses the encrypt method and encrypts files by reading the file path and storing the encrypted file on desktop
def encrypt_file(key):
    file_path = input("Enter File path : ")
    try:
        file1 = open(file_path, 'r+')
        f2 = open("C:/Users/MOHIT/Desktop/encrypted_file.txt", 'w')
        print("File opened Successfully!!")
        file_data = file1.read()
        print("File read Successfully!!")
        encrypted_data = encrypt(file_data, key)
        f2.write(encrypted_data)
        print("Encrypted File stored on Desktop successfully!!")

    except Exception as e:
        print("Problems with opening the file")
        print(str(e))

## main function

# encrypt("hello my name is mohit" , 3)
print("Program to implement Ceasar Cipher")
data = "This is program for vignere cipher"
key = input("Enter the Encryption key : ")
encrypt_file(key)
# print(encrypt(data, key))

'''

Program to implement Ceasar Cipher
Enter File path : F:/python_3.4/sample.txt
File opened Successfully!!
File read Successfully!!
Encrypted File stored on Desktop successfully!!

'''