Refactor configuration loading
This commit is contained in:
parent
e49312e63e
commit
45fe5b2156
@ -118,10 +118,10 @@ def main():
|
||||
|
||||
# Create bot context
|
||||
from nemubot import datastore
|
||||
from nemubot.bot import Bot, sync_act
|
||||
from nemubot.bot import Bot
|
||||
context = Bot(modules_paths=modules_paths,
|
||||
data_store=datastore.XML(args.data_path),
|
||||
verbosity=args.verbose)
|
||||
debug=args.verbose > 0)
|
||||
|
||||
if args.no_connect:
|
||||
context.noautoconnect = True
|
||||
@ -133,10 +133,34 @@ def main():
|
||||
|
||||
# Load requested configuration files
|
||||
for path in args.files:
|
||||
if os.path.isfile(path):
|
||||
sync_act("loadconf", path)
|
||||
else:
|
||||
if not os.path.isfile(path):
|
||||
logger.error("%s is not a readable file", path)
|
||||
continue
|
||||
|
||||
config = load_config(path)
|
||||
|
||||
# Preset each server in this file
|
||||
for server in config.servers:
|
||||
srv = server.server(config)
|
||||
# Add the server in the context
|
||||
if context.add_server(srv):
|
||||
logger.info("Server '%s' successfully added.", srv.name)
|
||||
else:
|
||||
logger.error("Can't add server '%s'.", srv.name)
|
||||
|
||||
# Load module and their configuration
|
||||
for mod in config.modules:
|
||||
context.modules_configuration[mod.name] = mod
|
||||
if mod.autoload:
|
||||
try:
|
||||
__import__(mod.name)
|
||||
except:
|
||||
logger.exception("Exception occurs when loading module"
|
||||
" '%s'", mod.name)
|
||||
|
||||
# Load files asked by the configuration file
|
||||
args.files += config.includes
|
||||
|
||||
|
||||
if args.module:
|
||||
for module in args.module:
|
||||
@ -205,5 +229,31 @@ def main():
|
||||
sys.exit(0)
|
||||
|
||||
|
||||
def load_config(filename):
|
||||
"""Load a configuration file
|
||||
|
||||
Arguments:
|
||||
filename -- the path to the file to load
|
||||
"""
|
||||
|
||||
from nemubot.channel import Channel
|
||||
from nemubot import config
|
||||
from nemubot.tools.xmlparser import XMLParser
|
||||
|
||||
try:
|
||||
p = XMLParser({
|
||||
"nemubotconfig": config.Nemubot,
|
||||
"server": config.Server,
|
||||
"channel": Channel,
|
||||
"module": config.Module,
|
||||
"include": config.Include,
|
||||
})
|
||||
return p.parse_file(filename)
|
||||
except:
|
||||
logger.exception("Can't load `%s'; this is not a valid nemubot "
|
||||
"configuration file.", filename)
|
||||
return None
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
@ -40,14 +40,14 @@ class Bot(threading.Thread):
|
||||
"""Class containing the bot context and ensuring key goals"""
|
||||
|
||||
def __init__(self, ip="127.0.0.1", modules_paths=list(),
|
||||
data_store=datastore.Abstract(), verbosity=0):
|
||||
data_store=datastore.Abstract(), debug=False):
|
||||
"""Initialize the bot context
|
||||
|
||||
Keyword arguments:
|
||||
ip -- The external IP of the bot (default: 127.0.0.1)
|
||||
modules_paths -- Paths to all directories where looking for modules
|
||||
data_store -- An instance of the nemubot datastore for bot's modules
|
||||
verbosity -- verbosity level
|
||||
debug -- enable debug
|
||||
"""
|
||||
|
||||
super().__init__(name="Nemubot main")
|
||||
@ -56,7 +56,7 @@ class Bot(threading.Thread):
|
||||
__version__,
|
||||
sys.version_info.major, sys.version_info.minor, sys.version_info.micro)
|
||||
|
||||
self.verbosity = verbosity
|
||||
self.debug = debug
|
||||
self.stop = None
|
||||
|
||||
# External IP for accessing this bot
|
||||
@ -149,6 +149,10 @@ class Bot(threading.Thread):
|
||||
self.cnsr_thrd_size = -1
|
||||
|
||||
|
||||
def __del__(self):
|
||||
self.datastore.close()
|
||||
|
||||
|
||||
def run(self):
|
||||
global sync_queue
|
||||
|
||||
@ -218,12 +222,6 @@ class Bot(threading.Thread):
|
||||
elif action == "launch_consumer":
|
||||
pass # This is treated after the loop
|
||||
|
||||
elif action == "loadconf":
|
||||
for path in args:
|
||||
logger.debug("Load configuration from %s", path)
|
||||
self.load_file(path)
|
||||
logger.info("Configurations successfully loaded")
|
||||
|
||||
sync_queue.task_done()
|
||||
|
||||
|
||||
@ -240,64 +238,6 @@ class Bot(threading.Thread):
|
||||
|
||||
|
||||
|
||||
# Config methods
|
||||
|
||||
def load_file(self, filename):
|
||||
"""Load a configuration file
|
||||
|
||||
Arguments:
|
||||
filename -- the path to the file to load
|
||||
"""
|
||||
|
||||
import os
|
||||
|
||||
# Unexisting file, assume a name was passed, import the module!
|
||||
if not os.path.isfile(filename):
|
||||
return self.import_module(filename)
|
||||
|
||||
from nemubot.channel import Channel
|
||||
from nemubot import config
|
||||
from nemubot.tools.xmlparser import XMLParser
|
||||
|
||||
try:
|
||||
p = XMLParser({
|
||||
"nemubotconfig": config.Nemubot,
|
||||
"server": config.Server,
|
||||
"channel": Channel,
|
||||
"module": config.Module,
|
||||
"include": config.Include,
|
||||
})
|
||||
config = p.parse_file(filename)
|
||||
except:
|
||||
logger.exception("Can't load `%s'; this is not a valid nemubot "
|
||||
"configuration file." % filename)
|
||||
return False
|
||||
|
||||
# Preset each server in this file
|
||||
for server in config.servers:
|
||||
srv = server.server(config)
|
||||
# Add the server in the context
|
||||
if self.add_server(srv, server.autoconnect):
|
||||
logger.info("Server '%s' successfully added." % srv.name)
|
||||
else:
|
||||
logger.error("Can't add server '%s'." % srv.name)
|
||||
|
||||
# Load module and their configuration
|
||||
for mod in config.modules:
|
||||
self.modules_configuration[mod.name] = mod
|
||||
if mod.autoload:
|
||||
try:
|
||||
__import__(mod.name)
|
||||
except:
|
||||
logger.exception("Exception occurs when loading module"
|
||||
" '%s'", mod.name)
|
||||
|
||||
|
||||
# Load files asked by the configuration file
|
||||
for load in config.includes:
|
||||
self.load_file(load.path)
|
||||
|
||||
|
||||
# Events methods
|
||||
|
||||
def add_event(self, evt, eid=None, module_src=None):
|
||||
@ -581,8 +521,6 @@ class Bot(threading.Thread):
|
||||
for cnsr in k:
|
||||
cnsr.stop = True
|
||||
|
||||
self.datastore.close()
|
||||
|
||||
if self.stop is False or sync_queue is not None:
|
||||
self.stop = True
|
||||
sync_act("end")
|
||||
|
||||
@ -108,7 +108,7 @@ class ModuleContext(_ModuleContext):
|
||||
self.config = context.modules_configuration[self.module_name]
|
||||
|
||||
self.context = context
|
||||
self.debug = context.verbosity > 0
|
||||
self.debug = context.debug
|
||||
|
||||
|
||||
def load_data(self):
|
||||
|
||||
Loading…
Reference in New Issue
Block a user