social.py

import asyncio
import logging as log
import json


class SocialFeatures(object):
    def __init__(self, _bnet_client):
        self.bnet_client = _bnet_client
        self._friends = {}

    async def get_friends(self):
        _friends_future = asyncio.get_running_loop().create_future()
        await self.bnet_client.fetch_friends_list(_friends_future)
        await _friends_future

        self._friends = {}
        for friend in _friends_future.result():
            _friend_battle_tag_future = asyncio.get_running_loop().create_future()
            await self.bnet_client.fetch_friend_battle_tag(friend.id, _friend_battle_tag_future)
            await _friend_battle_tag_future

            friend.battle_tag = _friend_battle_tag_future.result()["battle_tag"]
            self._friends[str(friend.id.low)] = friend

        log.debug(f"fetched {len(self._friends)} friends")

        return self._friends

    async def get_friend_presence(self, user_id):
        if user_id not in self._friends:
            return None

        _friend_presence_future = asyncio.get_running_loop().create_future()
        await self.bnet_client.fetch_friend_presence_account_details(self._friends[user_id].id, _friend_presence_future)
        await _friend_presence_future

        account_info = _friend_presence_future.result()

        if "game_accounts" in account_info:
            for game_account in account_info["game_accounts"]:
                _game_account_future = asyncio.get_running_loop().create_future()
                await self.bnet_client.fetch_friend_presence_game_account_details(game_account["id"], _game_account_future)
                await _game_account_future

                game_account.update(_game_account_future.result())

        log.debug(f"fetched friend presence ({user_id}): {json.dumps(account_info, default=str)}")

        return account_info