Как запросить у пользователя желаемую фигуру посредством выбора из существующих функций для рисования объектов?

Question

[Iceforest]

я написал 4 функции для рисования 4х фигур
задача такая: пользователь вводит число равное какой-то фигуре и запускается только нужная функция
можно сделать с помощью if elif else, но это долгий вариант, хочется упростить, предполагаю, что нужно сделать словари
но как в словари добавить именно вызов функции?

Answer 1

[soremix]

Полагаю, вы имеете в виду нечто в виде

def func1():
    print('func1')

def func2():
    print('func2')

funcs = {1: func1, 2: func2}

funcs[1]()

Comments

[Iceforest]

немного не так, пользователь введет сам значение
user_input = input('Введите, пожалуйста, номер фигуры с 1 - 4: ')
number_figure = int(user_input)
я пытаюсь как то так
figures = [triangle(point=point, angle=0, length=200), square(point=point, angle=0, length=200),
pentangle(point, angle=0, length=100), sixangle(point=point, angle=0, length=100)]
figure_numbers = [1, 2, 3, 4]
res = dict(zip(figure_numbers, figures))
но это чушь какая-то

[aRegius]

Iceforest, а чем этот ответ отличается от отмеченного вами как решение?

[Iceforest]

aRegius, сейчас понял, что ничем, просто тот код я сразу понял

Answer 2

[sswwssww]

def func1():
    print('func1')

def func2():
    print('func2')

funcs = {'1': func1, '2': func2}
func_nmbr = input('Введите, пожалуйста, номер фигуры с 1 - 4: ')
funcs[func_nmbr]()

Comments

[Iceforest]

точно, чего я все усложняю, а как сделать так, чтобы они не рисовали? все 4 функции в них прописано рисование фигур и пока код выполняется они рисуются до момента определения номера

[sswwssww]

Iceforest, если я ответил на ваш заданный вопрос, отметьте пожалуйста мой ответ как решение. Чтобы ответить на второй вопрос мне нужен ваш код.

[Iceforest]

sswwssww, да, ответили.

[Iceforest]

sswwssww,

spoiler

import simple_draw as sd
def triangle(point, angle=0, length=200):
v1 = sd.get_vector(start_point=point, angle=angle, length=200, width=3)
v1.draw(color=sd.random_color())

v2 = sd.get_vector(start_point=v1.end_point, angle=angle + 120, length=200, width=3)
v2.draw(color=sd.random_color())

v3 = sd.get_vector(start_point=v2.end_point, angle=angle + 240, length=200, width=3)
v3.draw(color=sd.random_color())

def square(point, angle=0, length=200):
v1 = sd.get_vector(start_point=point, angle=angle, length=200, width=3)
v1.draw(color=sd.random_color())

v2 = sd.get_vector(start_point=v1.end_point, angle=angle + 90, length=200, width=3)
v2.draw(color=sd.random_color())

v3 = sd.get_vector(start_point=v2.end_point, angle=angle + 180, length=200, width=3)
v3.draw(color=sd.random_color())

v4 = sd.get_vector(start_point=v3.end_point, angle=angle - 90, length=200, width=3)
v4.draw(color=sd.random_color())
def pentangle(point, angle=0, length=100):
v1 = sd.get_vector(start_point=point, angle=angle, length=100, width=3)
v1.draw(color=sd.random_color())

v2 = sd.get_vector(start_point=v1.end_point, angle=angle + 60, length=100, width=3)
v2.draw(color=sd.random_color())

v3 = sd.get_vector(start_point=v2.end_point, angle=angle + 120, length=100, width=3)
v3.draw(color=sd.random_color())

v4 = sd.get_vector(start_point=v3.end_point, angle=angle - 180, length=100, width=3)
v4.draw(color=sd.random_color())

v5 = sd.get_vector(start_point=v4.end_point, angle=angle - 120, length=100, width=3)
v5.draw(color=sd.random_color())

v6 = sd.get_vector(start_point=v5.end_point, angle=angle - 60, length=100, width=3)
v6.draw(color=sd.random_color())

def sixangle(point, angle=0, length=100):
v1 = sd.get_vector(start_point=point, angle=angle + 30, length=100, width=3)
v1.draw(color=sd.random_color())

v2 = sd.get_vector(start_point=v1.end_point, angle=angle + 90, length=100, width=3)
v2.draw(color=sd.random_color())

v3 = sd.get_vector(start_point=v2.end_point, angle=angle + 150, length=100, width=3)
v3.draw(color=sd.random_color())

v4 = sd.get_vector(start_point=v3.end_point, angle=angle + 210, length=100, width=3)
v4.draw(color=sd.random_color())

v5 = sd.get_vector(start_point=v4.end_point, angle=angle + 270, length=100, width=3)
v5.draw(color=sd.random_color())

v6 = sd.get_vector(start_point=v5.end_point, angle=angle + 330, length=100, width=3)
v6.draw(color=sd.random_color())

point = sd.get_point(300, 300)
figures = {'1': triangle(point=point, angle=0, length=200), '2': square(point=point, angle=0, length=200),
'3': pentangle(point, angle=0, length=100), '4': sixangle(point=point, angle=0, length=100)}
figures_numbers=[1,2,3,4]
user_input = input('Введите, пожалуйста, номер фигуры с 1 - 4: ')
number_figure = int(user_input)
# number_figure = 2
if number_figure in figures_numbers:
figures[number_figure]()
else:
print('Вы ввели неправильный номер фигуры, введите значение от 1 до 4')

sd.pause()