Update .steamcalc to use steamdb.info
This commit is contained in:
Steven Smith
parent
a8b830af4f
commit
6750b076d8
1 changed files with 76 additions and 57 deletions
133
plugins/steam.py
133
plugins/steam.py
|
|
@ -1,65 +1,84 @@
|
||||||
from util import hook, http, web, text
|
from util import hook, http, web, text
|
||||||
|
from urllib import urlencode
|
||||||
|
from datetime import datetime
|
||||||
import re
|
import re
|
||||||
|
|
||||||
# this is still beta code. some things will be improved later.
|
|
||||||
|
|
||||||
count_re = re.compile(r"Found (.*?) Games with a value of ")
|
|
||||||
value_re = re.compile(r'\$(\d+\.\d{2})')
|
|
||||||
|
|
||||||
|
|
||||||
def db_init(db):
|
def db_init(db):
|
||||||
"check to see that our db has the the top steam users table."
|
db.execute("create table if not exists steam(nick primary key, acc)")
|
||||||
db.execute("create table if not exists steam_rankings(id, value, count, "
|
|
||||||
"primary key(id))")
|
|
||||||
db.commit()
|
db.commit()
|
||||||
|
|
||||||
|
def timesince(dt, default="just now"):
|
||||||
|
now = datetime.utcnow()
|
||||||
|
diff = now - dt
|
||||||
|
periods = (
|
||||||
|
(diff.days / 365, "year", "years"),
|
||||||
|
(diff.days / 30, "month", "months"),
|
||||||
|
(diff.days / 7, "week", "weeks"),
|
||||||
|
(diff.days, "day", "days"),
|
||||||
|
(diff.seconds / 3600, "hour", "hours"),
|
||||||
|
(diff.seconds / 60, "minute", "minutes"),
|
||||||
|
(diff.seconds, "second", "seconds"),
|
||||||
|
)
|
||||||
|
for period, singular, plural in periods:
|
||||||
|
if period:
|
||||||
|
return "%d %s ago" % (period, singular if period == 1 else plural)
|
||||||
|
return default
|
||||||
|
|
||||||
@hook.command
|
@hook.command('sc', autohelp=False)
|
||||||
def steamcalc(inp, db=None):
|
|
||||||
"steamcalc <user> -- Check the value of <user>s steam account."
|
|
||||||
db_init(db)
|
|
||||||
|
|
||||||
if " " in inp:
|
|
||||||
return "Invalid Steam ID"
|
|
||||||
|
|
||||||
uid = inp.strip().lower()
|
|
||||||
url = "http://steamcalculator.com/id/{}".format(http.quote_plus(uid))
|
|
||||||
|
|
||||||
# get the web page
|
|
||||||
try:
|
|
||||||
page = http.get_html(url)
|
|
||||||
except Exception as e:
|
|
||||||
return "Could not get Steam game listing: {}".format(e)
|
|
||||||
|
|
||||||
# extract the info we need
|
|
||||||
try:
|
|
||||||
count_text = page.xpath("//div[@id='rightdetail']/text()")[0]
|
|
||||||
count = int(count_re.findall(count_text)[0])
|
|
||||||
|
|
||||||
value_text = page.xpath("//div[@id='rightdetail']/h1/text()")[0]
|
|
||||||
value = float(value_re.findall(value_text)[0])
|
|
||||||
except IndexError:
|
|
||||||
return "Could not get Steam game listing."
|
|
||||||
|
|
||||||
# save the info in the DB for steam rankings
|
|
||||||
db.execute("insert or replace into steam_rankings(id, value, count)"
|
|
||||||
"values(?,?,?)", (uid, value, count))
|
|
||||||
db.commit()
|
|
||||||
|
|
||||||
# shorten the URL
|
|
||||||
try:
|
|
||||||
short_url = web.isgd(url)
|
|
||||||
except web.ShortenError:
|
|
||||||
short_url = url
|
|
||||||
|
|
||||||
return u"\x02Games:\x02 {}, \x02Total Value:\x02 ${:.2f} USD - {}".format(count, value, short_url)
|
|
||||||
|
|
||||||
|
|
||||||
@hook.command(autohelp=False)
|
@hook.command(autohelp=False)
|
||||||
def steamtop(inp, db=None):
|
"""steamcalc [username] [currency] - Gets value of steam account and total hours played. Uses steamcommunity.com/id/<nickname>. Uses IRC nickname if none provided. """
|
||||||
"steamtop -- Shows the top five users from steamcalc."
|
currencies = {'USD': 'us', 'euro1': "de", 'euro2': 'no',
|
||||||
rows = []
|
'pound': 'uk', 'rubles': 'ru', 'real': 'br',
|
||||||
for row in db.execute("SELECT id, value, count FROM steam_rankings ORDER BY value DESC LIMIT 5"):
|
'yen': 'jp', 'dollars': 'us', 'german': 'de',
|
||||||
rows.append(u"{} - \x02${:.2f}\x02 ({} games)".format(text.munge(row[0], 1), row[1], row[2]))
|
'pounds': 'uk', 'russian': 'ru', 'brazil': 'br',
|
||||||
|
'japan': 'jp', 'us': 'us', 'de': 'de', 'no': 'no',
|
||||||
return u"Top Steam Users: {}".format(", ".join(rows))
|
'uk': 'uk', 'ru': 'ru', 'br': 'br', 'jp': 'jp'}
|
||||||
|
db_init(db)
|
||||||
|
currency = None
|
||||||
|
dontsave = False
|
||||||
|
if not inp:
|
||||||
|
user = db.execute("select acc from steam where nick=lower(?)", (nick,)).fetchone()
|
||||||
|
if not user:
|
||||||
|
inp = nick
|
||||||
|
else:
|
||||||
|
inp = user[0]
|
||||||
|
dontsave = True
|
||||||
|
else:
|
||||||
|
if len(inp.split(" ")) > 1:
|
||||||
|
if inp.split(" ")[1] in currencies:
|
||||||
|
currency = currencies[inp.split(" ")[1]]
|
||||||
|
else:
|
||||||
|
return "Invalid currency!"
|
||||||
|
inp = inp.split(" ")[0]
|
||||||
|
dontsave = False
|
||||||
|
urldata = urlencode({"player": inp, "currency": currency if currency else "us"})
|
||||||
|
soup = http.get_soup("http://steamdb.info/calculator/?" + urldata)
|
||||||
|
try:
|
||||||
|
name = soup.findAll('h1', {'class': 'header-title'})[1].text
|
||||||
|
status = soup.findAll('td')[7].text
|
||||||
|
if status == "Online":
|
||||||
|
status = "\x033\x02Online\x02\x0f"
|
||||||
|
elif status == "Offline":
|
||||||
|
status = "\x034\x02Offline\x02\x0f"
|
||||||
|
elif status == "Away":
|
||||||
|
status = "\x038\x02Away\x02\x0f"
|
||||||
|
elif status == "Busy":
|
||||||
|
status = "\x035\x02Busy\x02\x0f"
|
||||||
|
elif "Looking to" in status:
|
||||||
|
status = "\x036\x02%s\x02\x0f" % status
|
||||||
|
twdata = soup.find('h1', {'class': 'header-title pull-right'}).find('a')['data-text'].split(", ")
|
||||||
|
money = twdata[0].split("My #Steam account is worth ")[1]
|
||||||
|
time = twdata[1].split("and I spent ")[1].split(" playing games!")[0]
|
||||||
|
timeonsteam = soup.findAll('i')[1].text[1:-1].split(" ")
|
||||||
|
timestamp = datetime.strptime(timeonsteam[0]+" "+timeonsteam[1]+" "+timeonsteam[2] + " - " + timeonsteam[4]+" "+timeonsteam[5], "%B %d, %Y - %H:%M:%S UTC")
|
||||||
|
timeonsteam = timesince(timestamp)
|
||||||
|
totalgames = soup.find('b').text
|
||||||
|
notplayed = soup.findAll('b')[1].text
|
||||||
|
nppercent = soup.findAll('i')[3].text[1:-1]
|
||||||
|
if not dontsave:
|
||||||
|
db.execute("insert or replace into steam(nick, acc) values (?,?)", (nick.lower(), inp))
|
||||||
|
db.commit()
|
||||||
|
return u"%s (%s): This Steam account is worth \x02%s\x02, and they've spent \x02%s\x02 playing games! Their Steam account was created %s! They have \x02%s games in their Steam library\x02, but \x02%s of them haven't been touched\x02! That's %s! %s" % (name, status, money, time, timeonsteam, totalgames, notplayed, nppercent, web.isgd("http://steamdb.info/calculator/?"+ urldata))
|
||||||
|
except Exception as e:
|
||||||
|
print e
|
||||||
|
return u"\x02Unable to retrieve info for %s!\x02 Check that it's a valid Steam profile username (steamcommunity.com/id/<USERNAME>), check if your profile is private, or go here to search: %s" % (inp, web.isgd("http://steamdb.info/calculator/?"+ urldata))
|
||||||
|
|
|
||||||
Reference in a new issue