import pygame


class Widget(object):

    def __init__(self, rect):
        if not isinstance(rect, pygame.Rect):
            rect = pygame.Rect(rect, (0, 0))
        self.rect = rect

    def event(self, event):
        "Override me"
        pass

    def draw(self, surface):
        "Override me"
        pass


class Container(object):

    def __init__(self):
        self.children = []

    def event(self, event):
        for child in self.children:
            # TODO mouse events
            child.event(event)

    def add(self, widget):
        self.children.append(widget)

    def draw(self, surface):
        for child in self.children:
            child.draw(surface)