diff --git a/snake/main.py b/snake/main.py index b5cea8b..09caccb 100644 --- a/snake/main.py +++ b/snake/main.py @@ -1,10 +1,10 @@ import pygame +from window import Window if __name__ == '__main__': - pygame.init() - pygame.display.set_caption('Snake') - surface = pygame.display.set_mode(size=(800, 600)) + window = Window(title='Snake', size=(800, 600)) + clock = pygame.time.Clock() framerate = 25 @@ -15,7 +15,7 @@ if __name__ == '__main__': if event.type == pygame.QUIT: shall_abort = True - surface.fill((0, 0, 0)) + window.reset() clock.tick(framerate) pygame.display.flip() diff --git a/snake/window.py b/snake/window.py new file mode 100644 index 0000000..41eef91 --- /dev/null +++ b/snake/window.py @@ -0,0 +1,13 @@ +import pygame + + +class Window: + def __init__(self, title, size): + pygame.init() + pygame.display.set_caption(title) + self.__surface = pygame.display.set_mode(size=size) + + + def reset(self): + self.__surface.fill((0, 0, 0)) + \ No newline at end of file