import pyAesCrypt
def encrypt(key,source):
    """
    :param key: encryption key (text)
    :param source: File to be encrypted
    :return: Encrypted file  (original + .enc)
    """
    output = source+".enc"
    pyAesCrypt.encryptFile(source,output,key)
    return output

def decrypt(key,source):
    """
    :param key: decryption key (same as encrypting key - text)
    :param source: Decrypted file (source name + 'dec'+extension)
    :return:
    """
    dfile = source.split('.')
    output = dfile[0]+'dec.'+dfile[1]
    pyAesCrypt.decryptFile(source,output,key)
    return

def main():
    key = 'testkey'
    source = '1.bmp'
    file_enc=encrypt(key,source)
    decrypt(key,file_enc)

if __name__ == '__main__':
    main()