from myro import *

window = GraphWin()

def makeBall(radius):
    x = randomNumber() * getWidth(window)
    y = randomNumber() * getHeight(window)
    ball = Circle(Point(x, y), radius)
    color = pickOne("red", "blue", "green", "yellow")
    ball.direction = -1
    ball.setFill(color)
    return ball

list_of_balls = []

for x in range(10):
    ball = makeBall(20)
    list_of_balls.append( ball )

def drawList(list):
    for item in list:
        item.draw(window)

def updateList(list):
    for item in list:
        if item.getP2().y > getHeight(window):
            item.direction = - item.direction
        elif item.getP2().y < 0:
            item.direction = - item.direction
            
        item.move(0, item.direction)