You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

69 line
2.2KB

  1. #!/usr/bin/python3
  2. import requests
  3. import sys
  4. import textwrap
  5. api = 'http://api.steampowered.com/'
  6. key = '63578AF8E85EDF1E49411D5D4E18E166'
  7. uid = '76561198008294872' #whiteline
  8. #uid = '76561198111568728' #N33R
  9. # Fetch JSON from url and run it through transform, pretty printing errors
  10. # and the data worked on as exhaustively as possible.
  11. def json_query(url, transform, params={}):
  12. try:
  13. result = requests.get(url, params)
  14. except ConnectionError:
  15. print("Network connection error.")
  16. sys.exit(1)
  17. try:
  18. data = result.json()
  19. except requests.JSONDecodeError as err:
  20. print('Error when decoding JSON:\nFrom endpoint ' + url + ':\n' + err + '\n' + result + '\n')
  21. sys.exit(1)
  22. try:
  23. return transform(data)
  24. except (IndexError, KeyError) as err:
  25. print('Error when traversing JSON:\nFrom endpoint ' + url + ':\n' + err)
  26. pprint.PrettyPrinter(indent=2).pprint(data)
  27. sys.exit(1)
  28. def steam_query(endpoint, transform, params={}):
  29. new_params = params.copy()
  30. new_params['key'] = key
  31. return json_query(api + endpoint + '/', transform, new_params)
  32. def get_friend_list(steamid):
  33. params = {
  34. 'steamid' : steamid,
  35. 'relationship' : 'friend'
  36. }
  37. return steam_query('ISteamUser/GetFriendList/v0001', lambda data: data['friendslist']['friends'], params)
  38. def get_player_summaries(steamids):
  39. params = {
  40. 'steamids' : ','.join(steamids)
  41. }
  42. return steam_query('ISteamUser/GetPlayerSummaries/v0002', lambda data: data['response']['players'], params)
  43. def personastate(n):
  44. states = {
  45. 0 : 'Offline',
  46. 1 : 'Online',
  47. 2 : 'Busy',
  48. 3 : 'Away',
  49. 4 : 'Snooze',
  50. 5 : 'Looking to trade',
  51. 6 : 'Looking to play'
  52. }
  53. return states[n]
  54. def main():
  55. #print('\n'.join([friend['steamid'] for friend in get_friend_list(uid)]))
  56. for player in get_player_summaries([friend['steamid'] for friend in get_friend_list(uid)]):
  57. print(player['personaname'] + ': ' + personastate(player['personastate']) + (', playing ' + player['gameextrainfo'] if 'gameextrainfo' in player else ''))
  58. if __name__ == '__main__':
  59. main()