Remove messagetypes quirks

src/messagetypes/__init__.py

@@ -1,56 +1,19 @@
 import logging
-from importlib import import_module
-from os import listdir, path
-from string import lower
 
-import messagetypes
+from message import Message
-import paths
 
 logger = logging.getLogger('default')
 
 
-class MsgBase(object):  # pylint: disable=too-few-public-methods
-    """Base class for message types"""
-    def __init__(self):
-        self.data = {"": lower(type(self).__name__)}
-
-
 def constructObject(data):
-    """Constructing an object"""
+    """Constructing an object using `.Message` class"""
-    whitelist = ["message"]
+
-    if data[""] not in whitelist:
-        return None
     try:
-        classBase = getattr(getattr(messagetypes, data[""]), data[""].title())
+        obj = Message()
-    except (NameError, AttributeError):
+        obj.decode(data)
-        logger.error("Don't know how to handle message type: \"%s\"", data[""], exc_info=True)
-        return None
-    try:
-        returnObj = classBase()
-        returnObj.decode(data)
     except KeyError as e:
         logger.error("Missing mandatory key %s", e)
-        return None
     except:
         logger.error("classBase fail", exc_info=True)
-        return None
     else:
-        return returnObj
+        return obj
-
-
-if paths.frozen is not None:
-    import messagetypes.message
-    import messagetypes.vote
-else:
-    for mod in listdir(path.dirname(__file__)):
-        if mod == "__init__.py":
-            continue
-        splitted = path.splitext(mod)
-        if splitted[1] != ".py":
-            continue
-        try:
-            import_module(".{}".format(splitted[0]), "messagetypes")
-        except ImportError:
-            logger.error("Error importing %s", mod, exc_info=True)
-        else:
-            logger.debug("Imported message type module %s", mod)

src/messagetypes/_base.py

@@ -0,0 +1,7 @@
+"""Base class for message types"""
+
+
+class MsgBase(object):  # pylint: disable=too-few-public-methods
+    """Base class for message types"""
+    def __init__(self):
+        self.data = {"": type(self).__name__.lower()}

src/messagetypes/message.py

@@ -1,6 +1,6 @@
 import logging
 
-from messagetypes import MsgBase
+from _base import MsgBase
 
 logger = logging.getLogger('default')
 

src/messagetypes/vote.py

@@ -1,6 +1,6 @@
 import logging
 
-from messagetypes import MsgBase
+from . import MsgBase
 
 logger = logging.getLogger('default')