updated snake and food

Author: itspyguru

Snake/main.py

@@ -20,36 +20,66 @@
 GREEN = (0, 255, 0)
 WHITE = (255, 255, 255)
 
-apple = pygame.image.load('apple.png')
-
 def drawGrid():
 	for row in range(ROWS):
 		pygame.draw.line(win, WHITE, (0, row*CELLSIZE), (WIDTH, row*CELLSIZE), 1)
 	for col in range(COLS):
 		pygame.draw.line(win, WHITE, (col*CELLSIZE, 0), (col*CELLSIZE, HEIGHT))
 
-def randomApple():
-	return random.randint(0,ROWS-1), random.randint(0,COLS-1)
-
-dx = dy = 0
 
 class Snake:
 	def __init__(self):
-		self.length = 0
-		self.direction = UP
-		self.body = 20
-		self.x = WIDTH // 2
-		self.y = HEIGHT // 2
+		self.length = 1
+		self.direction = None
+		self.x = COLS // 2
+		self.y = ROWS // 2
+		self.head = (COLS//2 * CELLSIZE, ROWS//2 * CELLSIZE)
+		self.body = [self.head]
 
 	def update(self):
-		self.x += dx
-		self.y += dy
+		head = self.body[-1]
+		if self.direction == 'up':
+			head = (head[0], head[1] - CELLSIZE)
+		elif self.direction == 'down':
+			head = (head[0], head[1] + CELLSIZE)
+		elif self.direction == 'left':
+			head = (head[0] - CELLSIZE, head[1])
+		elif self.direction == 'right':
+			head = (head[0] + CELLSIZE, head[1])
+
+		self.head = head
+		self.body.append(self.head)
+		if self.length < len(self.body):
+			self.body.pop(0)
+
+	def eatFood(self):
+		self.length += 1
+
+	def checkFood(self, food):
+		if self.head[0] == food.x and self.head[1] == food.y:
+			self.eatFood()
+			food.respawn()
+
+	def draw(self):
+		for block in self.body:
+			x, y = block
+			pygame.draw.rect(win, GREEN, (x, y, CELLSIZE, CELLSIZE))
+
+class Food:
+	def __init__(self):
+		self.image = pygame.image.load('apple.png')
+		self.respawn()
+
+	def respawn(self):
+		self.x = random.randint(0,COLS-1) * CELLSIZE
+		self.y = random.randint(0,ROWS-1) * CELLSIZE
+		print(self.x, self.y)
 
 	def draw(self):
-		pygame.draw.rect(win, GREEN, self.x, self.y, self.body, self.body)
+		win.blit(self.image, (self.x, self.y))
 
-ay, ax = randomApple()
 snake = Snake()
+food = Food()
 
 running = True
 while running:
@@ -62,28 +92,23 @@ def draw(self):
 			if event.key == pygame.K_ESCAPE:
 				running = False
 
-			if event.key == pygame.K_RIGHT:
-				dx = CELLSIZE
-				dy = 0
+			if event.key == pygame.K_RIGHT and snake.direction != 'left':
+				snake.direction = 'right'
 
-			if event.key == pygame.K_LEFT:
-				dx = -CELLSIZE
-				dy = 0
+			if event.key == pygame.K_LEFT and snake.direction != 'right':
+				snake.direction = 'left'
 
-			if event.key == pygame.K_UP:
-				dx = 0
-				dy = -CELLSIZE
+			if event.key == pygame.K_UP and snake.direction != 'down':
+				snake.direction = 'up'
 
-			if event.key == pygame.K_DOWN:
-				dx = 0
-				dy = CELLSIZE
+			if event.key == pygame.K_DOWN and snake.direction != 'up':
+				snake.direction = 'down'
 
-	# drawGrid()
-	x += dx
-	y += dy
-	win.blit(apple, (ax*CELLSIZE, ay*CELLSIZE))
+	drawGrid()
 	snake.update()
+	snake.checkFood(food)
 	snake.draw()
+	food.draw()
 
 	clock.tick(FPS)
 	pygame.display.update()