initial commit
This commit is contained in:
cottongin
commit
aab0c5bbe9
|
|
@ -0,0 +1,49 @@
|
||||||
|
###
|
||||||
|
# Copyright (c) 2018, cottongin
|
||||||
|
# All rights reserved.
|
||||||
|
#
|
||||||
|
#
|
||||||
|
###
|
||||||
|
|
||||||
|
"""
|
||||||
|
GolfScores: Fetches golf scores
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import supybot
|
||||||
|
from supybot import world
|
||||||
|
|
||||||
|
# Use this for the version of this plugin. You may wish to put a CVS keyword
|
||||||
|
# in here if you're keeping the plugin in CVS or some similar system.
|
||||||
|
__version__ = ""
|
||||||
|
|
||||||
|
# XXX Replace this with an appropriate author or supybot.Author instance.
|
||||||
|
__author__ = supybot.authors.unknown
|
||||||
|
|
||||||
|
# This is a dictionary mapping supybot.Author instances to lists of
|
||||||
|
# contributions.
|
||||||
|
__contributors__ = {}
|
||||||
|
|
||||||
|
# This is a url where the most recent plugin package can be downloaded.
|
||||||
|
__url__ = ''
|
||||||
|
|
||||||
|
from . import config
|
||||||
|
from . import plugin
|
||||||
|
if sys.version_info >= (3, 4):
|
||||||
|
from importlib import reload
|
||||||
|
else:
|
||||||
|
from imp import reload
|
||||||
|
# In case we're being reloaded.
|
||||||
|
reload(config)
|
||||||
|
reload(plugin)
|
||||||
|
# Add more reloads here if you add third-party modules and want them to be
|
||||||
|
# reloaded when this plugin is reloaded. Don't forget to import them as well!
|
||||||
|
|
||||||
|
if world.testing:
|
||||||
|
from . import test
|
||||||
|
|
||||||
|
Class = plugin.Class
|
||||||
|
configure = config.configure
|
||||||
|
|
||||||
|
|
||||||
|
# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79:
|
||||||
|
|
@ -0,0 +1,33 @@
|
||||||
|
###
|
||||||
|
# Copyright (c) 2018, cottongin
|
||||||
|
# All rights reserved.
|
||||||
|
#
|
||||||
|
#
|
||||||
|
###
|
||||||
|
|
||||||
|
from supybot import conf, registry
|
||||||
|
try:
|
||||||
|
from supybot.i18n import PluginInternationalization
|
||||||
|
_ = PluginInternationalization('GolfScores')
|
||||||
|
except:
|
||||||
|
# Placeholder that allows to run the plugin on a bot
|
||||||
|
# without the i18n module
|
||||||
|
_ = lambda x: x
|
||||||
|
|
||||||
|
|
||||||
|
def configure(advanced):
|
||||||
|
# This will be called by supybot to configure this module. advanced is
|
||||||
|
# a bool that specifies whether the user identified themself as an advanced
|
||||||
|
# user or not. You should effect your configuration by manipulating the
|
||||||
|
# registry as appropriate.
|
||||||
|
from supybot.questions import expect, anything, something, yn
|
||||||
|
conf.registerPlugin('GolfScores', True)
|
||||||
|
|
||||||
|
|
||||||
|
GolfScores = conf.registerPlugin('GolfScores')
|
||||||
|
# This is where your configuration variables (if any) should go. For example:
|
||||||
|
# conf.registerGlobalValue(GolfScores, 'someConfigVariableName',
|
||||||
|
# registry.Boolean(False, _("""Help for someConfigVariableName.""")))
|
||||||
|
|
||||||
|
|
||||||
|
# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79:
|
||||||
|
|
@ -0,0 +1,168 @@
|
||||||
|
###
|
||||||
|
# Copyright (c) 2018, cottongin
|
||||||
|
# All rights reserved.
|
||||||
|
#
|
||||||
|
#
|
||||||
|
###
|
||||||
|
|
||||||
|
import pendulum
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from supybot import utils, plugins, ircutils, callbacks
|
||||||
|
from supybot.commands import *
|
||||||
|
try:
|
||||||
|
from supybot.i18n import PluginInternationalization
|
||||||
|
_ = PluginInternationalization('GolfScores')
|
||||||
|
except ImportError:
|
||||||
|
# Placeholder that allows to run the plugin on a bot
|
||||||
|
# without the i18n module
|
||||||
|
_ = lambda x: x
|
||||||
|
|
||||||
|
CURRENT_URL = 'https://statdata.pgatour.com/{trn_type}/current/message.json'
|
||||||
|
SCOREBOARD = 'https://statdata.pgatour.com/{trn_type}/{trn_id}/leaderboard-v2mini.json'
|
||||||
|
|
||||||
|
class GolfScores(callbacks.Plugin):
|
||||||
|
"""Fetches golf scores"""
|
||||||
|
threaded = True
|
||||||
|
|
||||||
|
def _fetchCurrent(self, type_='r'):
|
||||||
|
tmp = None
|
||||||
|
try:
|
||||||
|
jdata = requests.get(CURRENT_URL.format(trn_type=type_)).json()
|
||||||
|
tmp = jdata['tid']
|
||||||
|
return [type_, tmp]
|
||||||
|
except:
|
||||||
|
return [type_, tmp]
|
||||||
|
return [type_, tmp]
|
||||||
|
|
||||||
|
@wrap([getopts({'all': '', 'info': '', 'champions': ''}), optional('text')])
|
||||||
|
def golf(self, irc, msg, args, options, search=None):
|
||||||
|
""" [player name]
|
||||||
|
Fetches leaderboard for active or previous tournament,
|
||||||
|
optionally pass a name to find player in standings
|
||||||
|
"""
|
||||||
|
options = dict(options)
|
||||||
|
type_is_champions = options.get('champions')
|
||||||
|
|
||||||
|
if type_is_champions:
|
||||||
|
trn = self._fetchCurrent(type_='s')
|
||||||
|
else:
|
||||||
|
trn = self._fetchCurrent()
|
||||||
|
if not trn[1]:
|
||||||
|
irc.reply('Something went wrong getting the current/previous tournament')
|
||||||
|
return
|
||||||
|
|
||||||
|
if options.get('info'):
|
||||||
|
url = 'https://www.pgatour.com/bin/data/feeds/weather.json/{}{}'.format(
|
||||||
|
trn[0], trn[1])
|
||||||
|
print(url)
|
||||||
|
idata = requests.get(url).json()
|
||||||
|
url2 = 'https://statdata.pgatour.com/r/current/schedule-v2.json' #.format(trn[0])
|
||||||
|
#print(url2)
|
||||||
|
sdata = requests.get(url2).json()
|
||||||
|
#print(sdata)
|
||||||
|
|
||||||
|
# now get the leaderboard json
|
||||||
|
try:
|
||||||
|
jdata = requests.get(SCOREBOARD.format(trn_type=trn[0], trn_id=trn[1]))
|
||||||
|
print(jdata.url)
|
||||||
|
jdata = jdata.json()
|
||||||
|
except:
|
||||||
|
irc.reply('Something went wrong fetching the leaderboard')
|
||||||
|
return
|
||||||
|
|
||||||
|
leaderboard = jdata.get('leaderboard')
|
||||||
|
if not leaderboard:
|
||||||
|
irc.reply('No leaderboard found')
|
||||||
|
return
|
||||||
|
|
||||||
|
name = ircutils.bold(leaderboard['tournament_name'])
|
||||||
|
date = "{}-{}".format(
|
||||||
|
pendulum.parse(leaderboard['start_date'], strict=False).format('MMMD'),
|
||||||
|
pendulum.parse(leaderboard['end_date'], strict=False).format('MMMD'))
|
||||||
|
|
||||||
|
round_ = 'Round {}'.format(leaderboard['current_round'])
|
||||||
|
if leaderboard['round_state']:
|
||||||
|
round_ += ' ({})'.format(leaderboard['round_state'])
|
||||||
|
|
||||||
|
cut_line = leaderboard['cut_line'].get('cut_count') or len(leaderboard['players'])
|
||||||
|
|
||||||
|
positions = []
|
||||||
|
if not options.get('info'):
|
||||||
|
for idx, player in enumerate(leaderboard['players']):
|
||||||
|
if player['player_bio']['short_name']:
|
||||||
|
plyr_name = '{}.{}'.format(player['player_bio']['short_name'].replace('.', ''),
|
||||||
|
player['player_bio']['last_name'])
|
||||||
|
else:
|
||||||
|
plyr_name = '{}'.format(player['player_bio']['last_name'])
|
||||||
|
full_name = '{} {}'.format(player['player_bio']['first_name'],
|
||||||
|
player['player_bio']['last_name'])
|
||||||
|
if idx >= cut_line:
|
||||||
|
if player['status'] == 'wd':
|
||||||
|
rank = ircutils.mircColor('WD', 'orange')
|
||||||
|
else:
|
||||||
|
rank = ircutils.mircColor('CUT', 'red')
|
||||||
|
else:
|
||||||
|
rank = str(player['current_position'])
|
||||||
|
if player['thru']:
|
||||||
|
thru = ' {:+d} thru {} '.format(player['today'], ircutils.mircColor(str(player['thru']), 'green')) \
|
||||||
|
if player['thru'] != 18 else ' {:+d}'.format(player['today']) + ircutils.bold(ircutils.mircColor(' F ', 'red'))
|
||||||
|
else:
|
||||||
|
thru = ' '
|
||||||
|
score = '{:+d}'.format(player['total']) if player['total'] else '-'
|
||||||
|
string = '{} {}{}({})'.format(
|
||||||
|
ircutils.bold(ircutils.mircColor(rank, 'blue')),
|
||||||
|
plyr_name,
|
||||||
|
thru,
|
||||||
|
score)
|
||||||
|
if search:
|
||||||
|
if search.lower() in full_name.lower():
|
||||||
|
positions.append(string)
|
||||||
|
else:
|
||||||
|
positions.append(string)
|
||||||
|
|
||||||
|
if not positions:
|
||||||
|
positions.append('Player not found')
|
||||||
|
#return
|
||||||
|
|
||||||
|
if options.get('info'):
|
||||||
|
loc = idata['current_observation']['display_location']['full']
|
||||||
|
weather = idata['current_observation']['weather']
|
||||||
|
try:
|
||||||
|
temp = idata['current_observation']['temperature_string']
|
||||||
|
wind = idata['current_observation']['wind_string']
|
||||||
|
except:
|
||||||
|
temp = "{}F".format(idata['current_observation']['temp_f'])
|
||||||
|
wind = "{}mph {}".format(idata['current_observation']['wind_mph'], idata['current_observation']['wind_dir'])
|
||||||
|
w_string = '{} :: {} - {} - Wind: {}'.format(loc, weather, temp, wind)
|
||||||
|
year = sdata['currentYears'][trn[0]]
|
||||||
|
for item in sdata['years']:
|
||||||
|
if item['year'] == year:
|
||||||
|
data = item['tours']
|
||||||
|
for t in data:
|
||||||
|
if t['tourCodeLc'] == trn[0]:
|
||||||
|
tdata = t['trns']
|
||||||
|
for tour in tdata:
|
||||||
|
if tour['permNum'] == trn[1]:
|
||||||
|
tmp = tour
|
||||||
|
break
|
||||||
|
course = tmp['courses'][0]['courseName']
|
||||||
|
purse = tmp['Purse']
|
||||||
|
winnerPrize = tmp['winnersShare']
|
||||||
|
defending = '{} {}'.format(tmp['champions'][0]['playerName']['first'], tmp['champions'][0]['playerName']['last'])
|
||||||
|
w_string += ' :: {} :: \x02Defending champion:\x02 {} :: \x02Purse:\x02 ${} (${} to winner)'.format(
|
||||||
|
course, defending, purse, winnerPrize)
|
||||||
|
positions.append(w_string)
|
||||||
|
|
||||||
|
trunc = 10 if not options.get('all') else len(positions)
|
||||||
|
|
||||||
|
irc.reply('{} ({}) :: {} :: {}'.format(name, date, round_,
|
||||||
|
', '.join(p for p in positions[:trunc])))
|
||||||
|
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
|
Class = GolfScores
|
||||||
|
|
||||||
|
|
||||||
|
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79:
|
||||||
Loading…
Reference in New Issue