pythonhacktoberfeststeamauthenticationauthenticatorsteam-authenticatorsteam-clientsteam-guard-codessteam-websteamworksvalvewebapi
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.
90 lines
3.1 KiB
90 lines
3.1 KiB
from weakref import WeakValueDictionary
|
|
from steam.client.user import SteamUser
|
|
from steam.enums import EPersonaState
|
|
from steam.enums.emsg import EMsg
|
|
from steam.core.msg import MsgProto
|
|
from steam.util import proto_fill_from_dict
|
|
|
|
class User(object):
|
|
persona_state = EPersonaState.Online #: current persona state
|
|
user = None #: :class:`.SteamUser` instance once logged on
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(User, self).__init__(*args, **kwargs)
|
|
|
|
self._user_cache = WeakValueDictionary()
|
|
|
|
self.on(self.EVENT_DISCONNECTED, self.__handle_disconnect)
|
|
self.on(self.EVENT_LOGGED_ON, self.__handle_set_persona)
|
|
self.on(EMsg.ClientPersonaState, self.__handle_persona_state)
|
|
|
|
def __handle_disconnect(self):
|
|
self.user = None
|
|
|
|
def __handle_set_persona(self):
|
|
self.change_status(persona_state=self.persona_state)
|
|
self.user = self.get_user(self.steam_id)
|
|
|
|
def __handle_persona_state(self, message):
|
|
for friend in message.body.friends:
|
|
steamid = friend.friendid
|
|
|
|
if steamid in self._user_cache:
|
|
suser = self._user_cache[steamid]
|
|
suser._pstate = friend
|
|
suser._pstate_ready.set()
|
|
|
|
def change_status(self, **kwargs):
|
|
"""
|
|
Set name, persona state, flags
|
|
|
|
.. note::
|
|
Changing persona state will also change :attr:`persona_state`
|
|
|
|
:param persona_state: persona state (Online/Offlane/Away/etc)
|
|
:type persona_state: :class:`.EPersonaState`
|
|
:param player_name: profile name
|
|
:type player_name: :class:`str`
|
|
:param persona_state_flags: persona state flags
|
|
:type persona_state_flags: :class:`.EPersonaStateFlag`
|
|
"""
|
|
if not kwargs: return
|
|
|
|
self.persona_state = kwargs.get('persona_state', self.persona_state)
|
|
|
|
message = MsgProto(EMsg.ClientChangeStatus)
|
|
proto_fill_from_dict(message.body, kwargs)
|
|
self.send(message)
|
|
|
|
def request_persona_state(self, steam_ids):
|
|
"""Request persona state data
|
|
|
|
:param steam_ids: list of steam ids
|
|
:type steam_ids: :class:`list`
|
|
"""
|
|
m = MsgProto(EMsg.ClientRequestFriendData)
|
|
m.body.persona_state_requested = 4294967295 # request all possible flags
|
|
m.body.friends.extend(steam_ids)
|
|
self.send_job(m)
|
|
|
|
def get_user(self, steam_id, fetch_persona_state=True):
|
|
"""Get :class:`.SteamUser` instance for ``steam id``
|
|
|
|
:param steam_id: steam id
|
|
:type steam_id: :class:`int`, :class:`.SteamID`
|
|
:param fetch_persona_state: whether to request person state when necessary
|
|
:type fetch_persona_state: :class:`bool`
|
|
:return: SteamUser instance
|
|
:rtype: :class:`.SteamUser`
|
|
"""
|
|
steam_id = int(steam_id)
|
|
suser = self._user_cache.get(steam_id, None)
|
|
|
|
if suser is None:
|
|
suser = SteamUser(steam_id, self)
|
|
self._user_cache[steam_id] = suser
|
|
|
|
if fetch_persona_state:
|
|
self.request_persona_state([steam_id])
|
|
|
|
return suser
|
|
|