46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
import yaml
|
|
|
|
from .bot import Bot
|
|
from .settings import logger
|
|
|
|
|
|
class Loader:
|
|
def __init__(self, filename):
|
|
logger.info('Loading configuration from ' + filename)
|
|
with open(filename) as f:
|
|
self.dict = yaml.load(f)
|
|
self.bots = []
|
|
|
|
def load_bot_template(self, name, channel, serv, port):
|
|
template = self.dict['bots'][name]
|
|
b = Bot(nickname=name)
|
|
b.server = serv
|
|
b.channel = channel
|
|
b.port = port
|
|
for ping in template.get('on_ping', []):
|
|
b.add_ping(ping)
|
|
for ping in template.get('on_ping_python', []):
|
|
b.add_python_ping(ping)
|
|
|
|
matches = template.get('on_match', [])
|
|
for match in matches:
|
|
b.add_reaction(match, matches[match])
|
|
for match in template.get('on_match_python', []):
|
|
b.add_python_reaction(match, matches[match])
|
|
|
|
b.on_join = template.get('on_join', None)
|
|
b.min_time = template.get('min_time', 20)
|
|
|
|
return b
|
|
|
|
def load_bots(self):
|
|
for channel in self.dict['channels']:
|
|
name = channel['channel']
|
|
serv = channel['server']
|
|
port = channel.get('port', 6667)
|
|
bots_name = channel['bots']
|
|
|
|
for nickname in bots_name:
|
|
b = self.load_bot_template(nickname, name, serv, port)
|
|
self.bots.append(b)
|