from uuid import UUID
from typing import List
from buyercall.blueprints.agents.models import Agent, Team


class TeamRepository:

    def save(self, team: Team) -> Team:
        return Team.save(team)

    def get_by_id(self, team_id: int) -> Team:
        return Team.query.filter(Team.id == team_id).first()

    def get_by_sid(self, team_sid: UUID) -> Team:
        return Team.query.filter(Team.sid == team_sid).first()

    def get_agents(self, team_sid: UUID) -> List[Agent]:
        team = self.get_by_sid(team_sid=team_sid)
        return team.agents if team else []