Ejemplo n.º 1
0
    def get_game_play_by_play_dataframe_v7(event_id, game_unit,
                                           period_id_list):
        pbp_json_list = [
            FibaCommunicator.get_game_play_by_play_v7(
                event_id, game_unit, p)['content']['full']['Items']
            for p in period_id_list
        ]

        df = pd.DataFrame(sum(pbp_json_list, []))
        update_xy_v7(df)
        update_pbp_stats_v7(df)
        return df
Ejemplo n.º 2
0
    def get_game_stats_dataframe_v7(event_id, game_unit):
        game_json = FibaCommunicator.get_game_team_stats_v7(
            event_id, game_unit)
        team_stats_json = game_json['content']['full']['Competitors']

        for t in team_stats_json:
            #Team Stats
            t['Stats']['Name'] = t['Name']
            t['Stats']['TeamCode'] = t['TeamCode']
            t['Stats']['Periods'] = t['Periods']
            t['Stats']['PeriodIdList'] = [p['Id'] for p in t['Periods']]
            t['Stats']['SECS'] = 60 * 5 * game_time(len(t['Periods']))
            t['Stats']['TP'] = base60_to(t['Stats']['SECS'])

            #Player Stats
            for p in t['Children']:
                p['Stats']['TeamCode'] = t['TeamCode']
                p['Stats']['TeamId'] = t['Id']
                p['Stats']['JerseyNumber'] = p['JerseyNumber']
                p['Stats']['Name'] = p['Name']

        team_a_stats_json = team_stats_json[0]['Stats']
        team_b_stats_json = team_stats_json[1]['Stats']

        team_a_player_stats_list = [
            p['Stats'] for p in team_stats_json[0]['Children']
        ]
        team_b_player_stats_list = [
            p['Stats'] for p in team_stats_json[1]['Children']
        ]

        #Oppenent DREB for calculating OREB%
        team_a_stats_json['OPP_DR'] = team_b_stats_json['DR']
        team_b_stats_json['OPP_DR'] = team_a_stats_json['DR']

        team_a_stats_json['OppTeamCode'] = team_b_stats_json['TeamCode']
        team_b_stats_json['OppTeamCode'] = team_a_stats_json['TeamCode']

        team_stats_df = pd.DataFrame([team_a_stats_json, team_b_stats_json])
        player_stats_df = pd.DataFrame(team_a_player_stats_list +
                                       team_b_player_stats_list)
        update_secs_v7(player_stats_df)

        return team_stats_df, player_stats_df
Ejemplo n.º 3
0
    def get_game_details_dict_v7(event_id, game_unit):
        dtl_dict = FibaCommunicator.get_game_details_v7(
            event_id, game_unit)['content']['full']['Competitors']

        id_table = {
            k: ((v['TeamCode']) if v['IsTeam'] else
                ('{num} {short}{sur}'.format(num=v['Bib'].zfill(2),
                                             short=v['FirstNameShort'],
                                             sur=v['Name'])))
            for k, v in dtl_dict.iteritems()
        }

        team_id_list = [k for k, v in dtl_dict.iteritems() if v['IsTeam']]
        starter_dict = {
            t: {
                v['Id']
                for v in dtl_dict.itervalues()
                if (not v['IsTeam']) and v['Starter'] and v['ParentId'] == t
            }
            for t in team_id_list
        }

        return id_table, starter_dict
Ejemplo n.º 4
0
    def get_game_data_dataframe_v5(match_id):
        game_json = FibaCommunicator.get_game_data_v5(match_id)
        team_stats_json = game_json['tm']

        for t in team_stats_json.values():
            #Team Stats
            t['Name'] = t['nameInternational']
            t['TeamCode'] = t['codeInternational']
            t['Periods'] = [
                {
                    'Id': period.replace('_score', '').replace('p',
                                                               'q').upper(),
                    'Score': score
                } for period, score in t.iteritems()
                if period.startswith('p') and period.endswith('_score')
            ]
            if 'ot_score' in t:
                t['Periods'].append({'Id': 'OT', 'Score': t['ot_score']})
            t['PeriodIdList'] = [p['Id'] for p in t['Periods']]

            if base60_from(t['tot_sMinutes']) == 0:
                team_secs = sum(
                    [base60_from(p['sMinutes']) for p in t['pl'].values()])
                t['tot_sMinutes'] = base60_to(team_secs)

            #Player Stats
            for p in t['pl'].values():
                p['TeamCode'] = t['codeInternational']
                p['JerseyNumber'] = p['shirtNumber']
                p['Name'] = p['name'].replace(' ', '').upper()
                p['NumName'] = '{num} {name}'.format(
                    num=p['JerseyNumber'].zfill(2), name=p['Name'])

        team_a_stats_json = team_stats_json['1']
        team_b_stats_json = team_stats_json['2']

        team_a_player_stats_list = [
            p for p in team_stats_json['1']['pl'].values()
        ]
        team_b_player_stats_list = [
            p for p in team_stats_json['2']['pl'].values()
        ]

        #Oppenent DREB for calculating OREB%
        team_a_stats_json['OPP_DR'] = team_b_stats_json[
            'tot_sReboundsDefensive']
        team_b_stats_json['OPP_DR'] = team_a_stats_json[
            'tot_sReboundsDefensive']

        team_a_stats_json['OppTeamCode'] = team_b_stats_json['TeamCode']
        team_b_stats_json['OppTeamCode'] = team_a_stats_json['TeamCode']

        team_stats_df = pd.DataFrame([team_a_stats_json, team_b_stats_json])
        player_stats_df = pd.DataFrame(team_a_player_stats_list +
                                       team_b_player_stats_list)
        update_team_stats_v5_to_v7(team_stats_df)
        update_player_stats_v5_to_v7(player_stats_df)

        starter_dict = {
            t['TeamCode']:
            {p['NumName']
             for p in t['pl'].values() if p['starter'] == 1}
            for t in team_stats_json.values()
        }

        pbp_df = pd.DataFrame(reversed(game_json['pbp']))
        update_pbp_stats_v5_to_v7(pbp_df, team_a_stats_json['TeamCode'],
                                  team_b_stats_json['TeamCode'])

        shot_df = pd.DataFrame(
            sum([t['shot'] for t in team_stats_json.values()], []))
        update_pbp_stats_v5_to_v7(shot_df, team_a_stats_json['TeamCode'],
                                  team_b_stats_json['TeamCode'])
        update_xy_v5(shot_df)

        return team_stats_df, player_stats_df, starter_dict, pbp_df, shot_df
Ejemplo n.º 5
0
def test_get_game_team_stats_v7():
    result = FibaCommunicator.get_game_team_stats_v7(event_id='208053',
                                                     game_unit='24532-18-A')
    assert result['content']['full']['Competitors'][0]['Name'] == 'Pauian'
Ejemplo n.º 6
0
def test_get_game_data_v5():
    result = FibaCommunicator.get_game_data_v5(match_id='1143749')
    assert result['tm']['1']['name'] == 'Singapore Slingers'
Ejemplo n.º 7
0
def test_get_game_details_v7():
    result = FibaCommunicator.get_game_details_v7(event_id='208153',
                                                  game_unit='25098-A-3')
    assert result['content']['full']['Competitors']['T_57840'][
        'Name'] == 'Fubon Braves'
Ejemplo n.º 8
0
def test_get_game_play_by_play_v7():
    result = FibaCommunicator.get_game_play_by_play_v7(event_id='208053',
                                                       game_unit='24532-18-A',
                                                       period_id='Q1')
    assert result['content']['full']['Items'][0]['Action'] == 'Start of Game'