add broadcasts to single game queries
This commit is contained in:
cottongin
parent
b3eba18918
commit
7387c2d6c5
128
plugin.py
128
plugin.py
|
|
@ -10,7 +10,7 @@ import pendulum
|
|||
import requests
|
||||
import collections
|
||||
|
||||
from supybot import utils, plugins, ircutils, callbacks, conf, schedule
|
||||
from supybot import utils, plugins, ircutils, callbacks, conf, schedule, ircmsgs
|
||||
from supybot.commands import *
|
||||
try:
|
||||
from supybot.i18n import PluginInternationalization
|
||||
|
|
@ -22,7 +22,7 @@ except ImportError:
|
|||
|
||||
SCOREBOARD = ('http://site.api.espn.com/apis/site/v2/sports/basketball/'
|
||||
'mens-college-basketball/scoreboard?lang=en®ion=us'
|
||||
'&calendartype=blacklist&limit=300&dates={date}')
|
||||
'&calendartype=blacklist&limit=300&groups=50&dates={date}')
|
||||
|
||||
class CBBScores(callbacks.Plugin):
|
||||
"""Fetches College Basketball scores"""
|
||||
|
|
@ -36,14 +36,14 @@ class CBBScores(callbacks.Plugin):
|
|||
self.SCORES = self._checkscores()
|
||||
self.SCORES = self._checkscores()
|
||||
try: # check scores.
|
||||
schedule.addPeriodicEvent(checkcbbscores, 10,
|
||||
schedule.addPeriodicEvent(checkcbbscores, 20,
|
||||
now=False, name='checkcbbscores')
|
||||
except AssertionError:
|
||||
try:
|
||||
schedule.removeEvent('checkcbbscores')
|
||||
except KeyError:
|
||||
pass
|
||||
schedule.addPeriodicEvent(checkcbbscores, 10,
|
||||
schedule.addPeriodicEvent(checkcbbscores, 20,
|
||||
now=False, name='checkcbbscores')
|
||||
|
||||
def die(self):
|
||||
|
|
@ -65,6 +65,9 @@ class CBBScores(callbacks.Plugin):
|
|||
Defaults to today and all teams if no input given.
|
||||
Ex: --date 20181117 MICH
|
||||
"""
|
||||
channel = msg.args[0]
|
||||
if channel == irc.nick:
|
||||
channel = msg.nick
|
||||
options = dict(options)
|
||||
date = options.get('date')
|
||||
if date:
|
||||
|
|
@ -77,7 +80,7 @@ class CBBScores(callbacks.Plugin):
|
|||
date = pendulum.now().format('YYYYMMDD')
|
||||
else:
|
||||
try:
|
||||
pendulum.parse(date, strict=False).format('YYYYMMDD')
|
||||
date = pendulum.parse(date, strict=False).format('YYYYMMDD')
|
||||
except:
|
||||
irc.reply('Invalid date format')
|
||||
return
|
||||
|
|
@ -87,19 +90,42 @@ class CBBScores(callbacks.Plugin):
|
|||
if date not in self.SCORES:
|
||||
# fetch another day
|
||||
print('date not in scores db')
|
||||
pass
|
||||
SCORES = self._checkscores(cdate=date)
|
||||
else:
|
||||
if team:
|
||||
SCORES = self.SCORES
|
||||
|
||||
if team:
|
||||
if len(team) > 2:
|
||||
reply = []
|
||||
# single team
|
||||
for key,value in self.SCORES[date].items():
|
||||
if team.lower() in key.lower():
|
||||
irc.reply(value['long'])
|
||||
return
|
||||
pass
|
||||
else:
|
||||
# all teams
|
||||
irc.reply(' | '.join(value['short'] for item,value in self.SCORES[date].items()))
|
||||
for key,value in SCORES[date].items():
|
||||
if team.lower() in value['lookup']['abbr'].lower():
|
||||
#print(team.lower(), '\t', value['lookup']['abbr'].lower())
|
||||
reply.append(value['long'])
|
||||
#break
|
||||
if not reply:
|
||||
for key,value in SCORES[date].items():
|
||||
if team.lower() in value['lookup']['full'].lower():
|
||||
reply.append(value['long'])
|
||||
#break
|
||||
if not reply:
|
||||
irc.reply('ERROR: no match found for your input: {}'.format(team))
|
||||
return
|
||||
else:
|
||||
if len(reply) <= 4:
|
||||
for item in reply:
|
||||
irc.sendMsg(ircmsgs.privmsg(channel, item))
|
||||
else:
|
||||
for item in reply:
|
||||
irc.reply(item)
|
||||
return
|
||||
else:
|
||||
irc.reply('ERROR: search string too short')
|
||||
return
|
||||
else:
|
||||
# all teams
|
||||
irc.sendMsg(ircmsgs.privmsg(channel, ' | '.join(value['short'] for item,value in SCORES[date].items() if value['top25'])))
|
||||
return
|
||||
|
||||
return
|
||||
|
||||
|
|
@ -107,12 +133,18 @@ class CBBScores(callbacks.Plugin):
|
|||
# INTERNAL FUNCTIONS #
|
||||
######################
|
||||
|
||||
def _checkscores(self):
|
||||
today = pendulum.now().format('YYYYMMDD')
|
||||
yesterday = pendulum.yesterday().format('YYYYMMDD')
|
||||
tomorrow = pendulum.tomorrow().format('YYYYMMDD')
|
||||
|
||||
dates = [yesterday, today, tomorrow]
|
||||
def _checkscores(self, cdate=None):
|
||||
if cdate:
|
||||
#today = pendulum.parse(cdate, strict=False).format('YYYYMMDD')
|
||||
#yesterday = pendulum.parse(cdate, strict=False).subtract(days=1).format('YYYYMMDD')
|
||||
#tomorrow = pendulum.parse(cdate, strict=False).add(days=1).format('YYYYMMDD')
|
||||
dates = [cdate]
|
||||
else:
|
||||
today = pendulum.now().format('YYYYMMDD')
|
||||
yesterday = pendulum.yesterday().format('YYYYMMDD')
|
||||
tomorrow = pendulum.tomorrow().format('YYYYMMDD')
|
||||
dates = [yesterday, today, tomorrow]
|
||||
|
||||
data = collections.OrderedDict()
|
||||
for date in dates:
|
||||
tmp = requests.get(SCOREBOARD.format(date=date)).json()
|
||||
|
|
@ -131,51 +163,77 @@ class CBBScores(callbacks.Plugin):
|
|||
if d:
|
||||
games[day] = collections.OrderedDict()
|
||||
for event in d:
|
||||
key = '{} | {}'.format(event['name'], event['shortName'])
|
||||
key = event['id']
|
||||
lookup = {'abbr': '{}'.format(event['shortName']),
|
||||
'full': '{}'.format(event['name'])}
|
||||
comp = event['competitions'][0]
|
||||
time = pendulum.parse(comp['date'], strict=False).in_tz('US/Eastern')
|
||||
short_time = time.format('h:mm A zz')
|
||||
long_time = time.format('dddd, MMM Do, h:mm A zz')
|
||||
status = comp['status']['type']['state']
|
||||
is_ended = comp['status']['type']['completed']
|
||||
top25 = True if (0 < comp['competitors'][0]['curatedRank']['current'] <= 25
|
||||
or 0 < comp['competitors'][1]['curatedRank']['current'] <= 25) else False
|
||||
home_rank = '(#{})'.format(comp['competitors'][0]['curatedRank']['current']) \
|
||||
if 0 < comp['competitors'][0]['curatedRank']['current'] <= 25 else ''
|
||||
away_rank = '(#{})'.format(comp['competitors'][1]['curatedRank']['current']) \
|
||||
if 0 < comp['competitors'][1]['curatedRank']['current'] <= 25 else ''
|
||||
home_short = comp['competitors'][0]['team']['abbreviation']
|
||||
home_long = comp['competitors'][0]['team']['displayName']
|
||||
away_short = comp['competitors'][1]['team']['abbreviation']
|
||||
away_long = comp['competitors'][1]['team']['displayName']
|
||||
home_score = int(comp['competitors'][0]['score'])
|
||||
away_score = int(comp['competitors'][1]['score'])
|
||||
#print(home_short, away_short, ' || ', is_ended, status)
|
||||
# strings for pre/in games
|
||||
broadcasts = []
|
||||
try:
|
||||
for thing in comp['broadcasts']:
|
||||
for station in thing['names']:
|
||||
broadcasts.append(station)
|
||||
except:
|
||||
pass
|
||||
#print(home_short, away_short, '\t||\t', top25, status, comp['competitors'][0]['curatedRank']['current'],
|
||||
# comp['competitors'][1]['curatedRank']['current'])
|
||||
if status == 'pre':
|
||||
# pre
|
||||
short = '{} @ {} {}'.format(away_short, home_short, short_time)
|
||||
long = '{} @ {} | {}'.format(away_long, home_long, long_time)
|
||||
long = '{}{} @ {}{} | {}{}'.format(away_long, away_rank, home_long, home_rank, long_time,
|
||||
" [TV: {}]".format(", ".join(broadcasts) if broadcasts else "")
|
||||
)
|
||||
else:
|
||||
# inp
|
||||
if is_ended:
|
||||
clock_short = ircutils.mircColor('F', 'red')
|
||||
clock_long = ircutils.mircColor('Final', 'red')
|
||||
else:
|
||||
clock_short = ircutils.mircColor(comp['status']['type']['shortDetail'].replace(' - ', ' '), 'green')
|
||||
clock_long = ircutils.mircColor(comp['status']['type']['detail'], 'green')
|
||||
if 'Halftime' in comp['status']['type']['detail']:
|
||||
clock_short = ircutils.mircColor('HT', 'orange')
|
||||
clock_long = ircutils.mircColor('Halftime', 'orange')
|
||||
else:
|
||||
clock_short = ircutils.mircColor(comp['status']['type']['shortDetail'].replace(' - ', ' '), 'green')
|
||||
clock_long = ircutils.mircColor(comp['status']['type']['detail'], 'green')
|
||||
try:
|
||||
last_play = ' | \x02Last Play:\x02 {}'.format(comp['situation']['lastPlay']['text']) \
|
||||
if 'situation' in comp else ''
|
||||
except:
|
||||
last_play = ''
|
||||
if away_score > home_score:
|
||||
away_short_str = ircutils.bold('{} {}'.format(away_short, away_score))
|
||||
away_long_str = ircutils.bold('{} {}'.format(away_long, away_score))
|
||||
away_long_str = ircutils.bold('{}{} {}'.format(away_long, away_rank, away_score))
|
||||
home_short_str = '{} {}'.format(home_short, home_score)
|
||||
home_long_str = '{} {}'.format(home_long, home_score)
|
||||
home_long_str = '{}{} {}'.format(home_long, home_rank, home_score)
|
||||
elif home_score > away_score:
|
||||
away_short_str = '{} {}'.format(away_short, away_score)
|
||||
away_long_str = '{} {}'.format(away_long, away_score)
|
||||
away_long_str = '{}{} {}'.format(away_long, away_rank, away_score)
|
||||
home_short_str = ircutils.bold('{} {}'.format(home_short, home_score))
|
||||
home_long_str = ircutils.bold('{} {}'.format(home_long, home_score))
|
||||
home_long_str = ircutils.bold('{}{} {}'.format(home_long, home_rank, home_score))
|
||||
else:
|
||||
away_short_str = '{} {}'.format(away_short, away_score)
|
||||
away_long_str = '{} {}'.format(away_long, away_score)
|
||||
away_long_str = '{}{} {}'.format(away_long, away_rank, away_score)
|
||||
home_short_str = '{} {}'.format(home_short, home_score)
|
||||
home_long_str = '{} {}'.format(home_long, home_score)
|
||||
home_long_str = '{}{} {}'.format(home_long, home_rank, home_score)
|
||||
short = '{} {} {}'.format(away_short_str, home_short_str, clock_short)
|
||||
long = '{} @ {} - {}'.format(away_long_str, home_long_str, clock_long)
|
||||
games[day][key] = {'short': short, 'long': long, 'ended': is_ended}
|
||||
long = '{} @ {} - {}{}'.format(away_long_str, home_long_str, clock_long, last_play)
|
||||
games[day][key] = {'short': short, 'long': long, 'ended': is_ended, 'top25': top25, 'lookup': lookup}
|
||||
|
||||
# sort events
|
||||
games[day] = collections.OrderedDict(sorted(games[day].items(), key=lambda k: k[1]['ended']))
|
||||
|
|
|
|||
Loading…
Reference in New Issue