Coverage for bc/kwai-bc-training/src/kwai_bc_training/teams/team_tables.py: 100%

1"""Module that defines all dataclasses for the team tables.""" 

2 

3from dataclasses import dataclass 

4 

5from kwai_core.db.table_row import TableRow 

6 

7from kwai_bc_training.teams.team import TeamEntity, TeamIdentifier 

8 

9 

10@dataclass(kw_only=True, frozen=True, slots=True) 

11class TeamRow(TableRow): 

12 """Represent a row of the teams table.""" 

13 

14 __table_name__ = "teams" 

15 

16 id: int | None 

17 name: str | None 

18 

19 def create_entity(self) -> TeamEntity: 

20 """Create a Team entity of this row.""" 

21 return TeamEntity(id=TeamIdentifier(self.id), name=self.name) # type: ignore