Merge branch 'master' of https://github.com/krebscode/painload

3 files changed, 1 insertions, 268 deletions

diff --git a/news/controller.py b/news/controller.py
deleted file mode 100755
index e9e94476..00000000
--- a/news/controller.py
+++ /dev/null
@@ -1,151 +0,0 @@
-from time import sleep
-import irc.bot
-import _thread
-import rssbot
-import os
-import subprocess
-
-class NewsBot(irc.bot.SingleServerIRCBot):
-    def __init__(self, name, chans=['#news'], server='ire', port=6667, timeout=60):
-        irc.bot.SingleServerIRCBot.__init__(self, [(server, port)], name, name)
-        self.name = name
-        self.server = server
-        self.port = port
-        self.chans = chans
-        self.to = timeout
-
-    def start(self):
-        self.bot = _thread.start_new_thread(irc.bot.SingleServerIRCBot.start, (self,))
-
-    def on_welcome(self, connection, event):
-        for chan in self.chans:
-            connection.join(chan)
-
-    def send(self, target, string):
-        for line in string.split('\n'):
-            self.connection.action(target, line)
-            sleep(1)
-
-    def on_privmsg(self, connection, event):
-        args_array = event.arguments[0].split()
-        answer = self.read_message(args_array)
-        self.send(event.source.nick, answer)
-
-    def on_pubmsg(self, connection, event):
-        args_array = event.arguments[0].split()
-        if len(args_array) > 0 and args_array[0][:-1]==self.name:
-            answer = self.read_message(args_array[1:])
-            self.send(event.target, answer)
-
-    def on_invite(self, connection, event):
-        for chan in event.arguments:
-            connection.join(chan)
-
-    def read_message(self, args):
-        try:
-            if args[0] in [x for x in commands.__dict__.keys() if x.find('_')]:
-                func = getattr(commands, args[0])
-                return func(args)
-            else:
-                return 'command not found'
-        except:
-            return "mimimimi"
-
-
-
-class commands():
-    def add(args): 
-        if args[1] not in bots and not args[1]==knews.name:
-            bot = rssbot.RssBot(args[2], args[1], url_shortener=url_shortener)
-            bots[args[1]] = bot
-            bot.start()
-            return "bot " + args[1] + " added"
-        else:
-            return args[1] + ' does already exist'
-
-    def delete(args):
-        bots[args[1]].stop()
-        del bots[args[1]]
-        return "bot " + args[1] + " deleted"
-
-    def rename(args):
-        if args[1] in bots:
-            if args[2] in bots:
-                return args[2] + ' already taken'
-            else:
-                bots[args[1]].connection.nick(args[2])
-                bots[args[1]].name = args[2]
-                bots[args[2]] = bots[args[1]]
-                del bots[args[1]]
-                return 'renamed ' + args[1] + ' in ' + args[2]
-        else:
-            return args[1] + ' does not exist'
-
-    def save(args):
-        output_buffer = ''
-        for bot in bots:
-            if bots[bot].loop:
-                output_buffer += bot + '|' + bots[bot].url + '|' + ' '.join(bots[bot].channels) + '\n'
-
-        F = open(feedfile, "w")
-        F.writelines(output_buffer)
-        F.close()
-
-        return "bots saved to " + feedfile
-
-    def caps(args):
-        return ' '.join([x for x in commands.__dict__.keys() if x.find('_')])
-
-    def list(args):
-        output_buffer = ''
-        for bot in bots:
-            output_buffer += bot + ' url: ' + bots[bot].url + '\n'
-        return output_buffer
-
-    def info(args):
-        if args[1] in bots:
-            output_buffer = ''
-            for data in ['title', 'link', 'updated']:
-                if data in bots[args[1]].feed.feed:
-                    output_buffer += data + ': ' + bots[args[1]].feed.feed[data] + '\n'
-            output_buffer += 'lastnew: ' + bots[args[1]].lastnew.isoformat() + '\n'
-            output_buffer += 'rssurl: ' + bots[args[1]].url
-            return output_buffer
-        else:
-            return 'bot not found'
-
-    def search(args):
-        output = subprocess.check_output(['./GfindFeeds4bot', args[1]]).decode()
-        return output
-
-    def uptime(args):
-        output = subprocess.check_output(['uptime']).decode()
-        return output
-
-feedfile = 'new_feeds'
-url_shortener = 'http://wall'
-init_channels = ['#news']
-
-if 'FEEDFILE' in os.environ:
-    feedfile = os.environ['FEEDFILE']
-
-if 'URLSHORT' in os.environ:
-    url_shortener = os.environ['URLSHORT']
-
-bots = {}
-knews = NewsBot('knews')
-
-#config file reading
-F = open(feedfile, "r")
-lines = F.readlines()
-F.close()
-
-for line in lines:
-    line = line.strip('\n')
-    linear = line.split('|')
-    bot = rssbot.RssBot(linear[1], linear[0], init_channels + linear[2].split(), url_shortener)
-    bot.start()
-    bots[linear[0]] = bot
-
-knews.start()
-
diff --git a/news/newsbot.py b/news/newsbot.py
index 5850e4e9..54a2e7d2 100644
--- a/news/newsbot.py
+++ b/news/newsbot.py
@@ -221,7 +221,7 @@ url_shortener = 'http://wall'
 init_channels = ['#news']
 
 bots = {}
