Add client class registration

Author: linuxdaemon

cloudbot/client.py

@@ -7,6 +7,17 @@
 
 logger = logging.getLogger("cloudbot")
 
+CLIENTS = {}
+
+
+def client(_type):
+    def _decorate(cls):
+        CLIENTS[_type] = cls
+        cls._type = _type
+        return cls
+
+    return lambda cls: _decorate(cls)
+
 
 class Client:
     """
@@ -22,6 +33,8 @@ class Client:
     :type permissions: PermissionManager
     """
 
+    _type = None
+
     def __init__(self, bot, name, nick, *, channels=None, config=None):
         """
         :type bot: cloudbot.bot.CloudBot
@@ -155,3 +168,7 @@ def is_nick_valid(self, nick):
     @property
     def connected(self):
         raise NotImplementedError
+
+    @property
+    def type(self):
+        return self._type

cloudbot/clients/irc.py

@@ -7,7 +7,7 @@
 from functools import partial
 from ssl import SSLContext
 
-from cloudbot.client import Client
+from cloudbot.client import Client, client
 from cloudbot.event import Event, EventType, IrcOutEvent
 from cloudbot.util import async_util
 from cloudbot.util.parsers.irc import Message
@@ -50,6 +50,7 @@ def decode(bytestring):
     return bytestring.decode('utf-8', errors='ignore')
 
 
+@client("irc")
 class IrcClient(Client):
     """
     An implementation of Client for IRC.