diff options
Diffstat (limited to 'pyload/plugins/addons/IRCInterface.py')
| -rw-r--r-- | pyload/plugins/addons/IRCInterface.py | 203 |
1 files changed, 97 insertions, 106 deletions
diff --git a/pyload/plugins/addons/IRCInterface.py b/pyload/plugins/addons/IRCInterface.py index 821f80b6c..66fd15dcc 100644 --- a/pyload/plugins/addons/IRCInterface.py +++ b/pyload/plugins/addons/IRCInterface.py @@ -26,60 +26,60 @@ import time from time import sleep from traceback import print_exc import re +from pycurl import FORM_FILE from module.plugins.Addon import Addon from module.network.RequestFactory import getURL from module.utils import formatSize from module.Api import PackageDoesNotExists, FileDoesNotExists -from pycurl import FORM_FILE class IRCInterface(Thread, Addon): __name__ = "IRCInterface" __version__ = "0.11" __description__ = """connect to irc and let owner perform different tasks""" __config__ = [("activated", "bool", "Activated", "False"), - ("host", "str", "IRC-Server Address", "Enter your server here!"), - ("port", "int", "IRC-Server Port", "6667"), - ("ident", "str", "Clients ident", "pyload-irc"), - ("realname", "str", "Realname", "pyload-irc"), - ("nick", "str", "Nickname the Client will take", "pyLoad-IRC"), - ("owner", "str", "Nickname the Client will accept commands from", "Enter your nick here!"), - ("info_file", "bool", "Inform about every file finished", "False"), - ("info_pack", "bool", "Inform about every package finished", "True"), - ("captcha", "bool", "Send captcha requests", "True")] + ("host", "str", "IRC-Server Address", "Enter your server here!"), + ("port", "int", "IRC-Server Port", "6667"), + ("ident", "str", "Clients ident", "pyload-irc"), + ("realname", "str", "Realname", "pyload-irc"), + ("nick", "str", "Nickname the Client will take", "pyLoad-IRC"), + ("owner", "str", "Nickname the Client will accept commands from", "Enter your nick here!"), + ("info_file", "bool", "Inform about every file finished", "False"), + ("info_pack", "bool", "Inform about every package finished", "True"), + ("captcha", "bool", "Send captcha requests", "True")] __author_name__ = ("Jeix") __author_mail__ = ("Jeix@hasnomail.com") - + def __init__(self, core, manager): Thread.__init__(self) Addon.__init__(self, core, manager) self.setDaemon(True) # self.sm = core.server_methods - self.api = core.api #todo, only use api - + self.api = core.api # todo, only use api + def coreReady(self): self.new_package = {} - + self.abort = False - + self.links_added = 0 self.more = [] self.start() - - + def packageFinished(self, pypack): try: if self.getConfig("info_pack"): self.response(_("Package finished: %s") % pypack.name) except: pass - + def downloadFinished(self, pyfile): try: if self.getConfig("info_file"): - self.response(_("Download finished: %(name)s @ %(plugin)s ") % { "name" : pyfile.name, "plugin": pyfile.pluginname} ) + self.response( + _("Download finished: %(name)s @ %(plugin)s ") % {"name": pyfile.name, "plugin": pyfile.pluginname}) except: pass @@ -88,7 +88,8 @@ class IRCInterface(Thread, Addon): task.handler.append(self) task.setWaiting(60) - page = getURL("http://www.freeimagehosting.net/upload.php", post={"attached" : (FORM_FILE, task.captchaFile)}, multipart=True) + page = getURL("http://www.freeimagehosting.net/upload.php", + post={"attached": (FORM_FILE, task.captchaFile)}, multipart=True) url = re.search(r"\[img\]([^\[]+)\[/img\]\[/url\]", page).group(1) self.response(_("New Captcha Request: %s") % url) @@ -107,15 +108,14 @@ class IRCInterface(Thread, Addon): self.sock.send("JOIN %s\r\n" % t.strip()) self.logInfo("pyLoad IRC: Connected to %s!" % host) self.logInfo("pyLoad IRC: Switching to listening mode!") - try: + try: self.main_loop() - + except IRCError, ex: self.sock.send("QUIT :byebye\r\n") print_exc() self.sock.close() - def main_loop(self): readbuffer = "" while True: @@ -123,48 +123,47 @@ class IRCInterface(Thread, Addon): fdset = select([self.sock], [], [], 0) if self.sock not in fdset[0]: continue - + if self.abort: raise IRCError("quit") - + readbuffer += self.sock.recv(1024) temp = readbuffer.split("\n") readbuffer = temp.pop() for line in temp: - line = line.rstrip() + line = line.rstrip() first = line.split() if first[0] == "PING": self.sock.send("PONG %s\r\n" % first[1]) - + if first[0] == "ERROR": raise IRCError(line) - + msg = line.split(None, 3) if len(msg) < 4: continue - + msg = { - "origin":msg[0][1:], - "action":msg[1], - "target":msg[2], - "text":msg[3][1:] + "origin": msg[0][1:], + "action": msg[1], + "target": msg[2], + "text": msg[3][1:] } - + self.handle_events(msg) - - + def handle_events(self, msg): if not msg["origin"].split("!", 1)[0] in self.getConfig("owner").split(): return - + if msg["target"].split("!", 1)[0] != self.getConfig("nick"): return - + if msg["action"] != "PRIVMSG": return - + # HANDLE CTCP ANTI FLOOD/BOT PROTECTION if msg["text"] == "\x01VERSION\x01": self.logDebug("Sending CTCP VERSION.") @@ -175,9 +174,9 @@ class IRCInterface(Thread, Addon): self.sock.send("NOTICE %s :%d\r\n" % (msg['origin'], time.time())) return elif msg["text"] == "\x01LAG\x01": - self.logDebug("Received CTCP LAG.") # don't know how to answer + self.logDebug("Received CTCP LAG.") # don't know how to answer return - + trigger = "pass" args = None @@ -195,26 +194,25 @@ class IRCInterface(Thread, Addon): for line in res: self.response(line, msg["origin"]) except Exception, e: - self.logError("pyLoad IRC: "+ repr(e)) - - + self.logError("pyLoad IRC: " + repr(e)) + def response(self, msg, origin=""): if origin == "": for t in self.getConfig("owner").split(): self.sock.send("PRIVMSG %s :%s\r\n" % (t.strip(), msg)) else: self.sock.send("PRIVMSG %s :%s\r\n" % (origin.split("!", 1)[0], msg)) - - -#### Events + + #### Events + def event_pass(self, args): return [] - + def event_status(self, args): downloads = self.api.statusDownloads() if not downloads: return ["INFO: There are no active downloads currently."] - + temp_progress = "" lines = ["ID - Name - Status - Speed - ETA - Progress"] for data in downloads: @@ -225,71 +223,69 @@ class IRCInterface(Thread, Addon): temp_progress = "%d%% (%s)" % (data.percent, data.format_size) lines.append("#%d - %s - %s - %s - %s - %s" % - ( - data.fid, - data.name, - data.statusmsg, - "%s/s" % formatSize(data.speed), - "%s" % data.format_eta, - temp_progress - ) - ) + ( + data.fid, + data.name, + data.statusmsg, + "%s/s" % formatSize(data.speed), + "%s" % data.format_eta, + temp_progress + )) return lines - + def event_queue(self, args): ps = self.api.getQueueData() - + if not ps: return ["INFO: There are no packages in queue."] - + lines = [] for pack in ps: - lines.append('PACKAGE #%s: "%s" with %d links.' % (pack.pid, pack.name, len(pack.links) )) - + lines.append('PACKAGE #%s: "%s" with %d links.' % (pack.pid, pack.name, len(pack.links))) + return lines - + def event_collector(self, args): ps = self.api.getCollectorData() if not ps: return ["INFO: No packages in collector!"] - + lines = [] for pack in ps: - lines.append('PACKAGE #%s: "%s" with %d links.' % (pack.pid, pack.name, len(pack.links) )) - + lines.append('PACKAGE #%s: "%s" with %d links.' % (pack.pid, pack.name, len(pack.links))) + return lines - + def event_info(self, args): if not args: return ['ERROR: Use info like this: info <id>'] - + info = None try: info = self.api.getFileData(int(args[0])) - + except FileDoesNotExists: return ["ERROR: Link doesn't exists."] - return ['LINK #%s: %s (%s) [%s][%s]' % (info.fid, info.name, info.format_size, info.statusmsg, - info.plugin)] - + return ['LINK #%s: %s (%s) [%s][%s]' % (info.fid, info.name, info.format_size, info.statusmsg, info.plugin)] + def event_packinfo(self, args): if not args: return ['ERROR: Use packinfo like this: packinfo <id>'] - + lines = [] pack = None try: pack = self.api.getPackageData(int(args[0])) - + except PackageDoesNotExists: return ["ERROR: Package doesn't exists."] - + id = args[0] self.more = [] - - lines.append('PACKAGE #%s: "%s" with %d links' % (id, pack.name, len(pack.links)) ) + + lines.append('PACKAGE #%s: "%s" with %d links' % (id, pack.name, len(pack.links))) for pyfile in pack.links: self.more.append('LINK #%s: %s (%s) [%s][%s]' % (pyfile.fid, pyfile.name, pyfile.format_size, pyfile.statusmsg, pyfile.plugin)) @@ -301,74 +297,69 @@ class IRCInterface(Thread, Addon): lines.extend(self.more[:6]) self.more = self.more[6:] lines.append("%d more links do display." % len(self.more)) - - + return lines - + def event_more(self, args): if not self.more: return ["No more information to display."] - + lines = self.more[:6] self.more = self.more[6:] lines.append("%d more links do display." % len(self.more)) - + return lines - + def event_start(self, args): - + self.api.unpauseServer() return ["INFO: Starting downloads."] - + def event_stop(self, args): - + self.api.pauseServer() return ["INFO: No new downloads will be started."] - - + def event_add(self, args): if len(args) < 2: return ['ERROR: Add links like this: "add <packagename|id> links". ', - 'This will add the link <link> to to the package <package> / the package with id <id>!'] - + 'This will add the link <link> to to the package <package> / the package with id <id>!'] - pack = args[0].strip() links = [x.strip() for x in args[1:]] - + count_added = 0 count_failed = 0 try: - id = int(pack) + id = int(pack) pack = self.api.getPackageData(id) if not pack: return ["ERROR: Package doesn't exists."] - + #TODO add links - + return ["INFO: Added %d links to Package %s [#%d]" % (len(links), pack["name"], id)] - + except: # create new package id = self.api.addPackage(pack, links, 1) return ["INFO: Created new Package %s [#%d] with %d links." % (pack, id, len(links))] - - + def event_del(self, args): if len(args) < 2: return ["ERROR: Use del command like this: del -p|-l <id> [...] (-p indicates that the ids are from packages, -l indicates that the ids are from links)"] - + if args[0] == "-p": ret = self.api.deletePackages(map(int, args[1:])) return ["INFO: Deleted %d packages!" % len(args[1:])] - + elif args[0] == "-l": ret = self.api.delLinks(map(int, args[1:])) return ["INFO: Deleted %d links!" % len(args[1:])] else: return ["ERROR: Use del command like this: del <-p|-l> <id> [...] (-p indicates that the ids are from packages, -l indicates that the ids are from links)"] - + def event_push(self, args): if not args: return ["ERROR: Push package to queue like this: push <package id>"] @@ -401,11 +392,10 @@ class IRCInterface(Thread, Addon): task = self.core.captchaManager.getTaskByID(args[0]) if not task: return ["ERROR: Captcha Task with ID %s does not exists." % args[0]] - + task.setResult(" ".join(args[1:])) return ["INFO: Result %s saved." % " ".join(args[1:])] - def event_help(self, args): lines = ["The following commands are available:", "add <package|packid> <links> [...] Adds link to package. (creates new package if it does not exist)", @@ -422,10 +412,11 @@ class IRCInterface(Thread, Addon): "status Show general download status", "help Shows this help message"] return lines - - + + class IRCError(Exception): def __init__(self, value): self.value = value + def __str__(self): return repr(self.value) |