-knews = NewsBot('knews')
+knews = NewsBot('knews', init_channels)
 
 #config file reading
 F = open(feedfile, "r")
diff --git a/news/rssbot.py b/news/rssbot.py
deleted file mode 100755
index 87c58781..00000000
--- a/news/rssbot.py
+++ /dev/null
@@ -1,116 +0,0 @@
-#!/usr/bin/python
-import irc.bot
-from irc.client import IRC
-import feedparser
-import threading
-import math
-import re
-import subprocess
-from datetime import datetime
-from time import sleep
-
-class RssBot(irc.bot.SingleServerIRCBot):
-    def __init__(self, rss, name, chans=['#news'], url_shortener="http://localhost", server='ire', port=6667, timeout=60):
-        irc.bot.SingleServerIRCBot.__init__(self, [(server, port)], name, name)
-        self.url = rss
-        self.name = name
-        self.server = server
-        self.port = port
-        self.chans = chans
-        self.to = timeout
-        self.oldnews = []
-        self.sendqueue = []
-        self.loop = True
-        self.lastnew = datetime.now()
-        self.url_shortener = url_shortener
-
-        def better_loop(timeout=0.2):
-            while self.loop:
-                self.ircobj.process_once(timeout)
-        self.ircobj.process_forever = better_loop
-
-
-    def start(self):
-        self.upd_loop = threading.Thread(target=self.updateloop)
-        self.bot = threading.Thread(target=irc.bot.SingleServerIRCBot.start, args=(self,))
-        self.upd_loop.start()
-        self.bot.start()
-
-    def stop(self):
-        self.ircobj.disconnect_all()
-        self.loop = False
-        del self
-
-    def updateloop(self):
-        failcount=0
-        while True:
-          try:
-              self.feed = feedparser.parse(self.url)
-              for entry in self.feed.entries:
-                  self.oldnews.append(entry.link)
-              break
-          except:
-              print(self.name + ': rss timeout occured')
-              failcount+=1
-              if failcount>20:
-                  print(self.name + ' is broken, going to die')
-                  self.stop()
-                  return
-        while self.loop:
-            try:
-                self.feed = feedparser.parse(self.url)
-                for entry in self.feed.entries:
-                    if not entry.link in self.oldnews:
-                        #try:
-                        #    self.send(entry.title + " " + entry.link + " com: " + entry.comments)
-                        #except AttributeError:
-                        shorturl = self.shortenurl(entry.link)
-                        self.sendall(entry.title + ' ' + shorturl)
-                        self.oldnews.append(entry.link)
-                        self.lastnew = datetime.now()
-            except:
-                print(self.name + ': rss timeout occured')
-            sleep(self.to)
-
-    def shortenurl(self, url):
-      while True:
-          try:
-              shorturl = subprocess.check_output(["curl", "-sS", "-F", "uri=" + url, self.url_shortener]).decode().strip('\n').strip('\r') + '#' + url.partition('://')[2].partition('/')[0]
-              return shorturl
-          except:
-              print('url shortener error')
-              sleep(1)
-
-    def last(self, target, num):
-        for feed in [x for x in self.feed.entries][:num]:
-            self.send(target, feed.title + ' ' + self.shortenurl(feed.link))
-
-    def sendall(self, string):
-        for chan in self.channels:
-            self.send(chan, string)
-
-    def send(self, target, string):
-        if self.connection.connected:
-            for line in string.split('\n'):
-                if len(line) < 450:
-                    self.connection.privmsg(target, line)
-                    sleep(1)
-                else:
-                    space = 0
-                    for x in range(math.ceil(len(line)/400)):
-                        oldspace = space
-                        space = line.find(" ", (x+1)*400, (x+1)*400+50)
-                        self.connection.privmsg(target, line[oldspace:space])
-                        sleep(1)
-        else:
-            self.connection.reconnect()
-            sleep(1)
-            self.send(string)
-
-    def on_invite(self, connection, event):
-        for chan in event.arguments:
-            connection.join(chan)
-
-    def on_welcome(self, connection, event):
-        for chan in self.chans:
-            connection.join(chan)