ChCh/CloudBot
Archived
10
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

If data already exists for a user, return that and wait until after the command has finished to trigger an update

Browse source
This commit is contained in:
Luke Rogers 2013-09-10 12:00:29 +12:00
parent e644a0a0ca
commit 1f699c8f00
1 changed files with 22 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
plugins/steam_calc.py
View file

@ -1,6 +1,5 @@
from util import hook, http from util import hook, http
import csv import csv
import time
import StringIO import StringIO
gauge_url = "http://www.mysteamgauge.com/search?username={}" gauge_url = "http://www.mysteamgauge.com/search?username={}"
@ -8,6 +7,9 @@ gauge_url = "http://www.mysteamgauge.com/search?username={}"
api_url = "http://mysteamgauge.com/user/{}.csv" api_url = "http://mysteamgauge.com/user/{}.csv"
steam_api_url = "http://steamcommunity.com/id/{}/?xml=1" steam_api_url = "http://steamcommunity.com/id/{}/?xml=1"
def refresh_data(name): http.get(gauge_url.format(name), timeout=25, get_method='HEAD')
def get_data(name): return http.get(api_url.format(name))
def is_number(s): def is_number(s):
try: try:
@ -32,11 +34,16 @@ def steamcalc(inp, reply=None):
name = inp.strip() name = inp.strip()
try: try:
reply("Collecting data, this may take a while.") request = get_data(name)
http.get(gauge_url.format(name), timeout=25, get_method='HEAD') do_refresh = True
request = http.get(api_url.format(name))
except (http.HTTPError, http.URLError): except (http.HTTPError, http.URLError):
return "Could not get data for this user." try:
reply("Collecting data, this may take a while.")
refresh_data(name)
request = get_data(name)
do_refresh = False
except (http.HTTPError, http.URLError):
return "Could not get data for this user."
csv_data = StringIO.StringIO(request) # we use StringIO because CSV can't read a string csv_data = StringIO.StringIO(request) # we use StringIO because CSV can't read a string
reader = unicode_dictreader(csv_data) reader = unicode_dictreader(csv_data)
@ -50,9 +57,12 @@ def steamcalc(inp, reply=None):
# basic information # basic information
steam_profile = http.get_xml(steam_api_url.format(name)) steam_profile = http.get_xml(steam_api_url.format(name))
data["name"] = steam_profile.find('steamID').text try:
data["name"] = steam_profile.find('steamID').text
online_state = steam_profile.find('stateMessage').text
except AttributeError:
return "Could not get data for this user."
online_state = steam_profile.find('stateMessage').text
data["state"] = online_state.replace("<br/>", ": ") # will make this pretty later data["state"] = online_state.replace("<br/>", ": ") # will make this pretty later
# work out the average metascore for all games # work out the average metascore for all games
@ -81,6 +91,9 @@ def steamcalc(inp, reply=None):
data["size"] = "{0:.1f}".format(total_size) data["size"] = "{0:.1f}".format(total_size)
return "{name} ({state}) has {games} games with a total value of ${value}" \ reply("{name} ({state}) has {games} games with a total value of ${value}" \
" and a total size of {size}GB! The average metascore for these" \ " and a total size of {size}GB! The average metascore for these" \
" games is {average_metascore}.".format(**data) " games is {average_metascore}.".format(**data))
if do_refresh:
refresh_data(name)