h/plugins/tf.py

44 lines
1.2 KiB
Python

# tf.py: written by ipsum
#
# This skybot plugin retreives the number of items
# a given user has waiting from idling in Team Fortress 2.
from util import hook, http
@hook.command('hats')
@hook.command
def tf(inp):
""".tf/.hats <SteamID> -- Shows items waiting to be received in TF2."""
if not inp:
return tf.__doc__
if inp.isdigit():
link = 'profiles'
else:
link = 'id'
url = 'http://steamcommunity.com/%s/%s/tfitems?json=1' % \
(link, http.quote(inp.encode('utf8'), safe=''))
try:
inv = http.get_json(url)
except ValueError:
return '%s is not a valid profile' % inp
dropped,dhats,hats = 0,0,0
for item, data in inv.iteritems():
ind = int(data['defindex'])
if data['inventory'] == 0:
if 47 <= ind <= 55 or 94 <= ind <= 126 or 134 <= ind <= 152:
dhats += 1
else:
dropped += 1
else:
if 47 <= ind <= 55 or 94 <= ind <= 126 or 134 <= ind <= 152:
hats += 1
return '%s has had %s items and %s hats drop (%s total hats)' % \
(inp, dropped, dhats, dhats + hats)