def test_get_games_to_whistle_for_all_teams(self):
gameday = DBSetup().g62_status_empty()
first_game = Gameinfo.objects.first()
first_team = Team.objects.first()
Gameinfo.objects.filter(id=first_game.pk).update(gameFinished='12:00')
Gameinfo.objects.filter(id=first_game.pk +
1).update(officials=first_team.pk + 1)
gmw = GamedayModelWrapper(gameday.pk)
assert len(gmw.get_games_to_whistle('').index) == 10
class GamedayService:
@classmethod
def create(cls, gameday_pk):
try:
return cls(gameday_pk)
except Gameinfo.DoesNotExist:
return EmptyGamedayService
def __init__(self, pk):
self.gmw = GamedayModelWrapper(pk)
def get_schedule(self):
schedule = self.gmw.get_schedule()
columns = [SCHEDULED, FIELD, HOME, POINTS_HOME, POINTS_AWAY, AWAY, OFFICIALS_NAME, STANDING, STAGE, STATUS]
schedule = schedule[columns]
schedule[OFFICIALS_NAME] = schedule[OFFICIALS_NAME].apply('<i>{}</i>'.format)
schedule[SCHEDULED] = pd.to_datetime(schedule[SCHEDULED], format='%H:%M:%S').dt.strftime('%H:%M')
schedule = schedule.rename(columns=SCHEDULE_TABLE_HEADERS)
return schedule
def get_qualify_table(self):
qualify_table = self.gmw.get_qualify_table()
if qualify_table is '':
return EmptyQualifyTable
qualify_table = qualify_table[[STANDING, TEAM_NAME, POINTS, PF, PA, DIFF]]
qualify_table = qualify_table.rename(columns=TABLE_HEADERS)
return qualify_table
def get_final_table(self):
final_table = self.gmw.get_final_table()
if final_table.empty:
return final_table
final_table = final_table[[DFFL, TEAM_NAME, POINTS, PF, PA, DIFF]]
final_table = final_table.rename(columns=TABLE_HEADERS)
return final_table
def get_games_to_whistle(self, team):
if team == '*':
team = ''
games_to_whistle = self.gmw.get_games_to_whistle(team)
columns = [SCHEDULED, FIELD, OFFICIALS_NAME, STAGE, STANDING, HOME, POINTS_HOME, POINTS_AWAY, AWAY,
STATUS, ID_HOME, ID_AWAY, 'id']
games_to_whistle = games_to_whistle[columns]
games_to_whistle = games_to_whistle.rename(columns={OFFICIALS_NAME: OFFICIALS})
return games_to_whistle
