def execute()

in broadcast-monitoring/src/team_detection/app/team_text_check.py [0:0]

• 31 lines of code
• 9 McCabe index (conditional complexity)

    def execute(self, expected, data):

        team_info = TeamInfoFactory()
        expected_teams = parse_expected_teams(team_info, expected.get('Team_Info'))
        expected_teams.sort(key=lambda t: t.team_id)

        if expected_teams is None:
            logger.info('Expected Team Info not found')
            return

        expected_team_names = [team.name for team in expected_teams]
        detected_team_info = detect_team_from_text_in_image(team_info, data)

        result = {'Expected_Teams': expected_team_names, 'Detected_Teams': detected_team_info}

        # determine all keys in the detected teams dict that do NOT correspond with the
        # expected teams
        expected_ids = [team.team_id for team in expected_teams]
        non_matching_team_ids = [key for key in detected_team_info.keys() if key not in expected_ids]
        non_matching_team_ids.sort()
        logger.info(f'expected: {expected_ids}; found teams not expected: {non_matching_team_ids}')
        non_matching_team_ids = deque(non_matching_team_ids)

        for i, team in enumerate(expected_teams, 1):
            prefix = f'Team{i}_Text'
            result[f'{prefix}_Expected'] = {'id': team.team_id, 'name': team.name}
            expected_team_detected = detected_team_info.get(team.team_id)

            # the team_ids from the expected teams match the keys of the detections
            if expected_team_detected is not None:
                logger.info(f'Found expected team: {team.team_id}')
                result[f'{prefix}_Detected'] = expected_team_detected
                result[f'{prefix}_Status'] = True
            elif non_matching_team_ids:
                # team logos not matching the expected team were detected. This is a hard false
                result[f'{prefix}_Status'] = False
                non_matching_team = detected_team_info.get(non_matching_team_ids.popleft())
                result[f'{prefix}_Detected'] = non_matching_team
                logger.info(f'Found {non_matching_team} instead of team {team.team_id}')
            else:
                # when no text is detected, omit this attr.  This is an implicit None
                logger.info('No text status decided for Team%d - %s', i, team.name)

        return result