Ejemplo n.º 1
0
    def __init__(self):
        self.__session_classes = []
        self.__autostart_sessions = []
        self.__config = XigConfig()
        self.__sched = SchedAsync("xig_sched", self)

        self.helpfile = ""
        self.setLogLevels()

        self.__quit_flag = False
        self.__io_kernel = XigIOKernel(xig_core=self)
Ejemplo n.º 2
0
    def __init__(self):
        self.__session_classes = []
        self.__autostart_sessions = []
        self.__config = XigConfig()
        self.__sched = SchedAsync("xig_sched", self)

        self.helpfile = ""
        self.setLogLevels()

        self.__quit_flag = False
        self.__io_kernel = XigIOKernel(xig_core=self)
Ejemplo n.º 3
0
class Xig(object):
    def __init__(self):
        self.__session_classes = []
        self.__autostart_sessions = []
        self.__config = XigConfig()
        self.__sched = SchedAsync("xig_sched", self)

        self.helpfile = ""
        self.setLogLevels()

        self.__quit_flag = False
        self.__io_kernel = XigIOKernel(xig_core=self)

    def setLogLevels(self):
        
        log_level_map = {
            "debug": logging.DEBUG,
            "info":  logging.INFO,
            "warning": logging.WARNING,
            "error": logging.ERROR,
            "critical": logging.CRITICAL,
        }
        
        log_level_global = getattr(self.__config,
                               "log_level_global", "info").lower()
        log_level_io_kernel = getattr(self.__config,
                               "log_level_io_kernel", "debug").lower()
        
        if log_level_global in log_level_map:
            logger.setLevel(log_level_map[log_level_global]) 
        if log_level_io_kernel in log_level_map:
            logging.getLogger("xig.io_kernel").setLevel(log_level_map[log_level_io_kernel])
        

    def __gcTask(self):
        """A periodic task which invokes the garbage collector"""
        gc.collect()
        self.scheduleAfter(self.__config.global_gc_interval, self.__gcTask)

    def quit(self):
        self.__quit_flag = True

    def isXBeeXmitStatusSupported(self):
        if sys.platform == "digix3":
            return False
        return True

    def getShortName(self):
        return SHORTNAME

    def getVersion(self):
        return VERSION

    def getSessionClasses(self):
        # XigSession must always be last for proper command handling:
        return self.__session_classes + [sessions.XigSession]
 
    def getAutostartSessions(self, obj_type=None):
        if obj_type is None:
            return self.__autostart_sessions

        for obj in self.__autostart_sessions:
            if isinstance(obj, obj_type):
                return obj
        return None

    def getConfig(self):
        return self.__config

    def ioSampleSubcriberAdd(self, func):
        """\
        Called by another object if it wishes to receive a
        callback for each XBee I/O Sample packet.

        func argument must be a function which accepts two
        arguments buf and addr.
        """
        self.__io_kernel.ioSubscriberAdd(func)

    def ioSampleSubcriberRemove(self, func):
        """\
        Called by an object if it wishes to stop receiving
       callbacks for each XBee I/O Sample packet.

        The func argument must match the argument given
        previously to ioSampleSubscriberAdd()
        """
        self.__io_kernel.ioSubscriberRemove(func)

    def scheduleAfter(self, delay, action, *args):
        return self.__sched.schedule_after(delay, action, *args)


    def go(self):
        status = 255
        
        logger.info("Loading and initializing configured session types...")
        for session_type in self.__config.session_types:
            if session_type not in sessions.SESSION_MODEL_CLASS_MAP:
                logger.warn('unknown session type "%s", skipping.' % session_type)
                continue
            session_classes = sessions.SESSION_MODEL_CLASS_MAP[session_type]
            logger.debug("Loading %s from %s..." % (repr(session_classes), session_type))
            for session_class in session_classes:
                class_obj = sessions.classloader(module_name="sessions." + session_type, object_name=session_class)
                if issubclass(class_obj, sessions.AbstractSession):
                    self.__session_classes.append(class_obj)
                elif issubclass(class_obj, sessions.AbstractAutostartSession):
                    self.__autostart_sessions.append(class_obj(xig_core=self))

        logger.debug("Formatting help text...")
        sessionHelpText = "".join(map(lambda c: c.commandHelpText(), self.__session_classes))
        autostartHelpText = " (none)\n"
        if len(self.__autostart_sessions):
            autostartHelpText = "".join(map(lambda c: c.helpText(), self.__autostart_sessions))

        self.helpfile = (string.Template(HELPFILE_TEMPLATE)
                               .substitute(
                                  appName=NAME, appVersion=VERSION,
                                  ipAddr="127.0.0.1",
                                  sessionHelpText=sessionHelpText,
                                  autostartHelpText=autostartHelpText))
        self.helpfile = self.helpfile.replace('\n', '\r\n')
        logger.info("Starting scheduler...")
        self.__sched.start()
        self.__gcTask()
        logger.info("XIG startup complete, ready to serve requests.")
        while not self.__quit_flag:
            try:
                self.__io_kernel.ioLoop(timeout=1)
            except Exception, e:
                logger.error("Exception during I/O loop: %s"%e)
                traceback.print_exc(file=sys.stdout)
        logger.info("Shutting down.")
        if self.__quit_flag:
            status = 0
        # run one last time, with feeling:
        self.__io_kernel.ioLoop(timeout=5.0)
        self.__io_kernel.shutdown()

        # unregister rci callback
        ##TODO: move this elsewhere?
        #try:
        #    rci.stop_rci_callback("xig")
        #except:
        #    pass

        # From Digi support, to prevent the dreaded:
        # "zipimport.ZipImportError: bad local file header in WEB/python/_xig.zip" error
        import zipimport
        syspath_backup = list(sys.path)
        zipimport._zip_directory_cache.clear()
        sys.path = syspath_backup

        return status
