import pyglet
from objects import Grid, Snake
from pyglet.window import key

cell_size = 15
win_w = 301
win_h = 301
speed = 0.5
ori = None

fps = pyglet.clock.ClockDisplay(interval=speed)

class Window(pyglet.window.Window):
    def __init__(self, ori):
        super().__init__()
        self.set_size(win_w, win_h)
        self.ori = ori

    def on_draw(self):
        self.clear()
        Grid.draw_grid()
        Snake.move(self.ori)

    def on_key_press(self, sym, mod):
        if sym == key.W:
            self.ori = 0
        elif sym == key.D:
            self.ori = 1
        elif sym == key.S:
            self.ori = 2
        elif sym == key.A:
            self.ori = 3

if __name__ == '__main__':

    window = Window(ori)
    Grid = Grid(win_w, win_h, cell_size)
    Snake = Snake(win_w, win_h, cell_size)

    pyglet.app.run()