def init_name():
global player_name
# default is a random generated name
name = generateName()
font = pygame.font.Font(None, 50)
while True:
for evt in pygame.event.get():
if evt.type == pygame.KEYDOWN:
if evt.unicode.isalpha():
name += evt.unicode
elif evt.key == pygame.K_BACKSPACE:
name = name[:-1]
elif evt.key == pygame.K_RETURN:
# if new name entered, change it
if name != '':
player_name = name
return
elif evt.type == pygame.QUIT:
return
# generate avatar, changing according to name entered
avatar = generateAvatar(name)
global avatar_image
avatar_image = pygame.image.load(avatar)
screen.fill((0, 0, 0))
show_message(screen, 'Please Enter Your Name: ', whiteColour, 50, 10,
10)
screen.blit(avatar_image, (WIDTH // 2 - 25, HEIGHT // 2 + 50))
block = font.render(name, True, (255, 255, 255))
rect = block.get_rect()
rect.center = screen.get_rect().center
screen.blit(block, rect)
pygame.display.flip()
def help_loop():
font = pygame.font.SysFont("arial", 24)
font_height = font.get_linesize()
pygame.display.set_caption("Welcome To Snake")
background = 'data/bg.jpg'
bg = pygame.image.load(background).convert_alpha()
while True:
screen.fill((0, 0, 0))
screen.blit(bg, (0, 0))
pygame.draw.polygon(screen, whiteColour,
[[175, 225], [175, 450], [900, 450], [900, 225]],
4)
show_message(screen, 'Press WASD to move snake around!', blackColour,
40, 200, 250)
show_message(screen, 'move your snake to yellow wormhole to upgrade!',
blackColour, 40, 200, 300)
show_message(screen, 'eat as much food as possible!', blackColour, 40,
200, 350)
show_message(screen, 'Press F to return', blackColour, 40, 350, 400)
pygame.display.update()
for event in pygame.event.get():
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_f:
menu_loop()
if event.key == pygame.K_ESCAPE:
pygame.event.post(pygame.event.Event(pygame.QUIT))
def displayHelper(playSurface,
fpsClock,
level,
food_loc,
my_snake,
total_score,
player_name,
l=601,
op_new_cells=[]):
playSurface.fill(blackColour)
pygame.draw.polygon(playSurface, greenColour,
[[99, 99], [99, l], [l, l], [l, 99]], 1)
show_message(playSurface, 'Score: ' + str(total_score), whiteColour, 40,
10, 10)
show_message(playSurface, 'Level: ' + str(level), whiteColour, 40, 10, 50)
screen.blit(avatar_image, (20, 615))
show_message(playSurface, player_name, whiteColour, 40, 100, 630)
draw_surface(playSurface, redColour, food_loc, 10, 100)
draw_surface(playSurface, greenColour, my_snake.obstacles, 10, 100)
draw_surface(playSurface, whiteColour, my_snake.get_cells(), 10, 100)
draw_surface(playSurface, greyColour, op_new_cells, 10, 100)
pygame.display.flip()
# speed is changeable
fpsClock.tick(my_snake.speed)
def leader_board():
# set the data, before displaying
setData()
pygame.init()
pygame.mixer.init()
font = pygame.font.SysFont("arial", 24)
font_height = font.get_linesize()
board_screen = pygame.display.set_mode(Display_Size, pygame.DOUBLEBUF, 32)
pygame.display.set_caption("Welcome To Snake")
background = 'data/bg.jpg'
bg = pygame.image.load(background).convert_alpha()
# load firebase
db = firebase.database()
storage = firebase.storage()
# display the top 20 players with highest scores
users_by_score = db.child("players").get()
with open(SCORE_PATH, 'r') as f:
score_data = json.load(f)
while True:
board_screen.fill((0, 0, 0))
board_screen.blit(bg, (0, 0))
show_message(board_screen, 'Welcome to leaderBoard, press F to return',
blackColour, 30, 250, 200)
# the position of the first displaying data
x = 250
y = 520
# display avatar, name and score for each player
top_users = []
for user in users_by_score.each():
player_name = user.val()['name']
player_score = user.val()['score']
image_url = storage.child(
user.val()["avatarFilePath"]).get_url(None)
top_users.append((player_name, player_score, image_url))
top_users.sort(key=lambda x: x[1], reverse=False)
top_users = top_users[:5]
for user in top_users:
player_name, player_score, image_url = user[0], user[1], user[2]
response = requests.get(image_url)
image_file = io.BytesIO(response.content)
avatar_img = pygame.image.load(image_file)
board_screen.blit(avatar_img, (x, y))
show_message(board_screen,
player_name + ": " + str(player_score), blackColour,
25, x + 60, y + 20)
y -= 70
if (y < 240):
y = 520
x -= 220
# check input
pygame.display.update()
for event in pygame.event.get():
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_f:
menu_loop()
if event.key == pygame.K_ESCAPE:
pygame.event.post(pygame.event.Event(pygame.QUIT))
if event.type == pygame.QUIT:
quitHelper()
def duo_ai_mode():
global player1_name, player2_name, avatar1_image, avatar2_image
init_name()
final_score1 = 0
player1_name = player_name
avatar1_image = avatar_image
addPlayerData(player_name)
init_name()
final_score2 = 0
player2_name = player_name
avatar2_image = avatar_image
addPlayerData(player2_name)
screen_size = [700, 700]
pygame.init()
fpsClock = pygame.time.Clock()
# create pyGame screen
level = 1
playSurface = pygame.display.set_mode((screen_size[0], screen_size[1]))
my_board = Board(50, 50)
my_snake = Snake(50, 50)
op_snake = Snake(50, 50)
op_snake.obstacles = my_snake.obstacles
# set up food and snake
food_loc = [my_board.food]
draw_surface(playSurface, redColour, [my_board.food], 10,
100) # draw first food
pygame.display.set_caption('Food Snake')
total_score = 0
score = 0
isEnd = False
# check input and go as the direction
while True:
# give AI input, and let AI control snake
dir = AI1(my_board, my_snake)
my_snake.turn(dir)
# op is pure AI
dir = AI12(my_board, op_snake)
op_snake.turn(dir)
for event in pygame.event.get():
if event.type == pygame.QUIT:
quitHelper()
elif event.type == pygame.KEYDOWN:
# determine the event of keyBoard
if event.key == pygame.K_RIGHT or event.key == ord('d'):
my_snake.turn(RIGHT)
if event.key == pygame.K_LEFT or event.key == ord('a'):
my_snake.turn(LEFT)
if event.key == pygame.K_UP or event.key == ord('w'):
my_snake.turn(UP)
if event.key == pygame.K_DOWN or event.key == ord('s'):
my_snake.turn(DOWN)
if event.key == ord('q'):
isEnd = True
if event.key == pygame.K_ESCAPE:
pygame.event.post(pygame.event.Event(pygame.QUIT))
# increase along the moving direction
draw_surface(playSurface, blackColour, my_snake.get_cells(), 10, 100)
status1 = my_snake.tick(my_board.food, [])
my_snake.teleport_wall()
# increase along the moving direction
draw_surface(playSurface, blackColour, op_snake.get_cells(), 10, 100)
status2 = op_snake.tick(my_board.food, [])
op_snake.teleport_wall()
my_new_cells = my_snake.get_cells()
op_new_cells = op_snake.get_cells()
if my_snake.is_dead(op_new_cells) or op_snake.is_dead(
my_new_cells) or isEnd:
break
# When my snake eats the food
if status1 == 1 or status2 == 1:
score += my_board.foodWeight
if status1 == 1:
final_score1 += 1
else:
final_score2 += 1
my_board.new_food(my_new_cells, my_snake.obstacles, op_new_cells)
food_loc = [my_board.food]
if score - level > level / 2:
level += 1
if level % 2 == 0:
my_snake.add_obstcles(level)
if level % 3 == 0:
my_snake.speed += 1
playSurface.fill(blackColour)
pygame.draw.polygon(playSurface, greenColour,
[[99, 99], [99, 601], [601, 601], [601, 99]], 1)
show_message(playSurface, 'Score: ' + str(total_score), whiteColour,
40, 10, 10)
show_message(playSurface, 'Level: ' + str(level), whiteColour, 40, 10,
50)
screen.blit(avatar1_image, (20, 615))
show_message(playSurface, player1_name, whiteColour, 40, 100, 630)
screen.blit(avatar2_image, (20, 650))
show_message(playSurface, player2_name, whiteColour, 40, 100, 665)
draw_surface(playSurface, redColour, food_loc, 10, 100)
draw_surface(playSurface, greenColour, my_snake.obstacles, 10, 100)
draw_surface(playSurface, whiteColour, my_new_cells, 10, 100)
draw_surface(playSurface, greyColour, op_new_cells, 10, 100)
pygame.display.flip()
# speed is changeable
fpsClock.tick(my_snake.speed)
game_end(final_score1, final_score2)
def single_ai_mode():
# set up local player data
addPlayerData(player_name)
screen_size = [700, 700]
pygame.init()
fpsClock = pygame.time.Clock()
# create pyGame screen
playSurface = pygame.display.set_mode((screen_size[0], screen_size[1]))
my_board = Board(50, 50)
my_snake = Snake(50, 50)
# set up food and snake
food_loc = [my_board.food]
draw_surface(playSurface, redColour, [my_board.food], 10,
100) # draw first food
pygame.display.set_caption('Food Snake')
total_score = 0
score = 0
isEnd = False
while True:
# give AI input, and let AI control snake
dir = AI1(my_board, my_snake)
my_snake.turn(dir)
for event in pygame.event.get():
if event.type == pygame.QUIT:
quitHelper()
elif event.type == pygame.KEYDOWN:
if event.key == ord('q'):
isEnd = True
if event.key == pygame.K_ESCAPE:
pygame.event.post(pygame.event.Event(pygame.QUIT))
# increase along the moving direction
draw_surface(playSurface, blackColour, my_snake.get_cells(), 10, 100)
status = my_snake.tick(my_board.food, [])
my_snake.teleport_wall()
my_new_cells = my_snake.get_cells()
if my_snake.is_dead([]) or isEnd:
break
# When snake eats the food
if status == 1:
score += my_board.foodWeight
total_score += 1
my_board.new_food(my_new_cells, my_snake.obstacles, [])
food_loc = [my_board.food]
playSurface.fill(blackColour)
pygame.draw.polygon(playSurface, greenColour,
[[99, 99], [99, 601], [601, 601], [601, 99]], 1)
show_message(playSurface, 'Score: ' + str(total_score), whiteColour,
40, 10, 10)
screen.blit(avatar_image, (20, 615))
show_message(playSurface, player_name, whiteColour, 40, 100, 630)
draw_surface(playSurface, redColour, food_loc, 10, 100)
draw_surface(playSurface, greenColour, my_snake.obstacles, 10, 100)
draw_surface(playSurface, whiteColour, my_new_cells, 10, 100)
pygame.display.flip()
# speed is changeable
fpsClock.tick(my_snake.speed)
game_end(total_score)
def standard_mode():
# set up local player data
addPlayerData(player_name)
screen_size = [700, 700]
pygame.init()
fpsClock = pygame.time.Clock()
playSurface = pygame.display.set_mode((screen_size[0], screen_size[1]))
my_board = Board(50, 50)
my_snake = Snake(50, 50)
# set up food and snake
food_loc = [my_board.food]
worm_loc = []
draw_surface(playSurface, redColour, [my_board.food], 10,
100) # draw first food
pygame.display.set_caption('Food Snake')
total_score = 0
level = 1
score = 0
isEnd = False
while True:
for event in pygame.event.get():
if event.type == pygame.QUIT:
quitHelper()
elif event.type == pygame.KEYDOWN:
# determine the event of keyBoard
if event.key == pygame.K_RIGHT or event.key == ord('d'):
my_snake.turn(RIGHT)
if event.key == pygame.K_LEFT or event.key == ord('a'):
my_snake.turn(LEFT)
if event.key == pygame.K_UP or event.key == ord('w'):
my_snake.turn(UP)
if event.key == pygame.K_DOWN or event.key == ord('s'):
my_snake.turn(DOWN)
if event.key == ord('q'):
isEnd = True
if event.key == pygame.K_ESCAPE:
pygame.event.post(pygame.event.Event(pygame.QUIT))
# increase along the moving direction
draw_surface(playSurface, blackColour, my_snake.get_cells(), 10, 100)
status = my_snake.tick(my_board.food, my_board.wormhole)
#if all snake body disappeared, go to next level!
if my_snake.get_cells() == []:
my_snake.cells = [[1, 1], [1, 2],
[1, 3]] # reassign snake's position!
level += 1
score = 0
if level == 2:
for i in range(20, 30, 1):
my_snake.obstacles.append([i, 25])
elif level == 3:
for i in range(20, 30, 1):
my_snake.obstacles.append([i, 15])
my_snake.obstacles.append([i, 35])
worm_loc = []
my_board.new_food(my_new_cells, my_snake.obstacles, [])
food_loc = [my_board.food]
my_snake.teleport_wall()
my_new_cells = my_snake.get_cells()
if my_snake.is_dead([]) or isEnd:
break
if status == 1:
score += my_board.foodWeight
total_score += 1
if score > 1:
my_board.new_wormhole(my_new_cells, my_snake.obstacles, [])
worm_loc = [my_board.wormhole]
food_loc = []
else:
my_board.new_food(my_new_cells, my_snake.obstacles, [])
food_loc = [my_board.food]
playSurface.fill(blackColour)
pygame.draw.polygon(playSurface, greenColour,
[[99, 99], [99, 601], [601, 601], [601, 99]], 1)
show_message(playSurface, 'Score: ' + str(total_score), whiteColour,
40, 10, 10)
show_message(playSurface, 'Level: ' + str(level), whiteColour, 40, 10,
50)
screen.blit(avatar_image, (20, 615))
show_message(playSurface, player_name, whiteColour, 40, 100, 630)
draw_surface(playSurface, redColour, food_loc, 10, 100)
draw_surface(playSurface, yellowColour, worm_loc, 10, 100)
draw_surface(playSurface, greenColour, my_snake.obstacles, 10, 100)
draw_surface(playSurface, whiteColour, my_new_cells, 10, 100)
pygame.display.flip()
# speed is changeable
fpsClock.tick(my_snake.speed)
game_end(total_score)
def game_end(total_score, score2=-1):
# update total_score
if score2 == -1:
updatePlayerData(player_name, total_score)
else:
updatePlayerData(player1_name, total_score)
updatePlayerData(player2_name, score2)
# display ending message
while True:
pygame.init()
pygame.mixer.init()
font = pygame.font.SysFont("arial", 24)
font_height = font.get_linesize()
end_screen = pygame.display.set_mode(Display_Size, pygame.DOUBLEBUF,
32)
pygame.display.set_caption("Welcome To Snake")
background = 'data/bg.jpg'
bg = pygame.image.load(background).convert_alpha()
# check input
while True:
end_screen.fill((0, 0, 0))
end_screen.blit(bg, (0, 0))
show_message(end_screen, 'Game End', whiteColour, 60, 400, 250)
show_message(end_screen, 'Press S to come back menu.', blackColour,
40, 250, 350)
show_message(end_screen, 'Press L to show the leaderBoard.',
blackColour, 40, 250, 400)
if score2 == -1:
show_message(
end_screen,
player_name + " you got " + str(total_score) + " points",
whiteColour, 30, 250, 450)
else:
show_message(
end_screen,
player1_name + " you got " + str(total_score) + " points",
whiteColour, 30, 250, 450)
show_message(
end_screen,
player2_name + " you got " + str(score2) + " points",
whiteColour, 30, 250, 500)
pygame.display.update()
for event in pygame.event.get():
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_s:
menu_loop()
if event.key == pygame.K_l:
leader_board()
if event.type == pygame.QUIT:
return
quitHelper()
def menu_loop():
font = pygame.font.SysFont("arial", 24)
font_height = font.get_linesize()
pygame.display.set_caption("Welcome To Snake")
background = 'data/bg.jpg'
bg = pygame.image.load(background).convert_alpha()
# react to input
while True:
screen.fill((0, 0, 0))
screen.blit(bg, (0, 0))
pygame.draw.polygon(screen, whiteColour,
[[175, 225], [175, 550], [700, 550], [700, 225]],
4)
show_message(screen, 'Press S to play standard mode', blackColour, 40,
200, 250)
show_message(screen, 'Press A to play demo mode', blackColour, 40, 200,
300)
show_message(screen, 'Press D to play AI duo mode', blackColour, 40,
200, 350)
show_message(screen, 'Press H to get Help', blackColour, 40, 200, 400)
show_message(screen, 'Press L to go to LeaderBoard', blackColour, 40,
200, 450)
show_message(screen, 'Press C to clear the LeaderBoard', blackColour,
40, 200, 500)
pygame.display.update()
for event in pygame.event.get():
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_s:
init_name()
standard_mode()
if event.key == pygame.K_a:
init_name()
single_ai_mode()
if event.key == pygame.K_d:
duo_ai_mode()
if event.key == pygame.K_l:
leader_board()
if event.key == pygame.K_c:
clear_data()
if event.key == pygame.K_h:
help_loop()
if event.type == pygame.QUIT:
return
def leader_board():
# set the data, before displaying
setData()
pygame.init()
pygame.mixer.init()
font = pygame.font.SysFont("arial", 24)
font_height = font.get_linesize()
board_screen = pygame.display.set_mode(Display_Size, pygame.DOUBLEBUF, 32)
pygame.display.set_caption("Welcome To Snake")
background = 'data/bg.jpg'
bg = pygame.image.load(background).convert_alpha()
# load firebase
db = firebase.database()
storage = firebase.storage()
# display the top 20 players with highest scores
users_by_score = db.child("players").order_by_child("score").limit_to_last(
20).get().val()
print(users_by_score)
# with open(SCORE_PATH, 'r') as f:
# score_data = json.load(f)
back_button = Button()
while True:
board_screen.fill((0, 0, 0))
board_screen.blit(bg, (0, 0))
back_button.create_button(screen, whiteColour, 30, 100, 100, 50, 60,
"Back", steelBlue)
# the position of the first displaying data (800,520) bottom-left to top-right
x = 720
y = 520
# display avatar, name and score for each player
for playerName in users_by_score:
player = users_by_score[playerName]
# avatar_img = pygame.image.load(player["avatarFilePath"])
image_url = storage.child(player["avatarFilePath"]).get_url(None)
response = requests.get(image_url)
image_file = io.BytesIO(response.content)
avatar_img = pygame.image.load(image_file)
board_screen.blit(avatar_img, (x, y))
show_message(board_screen,
playerName + ": " + str(player["score"]),
blackColour, 25, x + 60, y + 20)
y -= 70
if (y < 240):
y = 520
x -= 220
# check input
pygame.display.update()
for event in pygame.event.get():
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_f:
menu_loop()
if event.key == pygame.K_ESCAPE:
pygame.event.post(pygame.event.Event(pygame.QUIT))
if event.type == pygame.QUIT:
quitHelper()
elif event.type == pygame.MOUSEBUTTONUP:
if back_button.pressed(pygame.mouse.get_pos()):
menu_loop()