31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
import pygame
|
|
|
|
class Button:
|
|
def __init__(self, x, y, width, height, text, color, hover_color, text_color, action, font, font_size, extra_data = None):
|
|
self.rect = pygame.Rect(x, y, width, height)
|
|
self.text = text
|
|
self.color = color
|
|
self.hover_color = hover_color
|
|
self.text_color = text_color
|
|
self.current_color = color
|
|
self.action = action
|
|
self.font = pygame.font.Font(font, font_size)
|
|
self.extra_data = extra_data
|
|
|
|
def draw(self, screen):
|
|
pygame.draw.rect(screen, self.current_color, self.rect)
|
|
text_surface = self.font.render(self.text, True, self.text_color)
|
|
text_rect = text_surface.get_rect(center=self.rect.center)
|
|
screen.blit(text_surface, text_rect)
|
|
|
|
def handle_event(self, event):
|
|
if event.type == pygame.MOUSEMOTION:
|
|
if self.rect.collidepoint(event.pos):
|
|
self.current_color = self.hover_color
|
|
else:
|
|
self.current_color = self.color
|
|
|
|
elif event.type == pygame.MOUSEBUTTONDOWN:
|
|
if self.rect.collidepoint(event.pos):
|
|
self.action(self)
|