|
- #!/usr/bin/python3
- import requests
- import sys
- import textwrap
- import pprint
- try:
- from simplejson.errors import JSONDecodeError
- except ImportError:
- from json.decoder import JSONDecodeError
-
- api = 'https://api.steampowered.com/'
- key = '63578AF8E85EDF1E49411D5D4E18E166'
- uid = '76561198008294872' #whiteline
- #uid = '76561198111568728' #N33R
-
- # Fetch JSON from url and run it through transform, pretty printing errors
- # and the data worked on as exhaustively as possible.
- def json_query(url, transform, params={}):
- try:
- result = requests.get(url, params)
- except requests.exceptions.ConnectionError:
- print("Network connection error.")
- sys.exit(1)
- try:
- data = result.json()
- except JSONDecodeError as err:
- print('Error when decoding JSON:\nFrom endpoint ' + url + ':\n' + str(err) + '\n' + str(result) + '\n')
- sys.exit(1)
- try:
- return transform(data)
- except (IndexError, KeyError) as err:
- print('Error when traversing JSON:\nFrom endpoint ' + url + ':\n' + str(err))
- pprint.PrettyPrinter(indent=2).pprint(data)
- sys.exit(1)
-
- def steam_query(endpoint, transform, params={}):
- new_params = params.copy()
- new_params['key'] = key
- return json_query(api + endpoint + '/', transform, new_params)
-
- def get_friend_list(steamid):
- params = {
- 'steamid' : steamid,
- 'relationship' : 'friend'
- }
- return steam_query('ISteamUser/GetFriendList/v0001', lambda data: data['friendslist']['friends'], params)
-
- def get_player_summaries(steamids):
- params = {
- 'steamids' : ','.join(steamids)
- }
- return steam_query('ISteamUser/GetPlayerSummaries/v0002', lambda data: data['response']['players'], params)
-
- def boldgreen(string):
- return '\033[92m' + '\033[1m' + string + '\033[0m'
-
- def boldred(string):
- return '\033[91m' + '\033[1m' + string + '\033[0m'
-
- def boldgray(string):
- return '\033[90m' + '\033[1m' + string + '\033[0m'
-
- def personastate(n):
- states = {
- 0 : boldred('Offline'),
- 1 : boldgreen('Online'),
- 2 : boldred('Busy'),
- 3 : boldgray('Away'),
- 4 : boldgray('Snooze'),
- 5 : boldgreen('Looking to trade'),
- 6 : boldgreen('Looking to play')
- }
- return states[n]
-
- def main():
- friends = [friend['steamid'] for friend in get_friend_list(uid)]
- players = sorted(get_player_summaries(friends), key=lambda player: player['personaname'])
- print('\n'.join([player['personaname'] + ': ' + personastate(player['personastate']) + (', playing ' + player['gameextrainfo'] if 'gameextrainfo' in player else '') for player in players]))
-
-
- if __name__ == '__main__':
- main()
|