import pygame
from models.Object import Object


class Table(Object):
    def __init__(self, square_size, screen_size, left_square, top_square):
        super().__init__(
            'table',
            square_size,
            screen_size,
            left_square,
            top_square
        )

        self.state = 'table'

    def new_order(self):
        self.state = 'order'
        self.update_pic()

    def wait_order(self):
        self.state = 'wait'
        self.update_pic()

    def done_order(self):
        self.state = 'done'
        self.update_pic()

    def reset_order(self):
        self.state = 'table'
        self.update_pic()

    def is_order(self) -> bool:
        return self.state == 'order'

    def is_done(self) -> bool:
        return self.state == 'done'

    def update_pic(self):
        self.image = pygame.transform.scale(pygame.image.load(
            'images/{0}.png'.format(self.state)), (self.square_size, self.square_size))