feat: add player to team
This commit is contained in:
parent
ab3ed9b497
commit
a97eee842e
@ -1,5 +1,6 @@
|
|||||||
from typing import Annotated
|
from typing import Annotated
|
||||||
from fastapi import APIRouter, Depends
|
from fastapi import APIRouter, Depends, Security
|
||||||
|
from fastapi.responses import PlainTextResponse
|
||||||
from sqlmodel import Session, select
|
from sqlmodel import Session, select
|
||||||
|
|
||||||
from cutt.db import Player, Team, engine
|
from cutt.db import Player, Team, engine
|
||||||
@ -7,6 +8,8 @@ from cutt.security import change_password, get_current_active_user, read_player_
|
|||||||
|
|
||||||
P = Player
|
P = Player
|
||||||
|
|
||||||
|
player_router = APIRouter(prefix="/player", tags=["player"])
|
||||||
|
|
||||||
|
|
||||||
def add_player(player: P):
|
def add_player(player: P):
|
||||||
with Session(engine) as session:
|
with Session(engine) as session:
|
||||||
@ -14,6 +17,19 @@ def add_player(player: P):
|
|||||||
session.commit()
|
session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def add_player_to_team(player_id: int, team_id: int):
|
||||||
|
with Session(engine) as session:
|
||||||
|
player = session.exec(select(P).where(P.id == player_id)).one()
|
||||||
|
team = session.exec(select(Team).where(Team.id == team_id)).one()
|
||||||
|
if player and team:
|
||||||
|
team.players.append(player)
|
||||||
|
session.add(team)
|
||||||
|
session.commit()
|
||||||
|
return PlainTextResponse(
|
||||||
|
f"added {player.display_name} ({player.username}) to {team.name}"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def add_players(players: list[P]):
|
def add_players(players: list[P]):
|
||||||
with Session(engine) as session:
|
with Session(engine) as session:
|
||||||
for player in players:
|
for player in players:
|
||||||
@ -21,6 +37,11 @@ def add_players(players: list[P]):
|
|||||||
session.commit()
|
session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
async def list_all_players():
|
||||||
|
with Session(engine) as session:
|
||||||
|
return session.exec(select(P)).all()
|
||||||
|
|
||||||
|
|
||||||
async def list_players(team_id: int):
|
async def list_players(team_id: int):
|
||||||
with Session(engine) as session:
|
with Session(engine) as session:
|
||||||
statement = select(Team).where(Team.id == team_id)
|
statement = select(Team).where(Team.id == team_id)
|
||||||
@ -32,14 +53,35 @@ async def list_players(team_id: int):
|
|||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
async def read_teams_me(user: Annotated[P, Depends(get_current_active_user)]):
|
def read_teams_me(user: Annotated[P, Depends(get_current_active_user)]):
|
||||||
with Session(engine) as session:
|
with Session(engine) as session:
|
||||||
return [p.teams for p in session.exec(select(P).where(P.id == user.id))][0]
|
return [p.teams for p in session.exec(select(P).where(P.id == user.id))][0]
|
||||||
|
|
||||||
|
|
||||||
player_router = APIRouter(prefix="/player", tags=["player"])
|
player_router.add_api_route(
|
||||||
player_router.add_api_route("/list/{team_id}", endpoint=list_players, methods=["GET"])
|
"/add",
|
||||||
player_router.add_api_route("/add", endpoint=add_player, methods=["POST"])
|
endpoint=add_player,
|
||||||
|
methods=["POST"],
|
||||||
|
dependencies=[Security(get_current_active_user, scopes=["admin"])],
|
||||||
|
)
|
||||||
|
player_router.add_api_route(
|
||||||
|
"/list/{team_id}",
|
||||||
|
endpoint=list_players,
|
||||||
|
methods=["GET"],
|
||||||
|
dependencies=[Depends(get_current_active_user)],
|
||||||
|
)
|
||||||
|
player_router.add_api_route(
|
||||||
|
"/list",
|
||||||
|
endpoint=list_all_players,
|
||||||
|
methods=["GET"],
|
||||||
|
dependencies=[Security(get_current_active_user, scopes=["admin"])],
|
||||||
|
)
|
||||||
|
player_router.add_api_route(
|
||||||
|
"/add/{player_id}/{team_id}",
|
||||||
|
endpoint=add_player_to_team,
|
||||||
|
methods=["GET"],
|
||||||
|
dependencies=[Security(get_current_active_user, scopes=["admin"])],
|
||||||
|
)
|
||||||
player_router.add_api_route("/me", endpoint=read_player_me, methods=["GET"])
|
player_router.add_api_route("/me", endpoint=read_player_me, methods=["GET"])
|
||||||
player_router.add_api_route("/me/teams", endpoint=read_teams_me, methods=["GET"])
|
player_router.add_api_route("/me/teams", endpoint=read_teams_me, methods=["GET"])
|
||||||
player_router.add_api_route(
|
player_router.add_api_route(
|
||||||
|
Loading…
x
Reference in New Issue
Block a user