Как ссылаться на функции в списке

# -*- coding: utf-8 -*-
 
import simple_draw as sd
 
# Запросить у пользователя желаемую фигуру посредством выбора из существующих
#   вывести список всех фигур с номерами и ждать ввода номера желаемой фигуры.
# и нарисовать эту фигуру в центре экрана
 
# Код функций из упр lesson_004/02_global_color.py скопировать сюда
# Результат решения см lesson_004/results/exercise_03_shape_select.jpg
 
point1 = sd.get_point(250, 250)
point2 = sd.get_point(250, 250)
point3 = sd.get_point(250, 250)
point4 = sd.get_point(250, 250)
angle = 0
 
 
def triangle(point, angle_1, length):
    for angle in range(0, 360, 120):
        v = sd.get_vector(start_point=point, angle=angle, length=length, width=2)
        v.draw()
        point = v.end_point
 
 
def square(point, angle_2, length):
    for angle in range(0, 360, 90):
        v = sd.get_vector(start_point=point, angle=angle, length=length, width=2)
        v.draw()
        point = v.end_point
 
 
def pentagon(point, angle_3, length, ):
    for angle in range(0, 360, 72):
        v = sd.get_vector(start_point=point, angle=angle, length=length, width=2)
        v.draw()
        point = v.end_point
 
 
def hexagon(point, angle_4, length, ):
    for angle in range(0, 360, 60):
        v = sd.get_vector(start_point=point, angle=angle, length=length, width=2)
        v.draw()
        point = v.end_point
 
 
# TODO Будет удобнее создать словарь в значениях которого будут храниться
#  названием фигур и ссылками на функции, а ключами словаря будут цифры,
#  вводимые пользователем.
#  Ссылки на функции, проще говоря их названия, можно поместить в структуру данных:
#  список, кортеж, словарь, ... На примере списков это работает так:
#  functs = [pentagon, hexagon, triangle]
#  draw_function = functs[0]
#  draw_function(start_point, start_angle, length
#  Также будет удобно сохранить в этой же структуре названия фигур.
 
def user_input():
    print('Возможные фигуры:\n1 : Треугольник \n2 : Квадрат \n3 : Пятиугольник \n4 : Шестиугольник')
    while True:
        user_shape = int(input('Введите желаемую фигуру: '))
        if 0 < user_shape < 5:
            if user_shape == 1:
                triangle(point=point1, angle_1=angle, length=100)
            if user_shape == 2:
                square(point=point2, angle_2=angle, length=100)
            if user_shape == 3:
                pentagon(point=point3, angle_3=angle, length=100)
            if user_shape == 4:
                hexagon(point=point4, angle_4=angle, length=100)
            return user_shape
        else:
            print('Вы ввели некорректный номер!')
            continue
 
 
user_input()
sd.pause()

    dic = {1: lambda: triangle(point=point1, angle_1=angle, length=100),
           2: lambda: square(point=point2, angle_2=angle, length=100),
           3: lambda: pentagon(point=point3, angle_3=angle, length=100),
           4: lambda: hexagon(point=point4, angle_4=angle, length=100)}
 
#...
 
while True:
        user_shape = int(input('Введите желаемую фигуру: '))
        if user_shape in dic.keys():
            dic[user_shape]()
            break
#...

def user_input():
    dic = {1: triangle, 2: square, 3: pentagon, 4: hexagon}
 
    print('Возможные фигуры:\n1 : Треугольник \n2 : Квадрат \n3 : Пятиугольник \n4 : Шестиугольник')
    while True:
        user_shape = int(input('Введите желаемую фигуру: '))
        if user_shape in dic.keys():
            dic[user_shape](point1, angle, 100)
            break
        else:
            print('Вы ввели некорректный номер!')
            continue