Move configuration file loading from prompt to tools
This commit is contained in:
parent
e17996d858
commit
b6c5bf4f10
@ -20,17 +20,10 @@ import imp
|
|||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
from tools.config import load_file
|
||||||
|
|
||||||
logger = logging.getLogger("nemubot.prompt.builtins")
|
logger = logging.getLogger("nemubot.prompt.builtins")
|
||||||
|
|
||||||
from server.IRC import IRC as IRCServer
|
|
||||||
import xmlparser
|
|
||||||
|
|
||||||
|
|
||||||
def get_boolean(d, k):
|
|
||||||
return (k in d and
|
|
||||||
mod["autoload"].lower() != "false" and
|
|
||||||
mod["autoload"].lower() != "off")
|
|
||||||
|
|
||||||
|
|
||||||
def end(toks, context, prompt):
|
def end(toks, context, prompt):
|
||||||
"""Quit the prompt for reload or exit"""
|
"""Quit the prompt for reload or exit"""
|
||||||
@ -67,92 +60,6 @@ def liste(toks, context, prompt):
|
|||||||
print (" Please give a list to show: servers, ...")
|
print (" Please give a list to show: servers, ...")
|
||||||
|
|
||||||
|
|
||||||
def load_file(filename, context):
|
|
||||||
if os.path.isfile(filename):
|
|
||||||
config = xmlparser.parse_file(filename)
|
|
||||||
|
|
||||||
# This is a true nemubot configuration file, load it!
|
|
||||||
if config.getName() == "nemubotconfig":
|
|
||||||
# Preset each server in this file
|
|
||||||
for server in config.getNodes("server"):
|
|
||||||
opts = {
|
|
||||||
"host": server["host"],
|
|
||||||
"ssl": server.hasAttribute("ssl") and server["ssl"].lower() == "true",
|
|
||||||
|
|
||||||
"nick": server["nick"] if server.hasAttribute("nick") else config["nick"],
|
|
||||||
"owner": server["owner"] if server.hasAttribute("owner") else config["owner"],
|
|
||||||
}
|
|
||||||
|
|
||||||
# Optional keyword arguments
|
|
||||||
for optional_opt in [ "port", "username", "realname",
|
|
||||||
"password", "encoding", "caps" ]:
|
|
||||||
if server.hasAttribute(optional_opt):
|
|
||||||
opts[optional_opt] = server[optional_opt]
|
|
||||||
elif optional_opt in config:
|
|
||||||
opts[optional_opt] = config[optional_opt]
|
|
||||||
|
|
||||||
# Command to send on connection
|
|
||||||
if "on_connect" in server:
|
|
||||||
def on_connect():
|
|
||||||
yield server["on_connect"]
|
|
||||||
opts["on_connect"] = on_connect
|
|
||||||
|
|
||||||
# Channels to autojoin on connection
|
|
||||||
if server.hasNode("channel"):
|
|
||||||
opts["channels"] = list()
|
|
||||||
for chn in server.getNodes("channel"):
|
|
||||||
opts["channels"].append((chn["name"], chn["password"])
|
|
||||||
if chn["password"] is not None
|
|
||||||
else chn["name"])
|
|
||||||
|
|
||||||
# Server/client capabilities
|
|
||||||
if "caps" in server or "caps" in config:
|
|
||||||
capsl = (server["caps"] if server.hasAttribute("caps")
|
|
||||||
else config["caps"]).lower()
|
|
||||||
if capsl == "no" or capsl == "off" or capsl == "false":
|
|
||||||
opts["caps"] = None
|
|
||||||
else:
|
|
||||||
opts["caps"] = capsl.split(',')
|
|
||||||
else:
|
|
||||||
opts["caps"] = list()
|
|
||||||
|
|
||||||
# Bind the protocol asked to the corresponding implementation
|
|
||||||
if "protocol" not in server or server["protocol"] == "irc":
|
|
||||||
srvcls = IRCServer
|
|
||||||
else:
|
|
||||||
raise Exception("Unhandled protocol '%s'" %
|
|
||||||
server["protocol"])
|
|
||||||
|
|
||||||
# Initialize the server
|
|
||||||
srv = srvcls(**opts)
|
|
||||||
|
|
||||||
# Add the server in the context
|
|
||||||
if context.add_server(srv, get_boolean(server, "autoconnect")):
|
|
||||||
print("Server '%s' successfully added." % srv.id)
|
|
||||||
else:
|
|
||||||
print("Can't add server '%s'." % srv.id)
|
|
||||||
|
|
||||||
# Load module and their configuration
|
|
||||||
for mod in config.getNodes("module"):
|
|
||||||
context.modules_configuration[mod["name"]] = mod
|
|
||||||
if get_boolean(mod, "autoload"):
|
|
||||||
__import__(mod["name"])
|
|
||||||
|
|
||||||
# Load files asked by the configuration file
|
|
||||||
for load in config.getNodes("include"):
|
|
||||||
load_file(load["path"], context)
|
|
||||||
|
|
||||||
# Other formats
|
|
||||||
else:
|
|
||||||
print (" Can't load `%s'; this is not a valid nemubot "
|
|
||||||
"configuration file." % filename)
|
|
||||||
|
|
||||||
# Unexisting file, assume a name was passed, import the module!
|
|
||||||
else:
|
|
||||||
tt = __import__(filename)
|
|
||||||
imp.reload(tt)
|
|
||||||
|
|
||||||
|
|
||||||
def load(toks, context, prompt):
|
def load(toks, context, prompt):
|
||||||
"""Load an XML configuration file"""
|
"""Load an XML configuration file"""
|
||||||
if len(toks) > 1:
|
if len(toks) > 1:
|
||||||
|
|||||||
133
tools/config.py
Normal file
133
tools/config.py
Normal file
@ -0,0 +1,133 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
# Nemubot is a modulable IRC bot, built around XML configuration files.
|
||||||
|
# Copyright (C) 2012 Mercier Pierre-Olivier
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU Affero General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU Affero General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU Affero General Public License
|
||||||
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
|
||||||
|
import xmlparser
|
||||||
|
|
||||||
|
logger = logging.getLogger("nemubot.tools.config")
|
||||||
|
|
||||||
|
|
||||||
|
def get_boolean(d, k, default=False):
|
||||||
|
return ((k in d and d[k].lower() != "false" and d[k].lower() != "off") or
|
||||||
|
(k not in d and default))
|
||||||
|
|
||||||
|
|
||||||
|
def _load_server(config, xmlnode):
|
||||||
|
"""Load a server configuration
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
config -- the global configuration
|
||||||
|
xmlnode -- the current server configuration node
|
||||||
|
"""
|
||||||
|
|
||||||
|
opts = {
|
||||||
|
"host": xmlnode["host"],
|
||||||
|
"ssl": xmlnode.hasAttribute("ssl") and xmlnode["ssl"].lower() == "true",
|
||||||
|
|
||||||
|
"nick": xmlnode["nick"] if xmlnode.hasAttribute("nick") else config["nick"],
|
||||||
|
"owner": xmlnode["owner"] if xmlnode.hasAttribute("owner") else config["owner"],
|
||||||
|
}
|
||||||
|
|
||||||
|
# Optional keyword arguments
|
||||||
|
for optional_opt in [ "port", "username", "realname",
|
||||||
|
"password", "encoding", "caps" ]:
|
||||||
|
if xmlnode.hasAttribute(optional_opt):
|
||||||
|
opts[optional_opt] = xmlnode[optional_opt]
|
||||||
|
elif optional_opt in config:
|
||||||
|
opts[optional_opt] = config[optional_opt]
|
||||||
|
|
||||||
|
# Command to send on connection
|
||||||
|
if "on_connect" in xmlnode:
|
||||||
|
def on_connect():
|
||||||
|
yield xmlnode["on_connect"]
|
||||||
|
opts["on_connect"] = on_connect
|
||||||
|
|
||||||
|
# Channels to autojoin on connection
|
||||||
|
if xmlnode.hasNode("channel"):
|
||||||
|
opts["channels"] = list()
|
||||||
|
for chn in xmlnode.getNodes("channel"):
|
||||||
|
opts["channels"].append((chn["name"], chn["password"])
|
||||||
|
if chn["password"] is not None
|
||||||
|
else chn["name"])
|
||||||
|
|
||||||
|
# Server/client capabilities
|
||||||
|
if "caps" in xmlnode or "caps" in config:
|
||||||
|
capsl = (xmlnode["caps"] if xmlnode.hasAttribute("caps")
|
||||||
|
else config["caps"]).lower()
|
||||||
|
if capsl == "no" or capsl == "off" or capsl == "false":
|
||||||
|
opts["caps"] = None
|
||||||
|
else:
|
||||||
|
opts["caps"] = capsl.split(',')
|
||||||
|
else:
|
||||||
|
opts["caps"] = list()
|
||||||
|
|
||||||
|
# Bind the protocol asked to the corresponding implementation
|
||||||
|
if "protocol" not in xmlnode or xmlnode["protocol"] == "irc":
|
||||||
|
from server.IRC import IRC as IRCServer
|
||||||
|
srvcls = IRCServer
|
||||||
|
else:
|
||||||
|
raise Exception("Unhandled protocol '%s'" %
|
||||||
|
xmlnode["protocol"])
|
||||||
|
|
||||||
|
# Initialize the server
|
||||||
|
return srvcls(**opts)
|
||||||
|
|
||||||
|
|
||||||
|
def load_file(filename, context):
|
||||||
|
"""Load the configuration file
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
filename -- the path to the file to load
|
||||||
|
"""
|
||||||
|
|
||||||
|
if os.path.isfile(filename):
|
||||||
|
config = xmlparser.parse_file(filename)
|
||||||
|
|
||||||
|
# This is a true nemubot configuration file, load it!
|
||||||
|
if config.getName() == "nemubotconfig":
|
||||||
|
# Preset each server in this file
|
||||||
|
for server in config.getNodes("server"):
|
||||||
|
srv = _load_server(config, server)
|
||||||
|
|
||||||
|
# Add the server in the context
|
||||||
|
if context.add_server(srv, get_boolean(server, "autoconnect")):
|
||||||
|
print("Server '%s' successfully added." % srv.id)
|
||||||
|
else:
|
||||||
|
print("Can't add server '%s'." % srv.id)
|
||||||
|
|
||||||
|
# Load module and their configuration
|
||||||
|
for mod in config.getNodes("module"):
|
||||||
|
context.modules_configuration[mod["name"]] = mod
|
||||||
|
if get_boolean(mod, "autoload", default=True):
|
||||||
|
__import__(mod["name"])
|
||||||
|
|
||||||
|
# Load files asked by the configuration file
|
||||||
|
for load in config.getNodes("include"):
|
||||||
|
load_file(load["path"], context)
|
||||||
|
|
||||||
|
# Other formats
|
||||||
|
else:
|
||||||
|
print (" Can't load `%s'; this is not a valid nemubot "
|
||||||
|
"configuration file." % filename)
|
||||||
|
|
||||||
|
# Unexisting file, assume a name was passed, import the module!
|
||||||
|
else:
|
||||||
|
tt = __import__(filename)
|
||||||
|
imp.reload(tt)
|
||||||
Loading…
Reference in New Issue
Block a user