66 lines
2.8 KiB
Python
66 lines
2.8 KiB
Python
# Welcome to
|
|
# __________ __ __ .__ __
|
|
# \______ \_____ _/ |__/ |_| | ____ ______ ____ _____ | | __ ____
|
|
# | | _/\__ \\ __\ __\ | _/ __ \ / ___// \\__ \ | |/ // __ \
|
|
# | | \ / __ \| | | | | |_\ ___/ \___ \| | \/ __ \| <\ ___/
|
|
# |________/(______/__| |__| |____/\_____>______>___|__(______/__|__\\_____>
|
|
#
|
|
# This file can be a nice home for your Battlesnake logic and helper functions.
|
|
#
|
|
# To get you started we've included code to prevent your Battlesnake from moving backwards.
|
|
# For more info see docs.battlesnake.com
|
|
|
|
from server.Files import read_file, save_file
|
|
from server.GameStorage import GameStorage
|
|
from config import SNAKE
|
|
|
|
from dotenv import load_dotenv, find_dotenv
|
|
from datetime import datetime
|
|
import json, os
|
|
|
|
# info is called when you create your Battlesnake on play.battlesnake.com
|
|
# and controls your Battlesnake's appearance
|
|
# TIP: If you open your Battlesnake URL in a browser you should see this data
|
|
game_state_storage = GameStorage(SNAKE.__class__.__name__)
|
|
|
|
def info() -> dict:
|
|
CONFIG_PATH = os.path.join(os.path.dirname(__file__), 'data', 'snake-config.json')
|
|
snake = read_file(CONFIG_PATH, json.load)
|
|
if not snake:
|
|
snake = {"apiversion":"1","author":"","color":"#888888","head":"default","tail":"default"}
|
|
save_file(CONFIG_PATH, snake, callback=json.dump, indent=2, ensure_ascii=False)
|
|
print("INFO", snake)
|
|
return snake
|
|
|
|
print("INFO Snake:", snake)
|
|
return snake
|
|
|
|
# start is called when your Battlesnake begins a game
|
|
def start(game_state:dict):
|
|
game_state_storage.start_new_game(game_state["game"], game_state["board"], game_state["you"])
|
|
SNAKE.clear_history()
|
|
print("GAME START:", game_state["game"])
|
|
|
|
# move is called when your Battlesnake game is running game
|
|
def move(game_state:dict) -> dict:
|
|
next_move = SNAKE.choose_move(game_state)
|
|
game_state_storage.add_moves(game_state["board"], next_move)
|
|
print("MOVE:", f"{next_move:5},", "Me:", {"head": game_state["you"]["head"], "length": game_state["you"]["length"]})
|
|
return {"move": next_move}
|
|
|
|
# end is called when your Battlesnake finishes a game
|
|
def end(game_state:dict):
|
|
HISTORY_PATH = os.path.join(os.path.dirname(__file__), 'data', 'history')
|
|
|
|
game_state_storage.add_end_state(game_state["board"], SNAKE.get_history())
|
|
game_state_storage.save(os.path.join(HISTORY_PATH, f"{SNAKE.__class__.__name__}_{datetime.now().strftime('%d.%m.%Y_%H%M%S')}_{game_state['game']['id']}.json"), callback=json.dump, indent=2, ensure_ascii=False)
|
|
|
|
print("GAME OVER\n")
|
|
|
|
# Start server when `python main.py` is run
|
|
if __name__ == "__main__":
|
|
load_dotenv(find_dotenv())
|
|
|
|
from server.server import run_server
|
|
run_server({"info": info, "start": start, "move": move, "end": end})
|