Fetch given friendlists in parallel

1 files changed, 5 insertions, 4 deletions

diff --git a/main.py b/main.py
index d6c655a..fa0b728 100755
--- a/main.py
+++ b/main.py
@@ -77,10 +77,11 @@ def lobby():
     friends = request.args.get('friends')
 
     if friends:
-        for steamid in friends.split(','):
-            for friend in steam.getFriends(steamid):
-                if friend not in steamids:
-                    steamids[friend] = {}
+        friendList = friends.split(',')
+        profiles = steam.getMultipleFriends(friendList)
+        for steamid, profile in profiles.items():
+            for friendid in profile['friends']:
+                steamids[friendid] = {}
     else:
         # Load config (steamids, names)
         with open(os.path.join(CONFIG, 'lobby.json'), 'rt') as config: