Coverage for kwai/modules/training/teams/team_tables.py: 100%

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

2from dataclasses import dataclass 

3 

4from kwai.core.db.table import Table 

5from kwai.modules.training.teams.team import TeamEntity, TeamIdentifier 

6 

7 

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

9class TeamRow: 

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

11 

12 id: int 

13 name: str 

14 

15 def create_entity(self) -> TeamEntity: 

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

17 return TeamEntity(id_=TeamIdentifier(self.id), name=self.name) 

18 

19 

20TeamsTable = Table("teams", TeamRow)