diff --git a/api/games/api_games.py b/api/games/api_games.py index a4853bb..b04c9b4 100644 --- a/api/games/api_games.py +++ b/api/games/api_games.py @@ -19,7 +19,12 @@ def get_room(game_id): # TODO create decorator that returns user from header auth_header = request.headers.get('Authorization') user = jwt.decode(auth_header.split(" ")[1], os.environ.get('SECRET_KEY'))['user'] - print(user) + user = json.loads(user) + if not game.player_black and game.player_white != user['id']: + game.player_black = user['id'] + db.session.add(game) + db.session.commit() + print(game_schema.dumps(game)) join_game_notice(game_id, user) return jsonify(response) diff --git a/models/Game.py b/models/Game.py index 16ac68c..3789042 100644 --- a/models/Game.py +++ b/models/Game.py @@ -80,6 +80,8 @@ class GameSchema(ma.ModelSchema): board_size = fields.Int() player = fields.Nested(user_schema) game_room = fields.Int() + player_black = fields.Int() + player_white = fields.Int() game_schema = GameSchema() games_schema = GameSchema(many=True) \ No newline at end of file