Fix message encoding bug

- prevent loading invalid message types
2018-02-13 16:39:35 +01:00 · 2018-02-13 16:39:35 +01:00 · 3a8016d31f
commit 3a8016d31f
parent 96ea36cfd2
1 changed files with 4 additions and 3 deletions
--- a/src/messagetypes/init.py
+++ b/src/messagetypes/init.py
@ -12,9 +12,10 @@ class MsgBase(object):

 def constructObject(data):
    try:
-        classBase = eval(data[""] + "." + data[""].title())
-    except NameError:
-        logger.error("Don't know how to handle message type: \"%s\"", data[""])
+        m = import_module("messagetypes." + data[""])
+        classBase = getattr(m, data[""].title())
+    except (NameError, ImportError):
+        logger.error("Don't know how to handle message type: \"%s\"", data[""], exc_info=True)
        return None
    try:
        returnObj = classBase()