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 util.get_csv_list(config.CONFIG.parser['favs']):
if fav in (game_rec['away']['abbrev'], game_rec['home']['abbrev']):
return True
return False
def apply_filter(game_rec, arg_filter, filters=None):
"""Returns the game_rec if the game matches the filter, or if no filtering is active.
"""
if arg_filter == 'favs':
arg_filter = config.CONFIG.parser['favs']
elif filters and arg_filter in filters:
arg_filter = filters[arg_filter]
elif not arg_filter:
return game_rec
# apply the filter
for team in util.get_csv_list(arg_filter):
if team in (game_rec['away']['abbrev'], game_rec['home']['abbrev']):
return game_rec
# no match
return None
def is_fav_by_id(team_id):
for team_abbrev in util.get_csv_list(config.CONFIG.parser['favs']):
if get_team_id(team_abbrev) == team_id:
return True
return False
def is_fav(long_team_name):
try:
return get_team_abbrev(long_team_name) in util.get_csv_list(config.CONFIG.parser['favs'])
except:
LOG.exception("Unexpected exception")
return False
def test_csv_list():
list1 = ['e1', 'e2', 'e3']
string1 = 'e1, e2, e3'
assert list1[1] == util.get_csv_list(string1)[1]
def _is_fav(long_team_name):
if long_team_name in TEAMS_TO_FAVS.keys():
return TEAMS_TO_FAVS[long_team_name] in util.get_csv_list(
config.CONFIG.parser['favs'])
return False