diff options
Diffstat (limited to 'SteamAPI.py')
| -rw-r--r-- | SteamAPI.py | 100 |
1 files changed, 91 insertions, 9 deletions
diff --git a/SteamAPI.py b/SteamAPI.py index 87b5c39..1790989 100644 --- a/SteamAPI.py +++ b/SteamAPI.py @@ -193,36 +193,118 @@ class SteamAPI(): return ( jsondata['response']['game_count'], jsondata['response']['games'] ) return None - def getPlayerGameStats(self, steamid, gameid): - """Fetch the list of achievements a person achieved in a game + def getUserstatsForGame(self, steamid, gameid): + """Fetch the available game stats for a player in the specified game. Args: steamid: Steamid of the particular user gameid: Appid of the game we want to check Returns: - Tuple with (number of games, gameinfo [appid, name, playtime_2weeks, playtime_forever, icons]) + dict() with statname and value """ if CACHE: - cache = Caching.readCache('playergamestats', '%s-%s' % (str(steamid), str(gameid)), 24*60*60) + cache = Caching.readCache('usergamestats', '%s-%s' % (str(steamid), str(gameid)), 24*60*60) if cache: - jsondata = json.loads(cache) - return jsondata + cachedata = json.loads(cache) + return cachedata - url = 'http://api.steampowered.com/ISteamUserStats/GetPlayerAchievements/v0001/?key=%s&steamid=%s&appid=%s' % (self.token, str(steamid), str(gameid)) + url = 'http://api.steampowered.com/ISteamUserStats/GetUserStatsForGame/v0001/?key=%s&steamid=%s&appid=%s' % (self.token, str(steamid), str(gameid)) try: response = urlopen(url) data = response.read().decode('utf-8') jsondata = json.loads(data) + statslist = jsondata['playerstats']['stats'] + userstats = dict() + for stat in statslist: + userstats[stat] = statslist[stat]['value'] except HTTPError: # f.e. profile is private + userstats = None + + if userstats: + if CACHE: + cache = Caching.writeCache('usergamestats', '%s-%s' % (str(steamid), str(gameid)), json.dumps(userstats)) + return userstats + return None + + def getMultipleUserUserstatsForGame(self, steamids, gameid): + executor = ThreadPoolExecutor(max_workers=10) + futures = dict() + for steamid in steamids: + futures[steamid] = executor.submit(self.getUserstatsForGame, steamid, gameid) + + result = dict() + for steamid in steamids: + result[steamid] = futures[steamid].result() + + return result + + def getOwnedGames(self, steamid): + """Fetch games owned by a player. + + Args: + steamid: Steamid of the particular user + Returns: + dict() with game_count and games, which contains appid + playtime_forever + """ + if CACHE: + cache = Caching.readCache('userownedgames', '%s' % (str(steamid)), 7*24*60*60) + if cache: + cachedata = json.loads(cache) + return cachedata + + url = 'http://api.steampowered.com/IPlayerService/GetOwnedGames/v0001/?key=%s&steamid=%s' % (self.token, str(steamid)) + try: + response = urlopen(url) + data = response.read().decode('utf-8') + jsondata = json.loads(data) + if 'response' in jsondata: + jsondata = jsondata['response'] + except HTTPError: jsondata = None - if 'playerstats' in jsondata: + if jsondata: if CACHE: - cache = Caching.writeCache('playergamestats', '%s-%s' % (str(steamid), str(gameid)), json.dumps(jsondata)) + cache = Caching.writeCache('userownedgames', '%s' % (str(steamid)), json.dumps(jsondata)) return jsondata return None + def getMultipleUserOwnedGames(self, steamids): + executor = ThreadPoolExecutor(max_workers=10) + futures = dict() + for steamid in steamids: + futures[steamid] = executor.submit(self.getOwnedGames, steamid) + + result = dict() + for steamid in steamids: + result[steamid] = futures[steamid].result() + + return result + + def getDataForPremadefinder(self, steamids): + executor = ThreadPoolExecutor(max_workers=20) + + futures = dict() + futures['profiles'] = executor.submit(self.getProfiles, steamids) + futures['ownedGames'] = dict() + futures['userstats'] = dict() + futures['friends'] = dict() + + for steamid in steamids: + futures['ownedGames'][steamid] = executor.submit(self.getOwnedGames, steamid) + futures['userstats'][steamid] = executor.submit(self.getUserstatsForGame, steamid, 730) + futures['friends'][steamid] = executor.submit(self.getFriends, steamid) + + profiles = futures['profiles'].result() + for steamid in profiles: + profiles[steamid]['_friends'] = futures['friends'][steamid].result() + profiles[steamid]['_userstats'] = futures['userstats'][steamid].result() + profiles[steamid]['_ownedGames'] = futures['ownedGames'][steamid].result() + profiles[steamid]['_ownedPlayedGames'] = len([game for game in profiles[steamid]['_ownedGames']['games'] if game['playtime_forever'] > 0]) + + + return profiles + if __name__ == "__main__": # TODO(andre): Maybe run tests here? print('This is a module.') |