129 lines
4.5 KiB
Python
129 lines
4.5 KiB
Python
import logging
|
|
import os
|
|
import pickle
|
|
|
|
import pygame
|
|
from pygame import locals as pglocals
|
|
|
|
from config import RESOLUTION, SNAKE_COLOR, BACKGROUND_COLOR, FONT_COLOR, INITIAL_SNAKE_SIZE, SCORES_FILE
|
|
from objects import Snake, Apple
|
|
from utils import get_score_text, Direction
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
pygame.init()
|
|
|
|
|
|
def main():
|
|
clock = pygame.time.Clock()
|
|
logger.debug('pygame initialized')
|
|
|
|
if not pygame.font:
|
|
logger.warning('Fonts disabled')
|
|
if not pygame.mixer:
|
|
logger.warning('Sound disabled')
|
|
|
|
screen = pygame.display.set_mode(RESOLUTION) # type: pygame.Surface
|
|
pygame.display.set_caption('Snake')
|
|
scores = []
|
|
if os.path.isfile(SCORES_FILE):
|
|
with open(SCORES_FILE, 'rb') as scores_file:
|
|
scores = pickle.load(scores_file)
|
|
|
|
logger.info(f'High scores : {scores}')
|
|
|
|
while True:
|
|
screen.fill(BACKGROUND_COLOR)
|
|
snake = Snake()
|
|
screen.fill(SNAKE_COLOR, snake.head)
|
|
apple = Apple()
|
|
apple.display(screen)
|
|
|
|
score = 0
|
|
score_text, score_rect = get_score_text(score)
|
|
screen.blit(score_text, score_rect)
|
|
|
|
pygame.display.flip()
|
|
|
|
while True:
|
|
for event in pygame.event.get():
|
|
if event.type == pglocals.QUIT:
|
|
logger.warning('Received QUIT event')
|
|
return
|
|
if event.type == pglocals.KEYDOWN:
|
|
if event.key == pglocals.K_DOWN or event.key == pglocals.K_s:
|
|
snake.direction = Direction.DOWN
|
|
elif event.key == pglocals.K_LEFT or event.key == pglocals.K_q:
|
|
snake.direction = Direction.LEFT
|
|
elif event.key == pglocals.K_UP or event.key == pglocals.K_z:
|
|
snake.direction = Direction.UP
|
|
elif event.key == pglocals.K_RIGHT or event.key == pglocals.K_d:
|
|
snake.direction = Direction.RIGHT
|
|
elif event.key == pglocals.K_ESCAPE:
|
|
pause = True
|
|
while pause:
|
|
clock.tick(5)
|
|
for event in pygame.event.get():
|
|
if event.type == pglocals.KEYDOWN and event.key == pglocals.K_ESCAPE:
|
|
pause = False
|
|
break
|
|
|
|
dirty_rects = snake.move(screen, apple)
|
|
if snake.dead:
|
|
logger.info(f'Vous avez perdu ! Score : {score}')
|
|
scores.append({'player': 'PLAYER 1', 'score': score})
|
|
scores = sorted(scores, key=lambda x: x['score'], reverse=True)[:20]
|
|
logger.info(f'High scores : {scores}')
|
|
with open(SCORES_FILE, 'wb') as scores_file:
|
|
pickle.dump(scores, scores_file)
|
|
break
|
|
|
|
if snake.head.colliderect(apple.rect):
|
|
apple.renew()
|
|
score += apple.score
|
|
logger.info(f'Apple eaten, new score : {score}')
|
|
|
|
screen.fill(BACKGROUND_COLOR, score_rect)
|
|
old_score_rect = score_rect
|
|
|
|
score_text, score_rect = get_score_text(score)
|
|
screen.blit(score_text, score_rect)
|
|
dirty_rects.append(score_rect.union(old_score_rect))
|
|
|
|
dirty_rects.append(apple.display(screen))
|
|
|
|
pygame.display.update(dirty_rects)
|
|
|
|
# Run faster as snake grows
|
|
clock.tick(10 - INITIAL_SNAKE_SIZE + len(snake.slots))
|
|
|
|
screen.fill(BACKGROUND_COLOR)
|
|
|
|
font = pygame.font.Font(None, 60)
|
|
text = font.render(f"PERDU ! Score : {score}", 1, FONT_COLOR)
|
|
text_rect = text.get_rect() # type: pygame.Rect
|
|
text_rect.center = screen.get_rect().center
|
|
|
|
screen.blit(text, text_rect)
|
|
pygame.display.flip()
|
|
|
|
while True:
|
|
restart = False
|
|
for event in pygame.event.get():
|
|
if event.type == pglocals.QUIT:
|
|
logger.info('Received QUIT event')
|
|
logger.info(f'High scores : {scores}')
|
|
return
|
|
if event.type == pglocals.KEYDOWN:
|
|
if event.key == pglocals.K_r:
|
|
logger.debug('Restarting game')
|
|
restart = True
|
|
if restart:
|
|
break
|
|
|
|
clock.tick(5)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|