Initial code commit

playwithfriends/.gitignore

@@ -0,0 +1 @@
+api_key.txt

playwithfriends/__init__.py

@@ -0,0 +1,2 @@
+__version__ = '0.0.1'
+__release = '0.0.1'

playwithfriends/intersections.py

@@ -0,0 +1,12 @@
+from playwithfriends import steam
+
+
+def intersecting_games(steamid_left, steamid_right):
+    api = steam.SteamAPI()
+
+    games_left = api.get_games(steamid_left)
+    games_right = api.get_games(steamid_right)
+
+    ids_mutual = set(games_left.keys()).intersection(games_right.keys())
+
+    return {game_id: games_left[game_id] for game_id in ids_mutual}

playwithfriends/steam.py

@@ -0,0 +1,108 @@
+import requests
+from enum import Enum
+from os import path
+from copy import deepcopy
+import logging
+
+
+_base_url = 'http://api.steampowered.com'
+
+
+class WebAPI(Enum):
+    news = 'ISteamNews'
+    user_stats = 'ISteamUserStats'
+    user = 'ISteamUser'
+    player_service = 'IPlayerService'
+
+
+class SteamAPI(object):
+
+    def __init__(self):
+        file_dir = path.abspath(path.dirname(__file__))
+        api_file = path.join(file_dir, 'api_key.txt')
+        with open(api_file, 'r') as handle:
+            self.api_key = handle.read().strip()
+
+    def _make_request(self, api, function, version, extra_params):
+        full_params = deepcopy(extra_params)
+        full_params['key'] = self.api_key
+        request_url = '/'.join([_base_url, api.value, function, version]) + '/'
+        logging.debug('GET: {}'.format(request_url))
+        response = requests.get(
+            request_url,
+            params=full_params
+        )
+        logging.debug('Return code: {}'.format(response.status_code))
+        return response.json()
+
+    def get_friends(self, steam_id):
+        """
+        Return the steam_id's of all users this user is friends with
+
+        :param steam_id: ID for the user we are looking up
+        :type steam_id: str
+        :return: List of all id's this user is friends with
+        :rtype: list[str]
+        """
+        friends_list = self._make_request(
+            WebAPI.user,
+            'GetFriendList',
+            'v0001',
+            {
+                'steamid': steam_id
+            })['friendslist']
+
+        return [
+            f['steamid'] for f in friends_list['friends']
+        ]
+
+    def get_games(self, steam_id):
+        """
+        Get all the appid's for a user that they can play. Structure
+        for return is a dict with `appid: name` values.
+
+        :param steam_id:
+        :return: List of appid's
+        :rtype: dict
+        """
+        games = self._make_request(
+            WebAPI.player_service,
+            'GetOwnedGames',
+            'v0001',
+            {
+                'steamid': steam_id,
+                'include_appinfo': 1,
+                'include_played_free_games': 1
+            })['response']['games']
+
+        return {g['appid']: g['name'] for g in games}
+
+    def get_recent_games(self, steam_id):
+        """
+        Get all appid's that a user has recently played
+
+        :param steam_id: ID for a steam user
+        :type steam_id: str
+        :return: List of appids the user has recently played
+        """
+        games = self._make_request(
+            WebAPI.player_service,
+            'GetRecentlyPlayedGames',
+            'v0001',
+            {
+                'steamid': steam_id
+            }
+        )['response']['games']
+
+        return [g['appid'] for g in games]
+
+    def get_player_summary(self, steam_id):
+
+        return self._make_request(
+            WebAPI.user,
+            'GetPlayerSummaries',
+            'v0002',
+            {
+                'steamids': steam_id
+            }
+        )