Spacing in a single letter viginere cipher

Question

Trying to account for spacing/whitespaces in a viginere cipher in python. this is a single letter encryption.

Please note that i'm very new to python!

def encrypt(message, key):
'''Vigenere encryption of message using key.'''

# Converted to uppercase.
# Non-alpha characters stripped out.
message = filter(str.isalpha, message.upper())

def enc(c, k):
'''Single letter encryption.'''

return chr(((ord(k) + ord(c) - 2 * ord('A')) % 26) + ord('A'))

return ''.join(starmap(enc, zip(message, cycle(key))))

def decrypt(message, key):
'''Vigenere decryption of message using key.'''

def dec(c, k):
'''Single letter decryption.'''

return chr(((ord(c) - ord(k) - 2 * ord('A')) % 26) + ord('A'))

return ''.join(starmap(dec, zip(message, cycle(key))))
Have already tried this but it does not work

def enc(c, k):
'''Single letter encryption.'''
if ord(c) == 32:
return chr(ord(c))
else:
return chr(((ord(k) + ord(c) - 2 * ord('A')) % 26) + ord('A'))
This is what i get with my current code:

"Hello World" --> "JKJLJYUPLY" key = "wasup"
I want:

"Hello world" --> "JKJLJ YUPLY" key "wasup"

Answer 1

from itertools import starmap, cycle

def func(s):
return s.isalpha() or s == " "

def enc(c, k):
return chr(((ord(k) + ord(c) - 2 * ord('A')) % 26) + ord('A'))

def encrypt(message, key):
message = "".join(filter(func, message.upper())).split()

return ' '.join(''.join(starmap(enc, zip(word, cycle(key)))) for word in message)

print(encrypt("Hello world", "wasup"))
# JKJLJ YUPLY