h/plugins/tf.py

47 lines
1.2 KiB
Python
Raw Normal View History

'''
tf.py
A skybot plugin for Team Fortress 2
By Ipsum
'''
import json
import urllib
from util import hook
@hook.command('hats')
@hook.command
def tf(inp):
'''.tf/.hats <SteamID> -- Displays the number of hats
and items waiting to be received by <SteamID>'''
if not inp:
return tf.__doc__
if inp.isdigit() : link = 'profiles'
else : link = 'id'
url = 'http://steamcommunity.com/%s/%s/tfitems?json=1' % \
(link,urllib.quote(inp, safe=''))
raw_data = urllib.urlopen(url).read().decode('utf-8')
try:
inv = json.JSONDecoder().decode(raw_data)
except ValueError:
return '%s is not a valid profile' % inp
dropped,dhats,hats = 0,0,0
for item in inv:
defindex = int(inv[item]['defindex'])
if inv[item]['inventory'] == 0:
if 47<=defindex<=55 or 94<=defindex<=126 or 134<=defindex<=152:
dhats += 1
else:
dropped += 1
else:
if 47<=defindex<=55 or 94<=defindex<=126 or 134<=defindex<=152:
hats += 1
return '%s has had %s items and %s hats drop (%s total hats)' % (inp,dropped,dhats,dhats+hats)