58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from __future__ import unicode_literals
|
|
from __future__ import print_function
|
|
from __future__ import division
|
|
from __future__ import absolute_import
|
|
from future import standard_library
|
|
standard_library.install_aliases()
|
|
from builtins import *
|
|
from builtins import object
|
|
from importlib import import_module
|
|
from os import path, listdir
|
|
|
|
from debug import logger
|
|
import messagetypes
|
|
import paths
|
|
|
|
class MsgBase(object):
|
|
def encode(self):
|
|
self.data = {"": type(self).__name__.lower()}
|
|
|
|
|
|
def constructObject(data):
|
|
whitelist = ["message"]
|
|
if data[""] not in whitelist:
|
|
return None
|
|
try:
|
|
classBase = getattr(getattr(messagetypes, data[""]), data[""].title())
|
|
except (NameError, AttributeError):
|
|
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
|
|
|
|
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)
|