polygon_turtle.py

from functools import partial
import turtle

window = turtle.Screen()

def draw_polygon(sides, color):
    turtle.color(color)
    
    angle = 360/sides
    for i in range(sides):
      turtle.forward(50)
      turtle.right(angle)


for sides, color in [(3, "red"), (4, "green")]:
    window.onkey(partial(draw_polygon, sides, color), str(sides))

window.listen()
window.mainloop()