pygame display buttons code example

Example: event.type == pygame.MOUSEBUTTONDOWN click

def handle_event(self, screen, event, player):
        if event.type == pygame.QUIT:
            sys.exit()

        if event.type == pygame.MOUSEMOTION:
            pygame.draw.rect(screen, BACKGROUND, (0,0, self.screen_width, SQUARESIZE))
            posx = event.pos[0]
            if player == 1:
                pygame.draw.circle(screen, P1STONE, (posx, int(SQUARESIZE/2)), RADIUS)
            else:
                pygame.draw.circle(screen, P2STONE, (posx, int(SQUARESIZE/2)), RADIUS)
            pygame.display.update()    
            return None
        
        if event.type == pygame.MOUSEBUTTONDOWN:
            pygame.draw.rect(screen, BACKGROUND, (0,0, self.screen_width, SQUARESIZE))
            posx   = event.pos[0]
            action = int(np.floor(posx/SQUARESIZE))
            return action