import pygame, sys, pymunk

def create_apple(space,pos):
    body = pymunk.Body(1,100,body_type = pymunk.Body.DYNAMIC)
    body.position = pos
    shape = pymunk.Circle(body,40)
    space.add(body,shape)
    return shape

def draw_apples(apples):
    for apple in apples:
        pos_x = int(apple.body.position.x)
        pos_y = int(apple.body.position.y)
        pygame.draw.circle(screen,(0,0,0),(pos_x,pos_y),40)

def static_ball(space,pos):
    body = pymunk.Body(body_type = pymunk.Body.STATIC)
    body.position = pos
    shape = pymunk.Circle(body,30)
    space.add(body,shape)
    return shape

def draw_static_ball(ball):
    for ball in balls:
        pos_x = int(ball.body.position.x)
        pos_y = int(ball.body.position.y)
        pygame.draw.circle(screen,(0,0,0),(pos_x,pos_y),30)

pygame.init()
pygame.display.set_caption("budget plinko")
screen = pygame.display.set_mode((800,800))
clock = pygame.time.Clock()
space = pymunk.Space()
space.gravity = (0,500)
apples = []


balls = []
balls.append(static_ball(space,(000,700)))
balls.append(static_ball(space,(200,700)))
balls.append(static_ball(space,(400,700)))
balls.append(static_ball(space,(600,700)))
balls.append(static_ball(space,(800,700)))

balls.append(static_ball(space,(100,500)))
balls.append(static_ball(space,(300,500)))
balls.append(static_ball(space,(500,500)))
balls.append(static_ball(space,(700,500)))

balls.append(static_ball(space,(200,300)))
balls.append(static_ball(space,(400,300)))
balls.append(static_ball(space,(600,300)))

balls.append(static_ball(space,(300,100)))
balls.append(static_ball(space,(500,100)))


while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        if event.type == pygame.MOUSEBUTTONDOWN:
            apples.append(create_apple(space,event.pos))
    
    screen.fill((217,217,217))
    draw_apples(apples)
    draw_static_ball(balls)
    space.step(1/50)
    pygame.display.update()
    clock.tick(120)