unity_body.py

import logging
from bml import to_xml_clean
from tcp_sender import TCPSender


logger = logging.getLogger(__name__)


class UnityBody:
    def __init__(self, character_name='ChrBrad'):
        self._conn = None
        self._character_name = character_name

    def __enter__(self):
        self._conn = TCPSender()
        self._conn.init_network()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self._conn.finit_network()

    def execute(self, bml_list):
        logger.debug('Executing BML commands:\n%s', bml_list)
        xml = to_xml_clean(bml_list)
        print(xml)
        return self._conn.send_msg(xml) # tcp only, does not include sending files

    def check_done(self, id):
        done = False
        while done == False:
            try:
                if self._stomp.check_msg_done(id):
                    done = True
            except Exception:
                done = True
            except:
                pass

    def execute_and_check(self, bml_list):
        #return self.check_done(self.execute(bml_list))
        return self.execute(bml_list)