Add an optioonal class builder paraeter cls to `client.get_player()…

…`. This allows for the default player to easily be adapted and overwritten. Useful for subclassing and making custom `Player` classes.

Updated version.

setup.py

@@ -40,7 +40,7 @@
     requirements.append('sphinxcontrib-websupport')
     requirements.append('Pygments')
 
-version = '0.1.80'
+version = '0.1.90'
 
 readme = ''
 with open('README.rst') as f:

wavelink/__init__.py

@@ -2,7 +2,7 @@
 __author__ = 'EvieePy'
 __license__ = 'MIT'
 __copyright__ = 'Copyright 2019 (c) EvieePy'
-__version__ = '0.1.80a'
+__version__ = '0.1.90a'
 
 from .client import Client
 from .errors import *

wavelink/client.py

@@ -175,13 +175,16 @@ def get_node_by_shard(self, shard_id: int) -> Optional[Node]:
 
         return sorted(nodes, key=lambda n: len(n.players))[0]
 
-    def get_player(self, guild_id: int) -> Optional[Player]:
+    def get_player(self, guild_id: int, cls=None) -> Optional[Player]:
         """Retrieve a player for the given guild ID. If None, a player will be created and returned.
 
         Parameters
         ------------
         guild_id:
             The guild ID to retrieve a player for.
+        Optional[cls: class]
+            An optional class to pass to build from, overriding the default :class:`Player` class.
+            This must be similar to :class:`Player`. E.g a subclass.
 
         Returns
         ---------
@@ -195,7 +198,6 @@ def get_player(self, guild_id: int) -> Optional[Player]:
         ZeroConnectedNodes
             There are no :class:`wavelink.node.Node`'s currently connected.
         """
-
         players = self.players
 
         try:
@@ -212,6 +214,9 @@ def get_player(self, guild_id: int) -> Optional[Player]:
         if not self.nodes:
             raise ZeroConnectedNodes('There are not any currently connected nodes.')
 
+        if not cls:
+            cls = Player
+
         shard_options = []
         region_options = []
         nodes = self.nodes.values()
@@ -226,7 +231,7 @@ def get_player(self, guild_id: int) -> Optional[Player]:
 
         if not shard_options or region_options:
             node = sorted(nodes, key=lambda n: len(n.players))[0]
-            player = Player(self.bot, guild_id, node)
+            player = cls(self.bot, guild_id, node)
             node.players[guild_id] = player
 
             return player
@@ -239,7 +244,7 @@ def get_player(self, guild_id: int) -> Optional[Player]:
         else:
             node = sorted(region_options, key=lambda n: len(n.players))[0]
 
-        player = Player(self.bot, guild_id, node)
+        player = cls(self.bot, guild_id, node)
         node.players[guild_id] = player
 
         return player