Refactor configuration loading
This commit is contained in:
parent
e49312e63e
commit
45fe5b2156
@ -118,10 +118,10 @@ def main():
|
|||||||
|
|
||||||
# Create bot context
|
# Create bot context
|
||||||
from nemubot import datastore
|
from nemubot import datastore
|
||||||
from nemubot.bot import Bot, sync_act
|
from nemubot.bot import Bot
|
||||||
context = Bot(modules_paths=modules_paths,
|
context = Bot(modules_paths=modules_paths,
|
||||||
data_store=datastore.XML(args.data_path),
|
data_store=datastore.XML(args.data_path),
|
||||||
verbosity=args.verbose)
|
debug=args.verbose > 0)
|
||||||
|
|
||||||
if args.no_connect:
|
if args.no_connect:
|
||||||
context.noautoconnect = True
|
context.noautoconnect = True
|
||||||
@ -133,10 +133,34 @@ def main():
|
|||||||
|
|
||||||
# Load requested configuration files
|
# Load requested configuration files
|
||||||
for path in args.files:
|
for path in args.files:
|
||||||
if os.path.isfile(path):
|
if not os.path.isfile(path):
|
||||||
sync_act("loadconf", path)
|
|
||||||
else:
|
|
||||||
logger.error("%s is not a readable file", 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:
|
if args.module:
|
||||||
for module in args.module:
|
for module in args.module:
|
||||||
@ -205,5 +229,31 @@ def main():
|
|||||||
sys.exit(0)
|
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__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
@ -40,14 +40,14 @@ class Bot(threading.Thread):
|
|||||||
"""Class containing the bot context and ensuring key goals"""
|
"""Class containing the bot context and ensuring key goals"""
|
||||||
|
|
||||||
def __init__(self, ip="127.0.0.1", modules_paths=list(),
|
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
|
"""Initialize the bot context
|
||||||
|
|
||||||
Keyword arguments:
|
Keyword arguments:
|
||||||
ip -- The external IP of the bot (default: 127.0.0.1)
|
ip -- The external IP of the bot (default: 127.0.0.1)
|
||||||
modules_paths -- Paths to all directories where looking for modules
|
modules_paths -- Paths to all directories where looking for modules
|
||||||
data_store -- An instance of the nemubot datastore for bot's modules
|
data_store -- An instance of the nemubot datastore for bot's modules
|
||||||
verbosity -- verbosity level
|
debug -- enable debug
|
||||||
"""
|
"""
|
||||||
|
|
||||||
super().__init__(name="Nemubot main")
|
super().__init__(name="Nemubot main")
|
||||||
@ -56,7 +56,7 @@ class Bot(threading.Thread):
|
|||||||
__version__,
|
__version__,
|
||||||
sys.version_info.major, sys.version_info.minor, sys.version_info.micro)
|
sys.version_info.major, sys.version_info.minor, sys.version_info.micro)
|
||||||
|
|
||||||
self.verbosity = verbosity
|
self.debug = debug
|
||||||
self.stop = None
|
self.stop = None
|
||||||
|
|
||||||
# External IP for accessing this bot
|
# External IP for accessing this bot
|
||||||
@ -149,6 +149,10 @@ class Bot(threading.Thread):
|
|||||||
self.cnsr_thrd_size = -1
|
self.cnsr_thrd_size = -1
|
||||||
|
|
||||||
|
|
||||||
|
def __del__(self):
|
||||||
|
self.datastore.close()
|
||||||
|
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
global sync_queue
|
global sync_queue
|
||||||
|
|
||||||
@ -218,12 +222,6 @@ class Bot(threading.Thread):
|
|||||||
elif action == "launch_consumer":
|
elif action == "launch_consumer":
|
||||||
pass # This is treated after the loop
|
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()
|
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
|
# Events methods
|
||||||
|
|
||||||
def add_event(self, evt, eid=None, module_src=None):
|
def add_event(self, evt, eid=None, module_src=None):
|
||||||
@ -581,8 +521,6 @@ class Bot(threading.Thread):
|
|||||||
for cnsr in k:
|
for cnsr in k:
|
||||||
cnsr.stop = True
|
cnsr.stop = True
|
||||||
|
|
||||||
self.datastore.close()
|
|
||||||
|
|
||||||
if self.stop is False or sync_queue is not None:
|
if self.stop is False or sync_queue is not None:
|
||||||
self.stop = True
|
self.stop = True
|
||||||
sync_act("end")
|
sync_act("end")
|
||||||
|
@ -108,7 +108,7 @@ class ModuleContext(_ModuleContext):
|
|||||||
self.config = context.modules_configuration[self.module_name]
|
self.config = context.modules_configuration[self.module_name]
|
||||||
|
|
||||||
self.context = context
|
self.context = context
|
||||||
self.debug = context.verbosity > 0
|
self.debug = context.debug
|
||||||
|
|
||||||
|
|
||||||
def load_data(self):
|
def load_data(self):
|
||||||
|
Loading…
Reference in New Issue
Block a user