141 lines
3.5 KiB
Python
141 lines
3.5 KiB
Python
from flask import Flask, request, jsonify
|
|
from flask_json import FlaskJSON, as_json_p
|
|
from flask_cors import CORS
|
|
from board import Board
|
|
import main
|
|
import random
|
|
from network import Network
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
app.config['JSON_ADD_STATUS'] = False
|
|
app.config['JSON_JSONP_OPTIONAL'] = False
|
|
|
|
json = FlaskJSON(app)
|
|
CORS(app)
|
|
|
|
config = main.config.copy()
|
|
config['model'] = "player_testings"
|
|
config['ply'] = "1"
|
|
config['board_representation'] = 'quack-fat'
|
|
network = Network(config, config['model'])
|
|
|
|
network.restore_model()
|
|
|
|
|
|
def calc_move_sets(from_board, roll, player):
|
|
board = from_board
|
|
sets = []
|
|
total = 0
|
|
for r in roll:
|
|
print("Value of r:", r)
|
|
sets.append([Board.calculate_legal_states(board, player, [r, 0]), r])
|
|
total += r
|
|
sets.append([Board.calculate_legal_states(board, player, [total, 0]), total])
|
|
# print(sets)
|
|
return sets
|
|
|
|
|
|
def tmp_name(from_board, to_board, roll, player, total_moves, is_quad=False):
|
|
sets = calc_move_sets(from_board, roll, player)
|
|
return_board = from_board
|
|
for idx, board_set in enumerate(sets):
|
|
board_set[0] = list(board_set[0])
|
|
# print(to_board)
|
|
# print(board_set)
|
|
if to_board in board_set[0]:
|
|
print("To board:", to_board)
|
|
print(board_set[0])
|
|
print(board_set[1])
|
|
total_moves -= board_set[1]
|
|
# if it's not the sum of the moves
|
|
if idx < (4 if is_quad else 2):
|
|
roll[idx] = 0
|
|
else:
|
|
roll = [0, 0]
|
|
return_board = to_board
|
|
break
|
|
|
|
print("Return board!:\n",return_board)
|
|
return total_moves, roll, return_board
|
|
|
|
def calc_move_stuff(from_board, to_board, roll, player):
|
|
|
|
is_quad = roll[0] == roll[1]
|
|
|
|
total_moves = roll[0] + roll[1] if not is_quad else int(roll[0]) * 4
|
|
if is_quad:
|
|
roll = [roll[0]] * 4
|
|
|
|
total_moves, roll, board = tmp_name(from_board, to_board, list(roll), player, total_moves, is_quad)
|
|
return board, total_moves, roll
|
|
|
|
|
|
@app.route('/get_board', methods=['GET'])
|
|
@as_json_p
|
|
def get_board():
|
|
return {'board':'0, 2, 0, 0, 0, 0, -5, 0, -3, 0, 0, 0, 5, -5, 0, 0, 0, 3, 0, 5, 0, 0, 0, 0, -2, 0'}
|
|
|
|
|
|
|
|
def check_move(prev, curr):
|
|
|
|
# TODO: Decide on player system and implement roll properly
|
|
legal_states = Board.calculate_legal_states(tuple(prev), -1, [1,2])
|
|
|
|
truth_list = [list(curr) == list(ele) for ele in legal_states]
|
|
|
|
return any(truth_list)
|
|
|
|
|
|
|
|
@app.route('/bot_move', methods=['POST'])
|
|
def bot_move():
|
|
data = request.get_json(force=True)
|
|
|
|
board = [int(x) for x in data['board'].split(',')]
|
|
|
|
roll = (random.randrange(1,7), random.randrange(1,7))
|
|
print(roll)
|
|
board, _ = network.make_move(tuple(board), roll, 1)
|
|
print("Boards!:",board)
|
|
|
|
return ",".join([str(x) for x in list(board)])
|
|
|
|
|
|
|
|
@app.route('/post_board', methods=['POST'])
|
|
def post_board():
|
|
data = request.get_json(force=True)
|
|
|
|
# TODO: Fix hardcoded player
|
|
player = -1
|
|
|
|
board = [int(x) for x in data['board'].split(',')]
|
|
prev_board = [int(x) for x in data['prev_board'].split(',')]
|
|
roll = [int(x) for x in data['roll'].split(',')]
|
|
|
|
|
|
print(board)
|
|
|
|
total_roll = int(data['total_roll'])
|
|
|
|
return_board, total_moves, roll = calc_move_stuff(tuple(prev_board), tuple(board), tuple(roll), player)
|
|
|
|
|
|
|
|
|
|
str_board = ",".join([str(x) for x in return_board])
|
|
str_roll = ",".join([str(x) for x in roll])
|
|
|
|
|
|
return_string = str_board + "#" + str(total_moves) + "#" + str_roll
|
|
|
|
print(return_string)
|
|
|
|
return return_string
|
|
|
|
if __name__ == '__main__':
|
|
app.run()
|