Split plugin outout on newlines and ensure plugins folder exists

pull/5/head
Sina Mashek 2017-11-27 07:10:40 +02:00
parent 6e47d13361
commit cf91e90451
2 changed files with 17 additions and 6 deletions

View File

@ -1,11 +1,14 @@
import imp
import os
import ssl
import time
import irc.bot
irc.client.ServerConnection.buffer_class.errors = 'replace'
class Message:
def __init__(self, channel, nick, cmd, arg, botnick, ops):
self.channel = channel
@ -45,6 +48,9 @@ class Bot(irc.bot.SingleServerIRCBot):
setattr(self, a, kwarguments[a])
def load_plugins(self):
# ensure plugin folder exists
if not os.path.exists(self.plugin_dir):
os.makedirs(self.plugin_dir)
# load all plugins
plugins = []
for m in os.listdir(self.plugin_dir):
@ -114,8 +120,10 @@ class Bot(irc.bot.SingleServerIRCBot):
print(e)
if output:
for msg in output.msg:
if output.msg_type == 'message':
c.privmsg(chan, output.msg)
c.privmsg(chan, msg)
elif output.msg_type == 'action':
c.action(chan, output.msg)
c.action(chan, msg)
time.sleep(.5)

View File

@ -3,7 +3,10 @@ cmds = []
class Output:
def __init__(self, msg_type, msg):
self.msg_type = msg_type
self.msg = msg
self.msg = self.sanitize(msg)
def sanitize(self, msg):
return msg.splitlines()
def action(msg):