Source code for netdef.utils

import logging
import os
import sys
from logging.handlers import RotatingFileHandler

from .Engines.BaseEngine import BaseEngine
from .Shared.Shared import Shared

# for assertion:
from .Shared.SharedConfig import Config

[docs]def setup_logging(config): """ Parse the config file for: .. code-block:: ini [logging] logglevel loggformat loggdatefmt loggfile to_console to_file Then the logging module is set according to the configs :param config: instance of :class:`netdef.Shared.SharedConfig.Config` Example:: ... from netdef.Shared import Shared from netdef.utils import setup_logging shared = Shared.Shared("First-App", install_path, proj_path, config_string) setup_logging(shared.config) ... """ assert isinstance(config, Config) logglevel = logging.INFO loggformat = logging.BASIC_FORMAT loggdatefmt = "%Y-%m-%d %H:%M:%S" if config: logglevel = config.config("logging", "logglevel", logglevel) loggformat = config.config("logging", "loggformat", loggformat) loggdatefmt = config.config("logging", "loggdatefmt", loggdatefmt) loggfile = config.config("logging", "loggfile", "log/application.log") to_console = config.config("logging", "to_console", 1) to_file = config.config("logging", "to_file", 0) handlers = [] if to_console: handlers.append(logging.StreamHandler()) if to_file: handlers.append( RotatingFileHandler(loggfile, maxBytes=10485760, backupCount=10) ) else: handlers = None logging.basicConfig( handlers=handlers, level=logglevel, format=loggformat, datefmt=loggdatefmt ) if config: def exception_logger(exc_type, exc_value, exc_traceback): logging.error("exception", exc_info=(exc_type, exc_value, exc_traceback)) sys.excepthook = exception_logger for package_name, level in config.get_dict("logginglevels").items(): logging.getLogger(package_name).setLevel(int(level))
[docs]def handle_restart(shared, engine): """ By calling this function your application will restart on SystemExit if shared.restart_on_exit is True. :param shared: instance of :class:`netdef.Shared.Shared` :param engine: instance or subclass of :class:`netdef.Engines.BaseEngine.BaseEngine` Example:: from netdef.utils import handle_restart ... engine.init() engine.start() engine.block() # until ctrl-c or SIG_TERM engine.stop() handle_restart(shared, engine) """ assert isinstance(shared, Shared) assert isinstance(engine, BaseEngine) # dersom restart-knappen i webadmin er benyttet: if shared.restart_on_exit: engine.wait() # venter på at alle trådene er stoppet if sys.argv[0].endswith(""): # support restart when "python -m APP" args = [sys.executable, "-m", __package__] + sys.argv[1:] else: # support entry_points from args = sys.argv try: os.execv(args[0], args) except (PermissionError, OSError) as error: # support python os.execv(sys.executable, [sys.executable] + args)