54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
|
import datetime
|
||
|
|
from typing import List
|
||
|
|
from flask import Blueprint, jsonify, request
|
||
|
|
import pydantic
|
||
|
|
from flask_pydantic import validate
|
||
|
|
from models import Player, PlayerTeam, Team, db
|
||
|
|
from middleware import requires_authentication
|
||
|
|
import models
|
||
|
|
|
||
|
|
|
||
|
|
api_team = Blueprint("team", __name__, url_prefix="/team")
|
||
|
|
|
||
|
|
@api_team.get("/view/")
|
||
|
|
@api_team.get("/view/<team_id>/")
|
||
|
|
@requires_authentication
|
||
|
|
def view(team_id = None, **kwargs):
|
||
|
|
player: Player = kwargs["player"]
|
||
|
|
|
||
|
|
q_filter = PlayerTeam.player_id == player.steam_id
|
||
|
|
if team_id is not None:
|
||
|
|
q_filter = q_filter & (PlayerTeam.team_id == team_id)
|
||
|
|
|
||
|
|
q = db.session.query(
|
||
|
|
Team
|
||
|
|
).join(
|
||
|
|
PlayerTeam
|
||
|
|
).join(
|
||
|
|
Player
|
||
|
|
).filter(
|
||
|
|
PlayerTeam.player_id == player.steam_id
|
||
|
|
)
|
||
|
|
|
||
|
|
def map_player_team_to_player_json(player_team: PlayerTeam):
|
||
|
|
return {
|
||
|
|
"steamId": player_team.player.steam_id,
|
||
|
|
"username": player_team.player.username,
|
||
|
|
}
|
||
|
|
|
||
|
|
def map_team_to_json(team: Team):
|
||
|
|
return {
|
||
|
|
"teamName": team.team_name,
|
||
|
|
"id": team.id,
|
||
|
|
"players": list(map(map_player_team_to_player_json, team.players)),
|
||
|
|
}
|
||
|
|
|
||
|
|
if team_id is None:
|
||
|
|
teams = q.all()
|
||
|
|
return jsonify(list(map(map_team_to_json, teams)))
|
||
|
|
else:
|
||
|
|
team = q.one_or_none()
|
||
|
|
if team:
|
||
|
|
return jsonify(map_team_to_json(team))
|
||
|
|
return jsonify(), 404
|