# 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 datetime import datetime import typing 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() -> typing.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: typing.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: typing.Dict) -> typing.Dict: next_move = SNAKE.choose_move(game_state) game_state_storage.add_moves(game_state["board"], next_move) print("MOVE:", f"{next_move},", "Me:", {"body": game_state["you"]["body"], "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: typing.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__": from server.server import run_server run_server({"info": info, "start": start, "move": move, "end": end})