def figures(x, y, start_angle, step, length):
    start_point = sd.get_point(x, y)
    end_point = sd.get_point(x, y)
    for angle in range(0, 360 - step, step):
        vector = sd.get_vector(start_point=start_point, angle=angle + start_angle, length=length, width=2)
        vector.draw()
        start_point = vector.end_point
    sd.line(start_point=start_point, end_point=end_point, width=2)


def triangle():
    figures(100, 100, 10, 120, 100)


def square():
    figures(300, 100, 10, 90, 100)


def pentagon():
    figures(100, 300, 10, 72, 100)


def hexagon():
    figures(300, 300, 10, 60, 100)


figures_total = {
    '0': {'figure_name': 'triangle', 'sd.name': triangle()},
    '1': {'figure_name': 'square', 'sd.name': square()},
    '2': {'figure_name': 'pentagon', 'sd.name': pentagon()},
    '3': {'figure_name': 'hexagon', 'sd.name': hexagon()},
}
for number, figure in figures_total.items():
    print(number, ' - ', figure['figure_name'])

figure_1 = input("Введите, пожалуйста, номер фигуры: ")

if figure_1 in figures_total:
    triangle(figures_total[figure_1]['sd.name'])
    square(figures_total[figure_1]['sd.name'])
    pentagon(figures_total[figure_1]['sd.name'])
    hexagon(figures_total[figure_1]['sd.name'])
    sd.pause()
else:
    print('Вы что-то напутали. Попробуйте еще раз!')