def __init__(self, transport, debug=False): self.transport = transport # initialize logging if debug: logger.setLevel(logging.DEBUG) else: logger.setLevel(logging.INFO) # stream handler stream_handler = logging.StreamHandler() stream_formatter = logging.Formatter(STREAM_LOG_FORMAT) stream_handler.setFormatter(stream_formatter) # file handler file_handler = logging.FileHandler(FILE_LOG) file_formatter = logging.Formatter(FILE_LOG_FORMAT) file_handler.setFormatter(file_formatter) # add handlers logger.addHandler(stream_handler) logger.addHandler(file_handler) # function to instantiate command and parsing objects instantiator = lambda class_: class_(self) # load modules self.commands = get_modules_objects(COMMANDS_DIR, f=instantiator) self.keywords = get_modules_objects(PARSING_DIR, f=instantiator)
def __init__(self, *args): super().__init__(*args) self.media = self.init_media() self.siteinfo = Siteinfo(self.msg_formats) self.sources = get_modules_objects(SOURCES_PATH, f=lambda x: x(self.bot))
def setup(): """Call all setup modules""" setup_objects = get_modules_objects(SETUP_DIR) for setup in setup_objects: print("Setting up {0}...".format(setup.__class__.__name__)) setup.setup() print("Setup complete!")