Refactor from nhlv into gamedata

.gitignore

@@ -1,4 +1,5 @@
 /.idea
+/venv
 __pycache__
 *.json
 *.log

mlbam/config.py

@@ -34,6 +34,9 @@ class Config:
             'cdn': 'akamai',
             'resolution': 'best',
             'video_player': 'mpv',
+            'streamlink_highlights': 'true',  # if false will send url direct to video_player (no resolution selection)
+            'streamlink_passthrough_highlights': 'true',  # allows seeking
+            'streamlink_passthrough': 'false',
             'audio_player': 'mpv',
             'debug': 'false',
             'verbose': 'false',

mlbam/gamedata.py

@@ -0,0 +1,285 @@
+"""
+
+"""
+import json
+import logging
+import os
+import requests
+import sys
+import time
+
+from datetime import datetime
+from datetime import timedelta
+
+import mlbam.auth as auth
+import mlbam.config as config
+import mlbam.util as util
+
+
+LOG = logging.getLogger(__name__)
+
+
+# this map is used to transform the statsweb feed name to something shorter
+FEEDTYPE_MAP = {
+    'away': 'away',
+    'home': 'home',
+    'french': 'fr',
+    'national': 'national',
+    'condensed': 'condensed',
+    'recap': 'recap',
+}
+
+
+def is_fav(game_rec):
+    if 'favourite' in game_rec:
+        return game_rec['favourite']
+    if config.CONFIG.parser['favs'] is None or config.CONFIG.parser['favs'] == '':
+        return False
+    for fav in config.CONFIG.parser['favs'].split(','):
+        fav = fav.strip()
+        if fav in (game_rec['away_abbrev'], game_rec['home_abbrev']):
+            return True
+    return False
+
+
+def filter_favs(game_rec):
+    """Returns the game_rec if the game matches the favourites, or if no filtering is active."""
+    if not config.CONFIG.parser.getboolean('filter', 'false'):
+        return game_rec
+    if config.CONFIG.parser['favs'] is None or config.CONFIG.parser['favs'] == '':
+        return game_rec
+    for fav in config.CONFIG.parser['favs'].split(','):
+        fav = fav.strip()
+        if fav in (game_rec['away_abbrev'], game_rec['home_abbrev']):
+            return game_rec
+    return None
+
+
+class GameData:
+
+    def __init__(self, feedtype_map=FEEDTYPE_MAP):
+        self.game_data_list = list()
+        self.feedtype_map = feedtype_map
+
+    def convert_feedtype_to_short(self, feedtype):
+        if feedtype in self.feedtype_map:
+            return self.feedtype_map[feedtype]
+        return feedtype
+
+    def convert_to_long_feedtype(self, feed):
+        if feed in self.feedtype_map:
+            return feed
+        for feedtype in self.feedtype_map:
+            if self.feedtype_map[feedtype] == feed:
+                return feedtype
+        return feed
+
+
+class NHLGameData(GameData):
+
+    TEAM_CODES = ('ana', 'ari', 'bos', 'buf', 'car', 'cbj', 'cgy', 'chi', 'col', 'dal', 'det', 'edm', 'fla', 'lak',
+                  'min', 'mtl', 'njd', 'nsh', 'nyi', 'nyr', 'ott', 'phi', 'pit', 'sjs', 'stl', 'tbl', 'tor', 'van',
+                  'vgk', 'wpg', 'wsh')
+
+    def __init__(self):
+        GameData.__init__(self)
+
+    def show_game_data(self, game_date, num_days=1):
+        game_data_list = list()
+        for i in range(0, num_days):
+            game_data = self.get_game_data(game_date)
+            if game_data is not None:
+                game_data_list.append(game_data)
+                if len(game_data_list) > 1:
+                    print('')  # add line feed between days
+                live_game_pks = list()
+                for game_pk in game_data:
+                    if game_data[game_pk]['abstractGameState'] == 'Live':
+                        if filter_favs(game_data[game_pk]) is not None:
+                            live_game_pks.append(game_pk)
+
+                # print header
+                date_hdr = '{:7}{}'.format('','{}'.format(game_date))
+                show_scores = config.CONFIG.parser.getboolean('scores')
+                if show_scores:
+                    print("{:63} | {:^5} | {:^9} | {}".format(date_hdr, 'Score', 'State', 'Feeds'))
+                    print("{}|{}|{}|{}".format('-' * 64, '-' * 7, '-' * 11, '-' * 12))
+                else:
+                    print("{:63} | {:^9} | {}".format(date_hdr, 'State', 'Feeds'))
+                    print("{}|{}|{}".format('-' * 64, '-' * 11, '-' * 12))
+
+                if len(live_game_pks) > 0:
+                    if show_scores:
+                        print("{:63} |{}|{}|{}".format('Live Games:', ' ' * 7, ' ' * 11, ' ' * 12))
+                    else:
+                        print("{:63} |{}|{}".format('Live Games:', ' ' * 11, ' ' * 12))
+                    for game_pk in live_game_pks:
+                        if filter_favs(game_data[game_pk]) is not None:
+                            self.show_game_details(game_pk, game_data[game_pk])
+                    if show_scores:
+                        print("{:63} |{}|{}|{}".format('-----', ' ' * 7, ' ' * 11, ' ' * 12))
+                    else:
+                        print("{:63} |{}|{}".format('-----', ' ' * 11, ' ' * 12))
+                for game_pk in game_data:
+                    if game_data[game_pk]['abstractGameState'] != 'Live':
+                        if filter_favs(game_data[game_pk]) is not None:
+                            self.show_game_details(game_pk, game_data[game_pk])
+            else:
+                LOG.info("No game data for {}".format(game_date))
+
+            game_date = datetime.strftime(datetime.strptime(game_date, "%Y-%m-%d") + timedelta(days=1), "%Y-%m-%d")
+
+        return game_data_list
+
+    @staticmethod
+    def get_game_data(date_str=None, overwrite_json=True):
+        if date_str is None:
+            date_str = time.strftime("%Y-%m-%d")
+        if config.SAVE_JSON_FILE_BY_TIMESTAMP:
+            json_file = os.path.join(config.CONFIG.dir, 'gamedata-{}.json'.format(time.strftime("%Y-%m-%d-%H%M")))
+        else:
+            json_file = os.path.join(config.CONFIG.dir, 'gamedata.json')
+        if overwrite_json or not os.path.exists(json_file):
+            LOG.debug('Getting game data...')
+            # query nhl.com for today's schedule
+            headers = {
+                'User-Agent': config.CONFIG.ua_iphone,
+                'Connection': 'close'
+            }
+            url = ('{0}/schedule?&startDate={1}&endDate={1}&expand='
+                   'schedule.teams,schedule.linescore,schedule.game.content.media.epg').format(config.CONFIG.api_url,
+                                                                                               date_str)
+            util.log_http(url, 'get', headers, sys._getframe().f_code.co_name)
+            r = requests.get(url, headers=headers, cookies=auth.load_cookies(), verify=config.VERIFY_SSL)
+
+            with open(json_file, 'w') as f:  # write date to json_file
+                f.write(r.text)
+
+        with open(json_file) as games_file:
+            json_data = json.load(games_file)
+
+        game_data = dict()  # we return this dictionary
+
+        if json_data['dates'] is None or len(json_data['dates']) < 1:
+            LOG.debug("get_game_data: no game data for {}".format(date_str))
+            return None
+
+        for game in json_data['dates'][0]['games']:
+            # LOG.debug('game: {}'.format(game))
+            game_pk_str = str(game['gamePk'])
+            game_data[game_pk_str] = dict()
+            game_rec = game_data[game_pk_str]
+            game_rec['game_pk'] = game_pk_str
+            game_rec['abstractGameState'] = str(game['status']['abstractGameState'])  # Preview, Live, Final
+            game_rec['detailedState'] = str(game['status']['detailedState'])  # is something like: Scheduled, Live, Final, In Progress, Critical
+            game_rec['nhldate'] = datetime.strptime(str(game['gameDate']), "%Y-%m-%dT%H:%M:%SZ")
+            game_rec['away_name'] = str(game['teams']['away']['team']['name'])
+            game_rec['away_abbrev'] = str(game['teams']['away']['team']['abbreviation'].lower())
+            game_rec['away_score'] = str(game['teams']['away']['score'])
+            game_rec['home_name'] = str(game['teams']['home']['team']['name'])
+            game_rec['home_abbrev'] = str(game['teams']['home']['team']['abbreviation'].lower())
+            game_rec['home_score'] = str(game['teams']['home']['score'])
+            game_rec['favourite'] = is_fav(game_rec)
+            # game_rec['nhltv_link'] = 'http://nhl.com/tv/{0}/'.format(game_pk_str)
+
+            # linescore
+            game_rec['linescore'] = dict()
+            game_rec['linescore']['currentPeriod'] = str(game['linescore']['currentPeriod'])
+            if 'currentPeriodOrdinal' in game['linescore']:
+                game_rec['linescore']['currentPeriodOrdinal'] = str(game['linescore']['currentPeriodOrdinal']) # : "2nd", "OT", "SO"
+                game_rec['linescore']['currentPeriodTimeRemaining'] = str(game['linescore']['currentPeriodTimeRemaining']) # : "18:58", "Final"
+                game_rec['linescore']['hasShootout'] = bool(game['linescore']['hasShootout'])
+            else:
+                game_rec['linescore']['currentPeriodOrdinal'] = 'Not Started'
+                game_rec['linescore']['currentPeriodTimeRemaining'] = '20:00'
+                game_rec['linescore']['hasShootout'] = False
+
+            # epg
+            game_rec['feed'] = dict()
+            for media in game['content']['media']['epg']:
+                if media['title'] == 'NHLTV':
+                    for stream in media['items']:
+                        if stream['mediaFeedType'] != 'COMPOSITE' and stream['mediaFeedType'] != 'ISO':
+                            feedtype = str(stream['mediaFeedType']).lower()  # home, away, national, french, ...
+                            game_rec['feed'][feedtype] = dict()
+                            game_rec['feed'][feedtype]['mediaPlaybackId'] = str(stream['mediaPlaybackId'])
+                            game_rec['feed'][feedtype]['eventId'] = str(stream['eventId'])
+                            game_rec['feed'][feedtype]['callLetters'] = str(stream['callLetters'])
+                elif media['title'] == 'Extended Highlights':
+                    feedtype = 'condensed'
+                    if len(media['items']) > 0:
+                        game_rec['feed'][feedtype] = dict()
+                        stream = media['items'][0]
+                        game_rec['feed'][feedtype]['mediaPlaybackId'] = str(stream['mediaPlaybackId'])
+                        for playback_item in stream['playbacks']:
+                            if playback_item['name'] == config.CONFIG.playback_scenario:
+                                game_rec['feed'][feedtype]['playback_url'] = playback_item['url']
+                elif media['title'] == 'Recap':
+                    feedtype = 'recap'
+                    if len(media['items']) > 0:
+                        game_rec['feed'][feedtype] = dict()
+                        stream = media['items'][0]
+                        game_rec['feed'][feedtype]['mediaPlaybackId'] = str(stream['mediaPlaybackId'])
+                        for playback_item in stream['playbacks']:
+                            if playback_item['name'] == config.CONFIG.playback_scenario:
+                                game_rec['feed'][feedtype]['playback_url'] = playback_item['url']
+        return game_data
+
+    def show_game_details(self, game_pk, game_rec):
+        color_on = ''
+        color_off = ''
+        if is_fav(game_rec):
+            if config.CONFIG.parser['fav_colour'] != '':
+                color_on = util.fg_ansi_colour(config.CONFIG.parser['fav_colour'])
+                color_off = util.ANSI_CONTROL_CODES['reset']
+        show_scores = config.CONFIG.parser.getboolean('scores')
+        game_info_str = "{}: {} ({}) at {} ({})".format(util.convert_time_to_local(game_rec['nhldate']),
+                                                        game_rec['away_name'], game_rec['away_abbrev'].upper(),
+                                                        game_rec['home_name'], game_rec['home_abbrev'].upper())
+        game_state = ''
+        game_state_color_on = color_on
+        game_state_color_off = color_off
+        if game_rec['abstractGameState'] not in ('Preview', ):
+            if not show_scores:
+                game_state = game_rec['abstractGameState']
+                if 'In Progress - ' in game_rec['detailedState']:
+                    game_state = game_rec['detailedState'].split('In Progress - ')[-1]
+                elif game_rec['detailedState'] not in ('Live', 'Final', 'Scheduled', 'In Progress'):
+                    game_state = game_rec['detailedState']
+            else:
+                if 'Critical' in game_rec['detailedState']:
+                    game_state_color_on = util.fg_ansi_colour(config.CONFIG.parser['game_critical_colour'])
+                    game_state_color_off = util.ANSI_CONTROL_CODES['reset']
+                if game_rec['linescore']['currentPeriodTimeRemaining'] == 'Final' \
+                        and game_rec['linescore']['currentPeriodOrdinal'] == '3rd':
+                    game_state = 'Final'
+                else:
+                    game_state = '{} {}'.format(game_rec['linescore']['currentPeriodTimeRemaining'].title(),
+                                                game_rec['linescore']['currentPeriodOrdinal'])
+        # else:
+        #    game_state = 'Pending'
+        short_feeds = list()
+        for feed in game_rec['feed'].keys():
+            short_feeds.append(self.convert_feedtype_to_short(feed))
+        short_feed_str = ', '.join(sorted(short_feeds))
+        if config.CONFIG.parser.getboolean('scores'):
+            score = ''
+            if game_rec['abstractGameState'] not in ('Preview', ):
+                score = '{}-{}'.format(game_rec['away_score'], game_rec['home_score'])
+            print("{0}{2:<63}{1} | {0}{3:^5}{1} | {4}{5:<9}{6} | {0}{7}{1}".format(color_on, color_off,
+                                                                                   game_info_str, score,
+                                                                               game_state_color_on,
+                                                                               game_state,
+                                                                               game_state_color_off,
+                                                                               short_feed_str))
+                                                                               #', '.join(sorted(game_rec['feed'].keys()))))
+        else:
+            print("{0}{2:<63}{1} | {0}{3:^9}{1} | {0}{4}{1}".format(color_on, color_off,
+                                                                    game_info_str, game_state,
+                                                                    ', '.join(sorted(game_rec['feed'].keys()))))
+        if config.CONFIG.parser.getboolean('debug') and config.CONFIG.parser.getboolean('verbose'):
+            for feedtype in game_rec['feed']:
+                print('    {}: {}  [game_pk:{}, mediaPlaybackId:{}]'.format(feedtype,
+                                                                            game_rec['abstractGameState'],
+                                                                            game_pk,
+                                                                            game_rec['feed'][feedtype]['mediaPlaybackId']))