h/plugins/tf.py

# 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 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)