Ejemplo n.º 4
0
class Xig(object):
    def __init__(self):
        self.__session_classes = []
        self.__autostart_sessions = []
        self.__config = XigConfig()
        self.__sched = SchedAsync("xig_sched", self)

        self.helpfile = ""
        self.setLogLevels()

        self.__quit_flag = False
        self.__io_kernel = XigIOKernel(xig_core=self)

    def setLogLevels(self):

        log_level_map = {
            "debug": logging.DEBUG,
            "info": logging.INFO,
            "warning": logging.WARNING,
            "error": logging.ERROR,
            "critical": logging.CRITICAL,
        }

        log_level_global = getattr(self.__config, "log_level_global",
                                   "info").lower()
        log_level_io_kernel = getattr(self.__config, "log_level_io_kernel",
                                      "debug").lower()

        if log_level_global in log_level_map:
            logger.setLevel(log_level_map[log_level_global])
        if log_level_io_kernel in log_level_map:
            logging.getLogger("xig.io_kernel").setLevel(
                log_level_map[log_level_io_kernel])

    def __gcTask(self):
        """A periodic task which invokes the garbage collector"""
        gc.collect()
        self.scheduleAfter(self.__config.global_gc_interval, self.__gcTask)

    def quit(self):
        self.__quit_flag = True

    def isXBeeXmitStatusSupported(self):
        if sys.platform == "digix3":
            return False

        return True

    def __getLocalIPLinux(self):
        import socket
        try:
            s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
            s.connect(("50.56.41.152", 3197))  # No connect is actually made,
            # address is my.idigi.com, EDP port
            return s.getsockname()[0]
        except:
            return "127.0.0.1"

    def getLocalIP(self):
        if sys.platform.startswith("linux"):
            return self.__getLocalIPLinux()

        query_string = """\
        <rci_request version="1.1">
            <query_state><boot_stats/></query_state>
        </rci_request>"""
        response = rci.process_request(query_string)
        ip_beg, ip_end = (0, 0)
        if sys.platform == "digix3":
            ip_beg = response.find("<ip_address>") + 1
            ip_end = response.find("</ip_address>")
        else:
            ip_beg = response.find("<ip>")
            ip_end = response.find("</ip>")

        return response[ip_beg + 4:ip_end].strip()

    def getShortName(self):
        return SHORTNAME

    def getVersion(self):
        return VERSION

    def getSessionClasses(self):
        # XigSession must always be last for proper command handling:
        return self.__session_classes + [sessions.XigSession]

    def getAutostartSessions(self, obj_type=None):
        if obj_type is None:
            return self.__autostart_sessions

        for obj in self.__autostart_sessions:
            if isinstance(obj, obj_type):
                return obj
        return None

    def getConfig(self):
        return self.__config

    def ioSampleSubcriberAdd(self, func):
        """\
        Called by another object if it wishes to receive a
        callback for each XBee I/O Sample packet.

        func argument must be a function which accepts two
        arguments buf and addr.
        """
        self.__io_kernel.ioSubscriberAdd(func)

    def ioSampleSubcriberRemove(self, func):
        """\
        Called by an object if it wishes to stop receiving
       callbacks for each XBee I/O Sample packet.

        The func argument must match the argument given
        previously to ioSampleSubscriberAdd()
        """
        self.__io_kernel.ioSubscriberRemove(func)

    def scheduleAfter(self, delay, action, *args):
        return self.__sched.schedule_after(delay, action, *args)

    def go(self):
        status = 255

        logger.info("Loading and initializing configured session types...")
        for session_type in self.__config.session_types:
            if session_type not in sessions.SESSION_MODEL_CLASS_MAP:
                logger.warn('unknown session type "%s", skipping.' %
                            session_type)
                continue
            session_classes = sessions.SESSION_MODEL_CLASS_MAP[session_type]
            logger.debug("Loading %s from %s..." %
                         (repr(session_classes), session_type))
            for session_class in session_classes:
                class_obj = sessions.classloader(module_name="sessions." +
                                                 session_type,
                                                 object_name=session_class)
                if issubclass(class_obj, sessions.AbstractSession):
                    self.__session_classes.append(class_obj)
                elif issubclass(class_obj, sessions.AbstractAutostartSession):
                    self.__autostart_sessions.append(class_obj(xig_core=self))

        logger.debug("Formatting help text...")
        sessionHelpText = "".join(
            map(lambda c: c.commandHelpText(), self.__session_classes))
        autostartHelpText = " (none)\n"
        if len(self.__autostart_sessions):
            autostartHelpText = "".join(
                map(lambda c: c.helpText(), self.__autostart_sessions))

        self.helpfile = (string.Template(HELPFILE_TEMPLATE).substitute(
            appName=NAME,
            appVersion=VERSION,
            ipAddr=self.getLocalIP(),
            sessionHelpText=sessionHelpText,
            autostartHelpText=autostartHelpText))
        self.helpfile = self.helpfile.replace('\n', '\r\n')
        logger.info("Starting scheduler...")
        self.__sched.start()
        self.__gcTask()
        logger.info("XIG startup complete, ready to serve requests.")
        while not self.__quit_flag:
            try:
                self.__io_kernel.ioLoop(timeout=1)
            except Exception, e:
                logger.error("Exception during I/O loop: %s" % e)
                traceback.print_exc(file=sys.stdout)
        logger.info("Shutting down.")
        if self.__quit_flag:
            status = 0
        # run one last time, with feeling:
        self.__io_kernel.ioLoop(timeout=5.0)
        self.__io_kernel.shutdown()

        # unregister rci callback
        #TODO: move this elsewhere?
        try:
            rci.stop_rci_callback("xig")
        except:
            pass

        # From Digi support, to prevent the dreaded:
        # "zipimport.ZipImportError: bad local file header in WEB/python/_xig.zip" error
        import zipimport
        syspath_backup = list(sys.path)
        zipimport._zip_directory_cache.clear()
        sys.path = syspath_backup

        return status