ozai-webui/main.py

133 lines
4.7 KiB
Python

# Python
from flask import Flask, render_template, request
import requests
ozai_url = 'http://localhost:8000/api/'
app = Flask(__name__)
# @app.route('/azul.webp')
# def background():
# #return the background image
# return app.send_static_file('azul.webp')
@app.route('/')
def home():
return render_template('home.html')
@app.route('/azul.webp')
def background():
#return the background image
return app.send_static_file('azul.webp')
@app.route('/create', methods=['GET', 'POST'])
def create():
if request.method == 'POST':
# Handle the post request here
#get all the player names data from the form
player_names = []
for i in range(1, 5):
player_name = request.form['player' + str(i)]
#purge to long names
if len(player_name) > 20:
player_name = player_name[:20]
#sanitize the input
player_name = player_name.strip()
#replace spaces with underscores
player_name = player_name.replace(' ', '_')
# replace illegal characters with nothing
illegal_chars = ['<', '>', '(', ')', '[', ']', ',', '.', ':', ';', '-', '/', '|', '\\','=', 'exec', 'spectator']
for char in illegal_chars:
player_name = player_name.replace(char, '')
if player_name:
player_names.append(player_name)
player_names = {"player_names": player_names}
print(player_names)
response = requests.post(ozai_url + 'game', json=player_names)
if response.status_code == 200:
game_id = response.json()
print("Game ID:", game_id)
return render_template('create.html', game_id=game_id)
else:
print("Error:", response.status_code, response.text)#log the game id
#return the game id to the user
pass
return render_template('create.html')
@app.route('/join', methods=['GET', 'POST'])
def join():
if request.method == 'POST':
#get game id from the form and redirect to the join_game page
game_id = request.form['game_id']
return render_template('join.html', game_id=game_id)
return render_template('join.html')
@app.route('/join_game/<game_id>/', methods=['GET','POST'])
def join_game(game_id):
#get players from servre api and return them to the user
response = requests.get(ozai_url + 'game/' + game_id)
if response.status_code != 200:
#return error message
return 'Game not found', 404
players = response.json().get('players')
if request.method == 'POST':
#get the chosen player name from the form and redirect to the game page
player_name = request.form['player']
print('Player Name:', player_name)
return render_template('join_game.html', game_id=game_id, players=players, player_name=player_name)
return render_template('join_game.html', game_id=game_id, players=players)
@app.route('/game/<game_id>/player/<player_name>', methods=['GET', 'POST'])
def game(game_id, player_name):
gamestate = requests.get(ozai_url + 'game/' + game_id).json()
move_status = False
#if a move was made.
if request.method == 'POST':
source = request.form['source']
color = request.form['color']
destination = request.form['destination']
if source != "market":
source = int(source)-1
if destination != "floor":
destination = int(destination)-1
move = {
'player': str(player_name),
'policy': 'strict', # or 'loose' depending on your needs
'color': str(color),
'source': source,
'destination': destination
}
print(move)
response = requests.put(ozai_url + 'game/' + game_id, json=move)
if response.status_code == 200:
move_status = "Move Successful"
else:
move_status = response.text
#dont send player parameter if the player is a spectator
while True:
if player_name == 'spectator':
gamestate = requests.get(ozai_url + 'game/' + game_id).json()
return render_template('game.html', game_id=game_id, gamestate=gamestate)
elif player_name in gamestate['players']:
gamestate = requests.get(ozai_url + 'game/' + game_id + '?player=' + player_name).json()
return render_template('game.html', game_id=game_id, gamestate=gamestate, player_name=player_name, move_status=move_status)
else:
return 'Player not found', 404
return 'Player not found', 404
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=5000)