Торт на день рождения: задача на Python

Напишите функцию, которая будет создавать прямоугольный «торт», основываясь на переданном имени (name) и возрасте (age). Торт будет представлять собой строки в списке. Поздравление с днем рождения на торте окружается рамочкой:

• если возраст — четное число, рамочка состоит из символов «#»
• если возраст — нечетное число, рамочка состоит из символов «*».

Примечания:

• Поздравление должно быть в следующем формате: {age} Happy Birthday {name}! {age}
• Между рамочкой и цифрами возраста должен оставаться пробел.

Примеры

get_birthday_cake("Jack", 10) ➞ [
  "##############################",
  "# 10 Happy Birthday Jack! 10 #",
  "##############################"
]

get_birthday_cake("Russell", 19) ➞ [
  "*********************************",
  "* 19 Happy Birthday Russell! 19 *",
  "*********************************"
]

get_birthday_cake("Isabelle", 2) ➞ [
  "################################",
  "# 2 Happy Birthday Isabelle! 2 #",
  "################################"
]

Решение

def get_birthday_cake(name, age):
    char = ['#', '*'][age%2]
    msg = '{0} {1} Happy Birthday {2}! {1} {0}'.format(char, age, name)
    wall = char * len(msg)
    return [wall, msg, wall]