add: initial AI support

plugins/flibusta.py

@@ -0,0 +1,97 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# --------------------------------------------------------------------------- #
+#                                                                             #
+#    Plugin for iSida Jabber Bot                                              #
+#    Copyright (C) Vit@liy <vitaliy@root.ua>                                  #
+#                                                                             #
+#    This program is free software: you can redistribute it and/or modify     #
+#    it under the terms of the GNU 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 General Public License for more details.                             #
+#                                                                             #
+#    You should have received a copy of the GNU General Public License        #
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.    #
+#                                                                             #
+# --------------------------------------------------------------------------- #
+
+from xml.dom.minidom import parseString
+
+def flibusta(type, jid, nick, text):
+	text = re.sub(' +', ' ', text).strip()
+	msg = ''
+	if text == 'new':
+		c=urllib.urlopen('http://xpresslib.mobile-master.org/new').read()
+		xml=parseString(c)
+		nodes=xml.getElementsByTagName('book')
+		for el in nodes:
+			f=el.getAttribute('file')
+			size=el.getAttribute('size')
+			author=unicode(el.getAttribute('author'))
+			title=unicode(el.firstChild.data)
+			msg += u'\n%s - %s, %s кБ, http://flibusta.net/b/%s' % (author,title,round(float(size)/1024,1),f.split('.')[0]+'/fb2')
+		if not msg: msg = L('No news','%s/%s'%(jid,nick))
+	else:
+		opt = [i for i in (' ' + text).split(' -') if len(i)>2 and i[1]==' ']
+		opt = dict([i.split(' ', 1) for i in opt])
+		if opt.has_key('n'):
+			if re.match('\d+$', opt['n']): a = b = int(opt['n'])
+			elif re.match('\d+-\d+$', opt['n']): a, b = map(int,opt['n'].split('-'))
+			else: a, b = 1, 5
+		else: a, b = 1, 5
+		if opt.has_key('t'):
+			i={}
+			i['search']=opt['t'].encode('utf-8')
+			d=urllib.urlencode(i)
+			s=urllib.urlopen('http://xpresslib.mobile-master.org/search',d).read()
+			xml=parseString(s)
+			nodes=xml.getElementsByTagName('book')
+			t_book=[]
+			for el in nodes:
+				f0=el.getAttribute('file')
+				size=el.getAttribute('size')
+				author=unicode(el.getAttribute('author'))
+				title=unicode(el.firstChild.data)
+				t_book.append([author,title,size,f0])
+			if opt.has_key('a'): t_book = filter(lambda x: opt['a'].lower() in x[0].lower(), t_book)
+			if t_book:
+				c = len(t_book)
+				if b > c:
+					if b != a:  b = c
+					else: a = b = c
+				msg += L('Total found %s matches. Result(s) %s:\n','%s/%s'%(jid,nick)) % (c, a if a == b else '%s-%s' % (a,b))
+ 				for i in t_book[a-1:b]:
+					msg += u'%s- %s, %s кБ, http://flibusta.net/b/%s\n' % (i[0],i[1],round(float(i[2])/1024,1),i[3].split('.')[0]+'/fb2')
+		elif opt.has_key('a'):
+			i={}
+			i['search']=opt['a'].encode('utf-8')
+			d=urllib.urlencode(i)
+			s=urllib.urlopen('http://xpresslib.mobile-master.org/authors',d).read()
+			xml=parseString(s)
+			nodes=xml.getElementsByTagName('author')
+			a_book=[]
+			for el in nodes:
+				id=el.getAttribute('id')
+				title=unicode(el.firstChild.data)
+				a_book.append([id,title])
+			if a_book:
+				c = len(a_book)
+				if b > c:
+					if b != a:  b = c
+					else: a = b = c
+				msg += L('Total found %s matches. Result(s) %s:\n','%s/%s'%(jid,nick)) % (c, a if a == b else '%s-%s' % (a,b))
+				for i in a_book[a-1:b]:
+					msg += '%s- http://flibusta.net/a/%s\n' % (i[1], i[0])
+
+	if not msg: msg = L('What?','%s/%s'%(jid,nick))
+	send_msg(type, jid, nick, msg)
+
+global execute
+
+execute = [(3, 'flibusta', flibusta, 2, 'Search books and authors on flibusta.net.\nflibusta [-a author] [-t title] [-n number of result]')]