def get_message(self, command=None):
        try:
            if command is not None and self._Messages[command]:
                _msg = __import__(
                    self._Messages[command],
                    globals(),
                    locals(),
                    [command]
                )

                # Get the right instance from Import List
                _Message = getattr(_msg, command)
                if self._connection.auth_token != b'':
                    token = self._connection.auth_token
                else:
                    token = self._auth_token

                message_instance = _Message(self._connection)\
                    .set_session_token(token)
                message_instance._push_callback = self._push_received
                return message_instance

        except __HOLE__ as e:
            self._connection.close()
            raise PyOrientBadMethodCallException(
                "Unable to find command " + str(e), []
            )
dataset/ETHPy150Open mogui/pyorient/pyorient/orient.py/OrientDB.get_message