def __init__(self, config):
super(Client, self).__init__()
self.config = config
self.events = Emitter()
self.packets = Emitter()
self.api = APIClient(self.config.token, self)
self.gw = GatewayClient(self, self.config.max_reconnects,
self.config.encoder)
self.state = State(self, StateConfig(self.config.get('state', {})))
if self.config.manhole_enable:
self.manhole_locals = {
'client': self,
'state': self.state,
'api': self.api,
'gw': self.gw,
}
self.manhole = DiscoBackdoorServer(
self.config.manhole_bind,
banner='Disco Manhole',
localf=lambda: self.manhole_locals)
self.manhole.start()
class Client(LoggingClass):
"""
Class representing the base entry point that should be used in almost all
implementation cases. This class wraps the functionality of both the REST API
(`disco.api.client.APIClient`) and the realtime gateway API
(`disco.gateway.client.GatewayClient`).
Parameters
----------
config : `ClientConfig`
Configuration for this client instance.
Attributes
----------
config : `ClientConfig`
The runtime configuration for this client.
events : `Emitter`
An emitter which emits Gateway events.
packets : `Emitter`
An emitter which emits Gateway packets.
state : `State`
The state tracking object.
api : `APIClient`
The API client.
gw : `GatewayClient`
The gateway client.
manhole_locals : dict
Dictionary of local variables for each manhole connection. This can be
modified to add/modify local variables.
manhole : Optional[`BackdoorServer`]
Gevent backdoor server (if the manhole is enabled).
"""
def __init__(self, config):
super(Client, self).__init__()
self.config = config
self.events = Emitter()
self.packets = Emitter()
self.api = APIClient(self.config.token, self)
self.gw = GatewayClient(self, self.config.max_reconnects, self.config.encoder)
self.state = State(self, StateConfig(self.config.get('state', {})))
if self.config.manhole_enable:
self.manhole_locals = {
'client': self,
'state': self.state,
'api': self.api,
'gw': self.gw,
}
self.manhole = DiscoBackdoorServer(self.config.manhole_bind,
banner='Disco Manhole',
localf=lambda: self.manhole_locals)
self.manhole.start()
def update_presence(self, status, game=None, afk=False, since=0.0):
"""
Updates the current clients presence.
Params
------
status : `user.Status`
The clients current status.
game : `user.Game`
If passed, the game object to set for the users presence.
afk : bool
Whether the client is currently afk.
since : float
How long the client has been afk for (in seconds).
"""
if game and not isinstance(game, Game):
raise TypeError('Game must be a Game model')
if status is Status.IDLE and not since:
since = int(time.time() * 1000)
payload = {
'afk': afk,
'since': since,
'status': status.value.lower(),
'game': None,
}
if game:
payload['game'] = game.to_dict()
self.gw.send(OPCode.STATUS_UPDATE, payload)
def run(self):
"""
Run the client (e.g. the `GatewayClient`) in a new greenlet.
"""
return gevent.spawn(self.gw.run)
def run_forever(self):
"""
Run the client (e.g. the `GatewayClient`) in the current greenlet.
"""
return self.gw.run()
class Client(object):
"""
Class representing the base entry point that should be used in almost all
implementation cases. This class wraps the functionality of both the REST API
(:class:`disco.api.client.APIClient`) and the realtime gateway API
(:class:`disco.gateway.client.GatewayClient`).
Parameters
----------
config : :class:`ClientConfig`
Configuration for this client instance.
Attributes
----------
config : :class:`ClientConfig`
The runtime configuration for this client.
events : :class:`Emitter`
An emitter which emits Gateway events.
packets : :class:`Emitter`
An emitter which emits Gateway packets.
state : :class:`State`
The state tracking object.
api : :class:`APIClient`
The API client.
gw : :class:`GatewayClient`
The gateway client.
manhole_locals : dict
Dictionary of local variables for each manhole connection. This can be
modified to add/modify local variables.
manhole : Optional[:class:`BackdoorServer`]
Gevent backdoor server (if the manhole is enabled).
"""
def __init__(self, config):
super(Client, self).__init__()
self.config = config
self.events = Emitter(gevent.spawn)
self.packets = Emitter(gevent.spawn)
self.state = State(self)
self.api = APIClient(self)
self.gw = GatewayClient(self, self.config.encoder)
if self.config.manhole_enable:
self.manhole_locals = {
'client': self,
'state': self.state,
'api': self.api,
'gw': self.gw
}
self.manhole = DiscoBackdoorServer(
self.config.manhole_bind,
banner='Disco Manhole',
localf=lambda: self.manhole_locals)
self.manhole.start()
def run(self):
"""
Run the client (e.g. the :class:`GatewayClient`) in a new greenlet
"""
return gevent.spawn(self.gw.run)
def run_forever(self):
"""
Run the client (e.g. the :class:`GatewayClient`) in the current greenlet
"""
return self.gw.run()