main2.py

##### 10 - Game over ####
import pygame, random, numpy as np
from snake_agent import agent
from pygame.locals import *

import socket
import atexit
import json
import numpy as np

# ----------- Server Config ----------
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)


# Helper functions
def on_grid_random():
	x = random.randint(3, 56)
	y = random.randint(3, 56)

	return (x * 10, y * 10)


def collision(c1, c2):
	return (c1[0] == c2[0]) and (c1[1] == c2[1])


# Macro definition for snake movement.
UP = 0
RIGHT = 1
DOWN = 2
LEFT = 3

pygame.init()
screen = pygame.display.set_mode((600, 600))
pygame.display.set_caption('Snake With AI =)')

# Board
food_board = np.random.randint(50, size=(60, 60))

snake = [on_grid_random()]

snake_skin = pygame.Surface((10, 10))
snake_skin.fill((255, 255, 255))  # White
snake_energy = 3

my_direction = LEFT

clock = pygame.time.Clock()

font = pygame.font.Font('freesansbold.ttf', 18)
score = 0
movement = 0

game_over = False
# while not game_over:

# !/usr/bin/env python3

import socket
import atexit
import json
import numpy as np

HOST = '127.0.0.1'  # Standard loopback interface address (localhost)
# HOST = 'localhost'  # Standard loopback interface address (localhost)
# HOST = '172.20.123.64'  # Standard loopback interface address (localhost)
PORT = 1397  # Port to listen on (non-privileged ports are > 1023)

while True:
	with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
		s.bind((HOST, PORT))
		# s.listen()
		conn, addr = s.accept()
		with conn:
			print('Connected by', addr)
			while True:
				data = conn.recv(1024)
				if not data:
					break
				while score <= 200:
					clock.tick(7)
					for event in pygame.event.get():
						if event.type == QUIT:
							pygame.quit()
							exit()


					def add_to_snake_tail():
						snake.append((0, 0))


					if snake_energy == 0 and len(snake) == 1:
						snake_energy = food_board[snake[0][0] // 10][snake[0][1] // 10]
						# snake.append((0, 0))
						score = score + snake_energy * 5 + 1

					my_direction = agent(food_board.copy(), snake.copy(), snake_energy, score)

					msg = s.listen()
					print(msg)
					data = {
							'food_board': food_board.tolist(),
					}
					conn.send(bytes(json.dumps(data), 'utf-8'))
					movement += 1
					if snake_energy:
						snake_energy -= 1
						add_to_snake_tail()
					else:
						if not len(snake) == 1:
							snake.pop(len(snake) - 1)
					# Check if snake collided with boundaries
					if snake[0][0] == 600 or snake[0][1] == 600 or snake[0][0] < 0 or snake[0][1] < 0:
						game_over = True
						print('hit boundaries')
						break

					# Check if the snake has hit itself
					for i in range(1, len(snake) - 1):
						if snake[0][0] == snake[i][0] and snake[0][1] == snake[i][1]:
							game_over = True
							print('hit itself')
							break

					if game_over:
						break

					for i in range(len(snake) - 1, 0, -1):
						snake[i] = (snake[i - 1][0], snake[i - 1][1])

					# Actually make the snake move.
					if my_direction == UP:
						snake[0] = (snake[0][0], snake[0][1] - 10)
					if my_direction == DOWN:
						snake[0] = (snake[0][0], snake[0][1] + 10)
					if my_direction == RIGHT:
						snake[0] = (snake[0][0] + 10, snake[0][1])
					if my_direction == LEFT:
						snake[0] = (snake[0][0] - 10, snake[0][1])

					screen.fill((0, 0, 0))

					for x in range(0, 600, 10):  # Draw vertical lines
						pygame.draw.line(screen, (40, 40, 40), (x, 0), (x, 600))
					for y in range(0, 600, 10):  # Draw vertical lines
						pygame.draw.line(screen, (40, 40, 40), (0, y), (600, y))

					# ---------------- Refresh Score On Screen ------------------

					score_font = font.render('Score: %s' % (score), True, (255, 255, 255))
					score_rect = score_font.get_rect()
					score_rect.topleft = (600 - 120, 10)
					screen.blit(score_font, score_rect)

					energy_font = font.render('Energy: %s' % (snake_energy), True, (255, 255, 255))
					energy_rect = score_font.get_rect()
					energy_rect.topleft = (80, 10)
					screen.blit(energy_font, energy_rect)

					move_font = font.render('move: %s' % (movement), True, (255, 255, 255))
					move_rect = score_font.get_rect()
					move_rect.topleft = (250, 10)
					screen.blit(move_font, move_rect)

					for pos in snake:
						screen.blit(snake_skin, pos)
					# -----------------------------------------------------------

					pygame.display.update()

				while True:
					game_over_font = pygame.font.Font('freesansbold.ttf', 75)
					# game_over_screen = game_over_font.render('Game Over', True, (255, 255, 255))
					game_over_screen = game_over_font.render('move: {}'.format(movement), True, (255, 255, 255))
					game_over_rect = game_over_screen.get_rect()
					game_over_rect.midtop = (600 / 2, 60)
					screen.blit(game_over_screen, game_over_rect)
					pygame.display.update()
					pygame.time.wait(1000)
					while True:
						for event in pygame.event.get():
							if event.type == QUIT:
								pygame.quit()
								exit()
				conn.send(bytes(json.dumps(1), 'utf-8'))


# @atexit.register
# def f():
# 	s.close()
# 	print(f'Connection {addr} was closed...')


@atexit.register
def f():
	s.close()