diff --git a/snake/main.py b/snake/main.py index 09caccb..843363e 100644 --- a/snake/main.py +++ b/snake/main.py @@ -3,7 +3,7 @@ from window import Window if __name__ == '__main__': - window = Window(title='Snake', size=(800, 600)) + window = Window(title='Snake', size=(800, 600), background_color=(0, 128, 0)) clock = pygame.time.Clock() framerate = 25 diff --git a/snake/window.py b/snake/window.py index 41eef91..214c734 100644 --- a/snake/window.py +++ b/snake/window.py @@ -2,12 +2,13 @@ import pygame class Window: - def __init__(self, title, size): + def __init__(self, title, size, background_color=(0, 0, 0)): pygame.init() pygame.display.set_caption(title) self.__surface = pygame.display.set_mode(size=size) + self.__color = background_color def reset(self): - self.__surface.fill((0, 0, 0)) + self.__surface.fill(self.__color) \ No newline at end of file