Mercurial > cobalt-mk2 / changeset

--- a/cobalt.py	Tue May 13 23:29:37 2014 +0300
+++ b/cobalt.py	Fri May 23 19:25:37 2014 +0300
@@ -34,6 +34,8 @@
 import traceback
 import re
 import json
+import urllib
+import urllib2
 from suds.xsd.doctor import Import
 from suds.xsd.doctor import ImportDoctor
 from suds.client import Client
@@ -48,6 +50,8 @@
 g_admins = g_config['admins']
 g_mynick = g_config['nickname']
 
+g_idgamesSearchURL = 'http://www.doomworld.com/idgames/api/api.php?action=search&query=%s&type=title&sort=date&out=json'
+
 #
 # SOAP stuff
 #
@@ -164,7 +168,8 @@
 
 	def handle_read (self):
 		lines = self.recv (4096).splitlines()
-		for line in lines:
+		for utfline in lines:
+			line = utfline.decode("utf-8").encode("ascii","ignore")
 			print "[%s] -> %s" % (self.name, line)
 
 			if line.startswith ("PING :"):
@@ -241,10 +246,45 @@
 			if len(args) < 2:
 				raise logical_exception ("usage: .%s <target> <message...>" % command)
 			self.privmsg (args[0], " ".join (args[1:]))
-		elif command == "ticket":
+		elif command == 'ticket':
 			if len(args) != 1:
 				raise logical_exception ("usage: .%s <ticket>" % command)
 			self.get_ticket_data (replyto, args[0], True)
+		elif command == 'idgames':
+			try:
+				if len(args) < 1:
+					raise logical_exception ('usage: .%s <keywords>' % command)
+
+				url = g_idgamesSearchURL % urllib.quote (" ".join (args[0:]))
+				response = urllib2.urlopen (url).read()
+				data = json.loads (response)
+
+				if 'content' in data and 'file' in data['content']:
+					if type (data['content']['file']) is list:
+						files = data['content']['file']
+					else:
+						files = [data['content']['file']]
+
+					i = 0
+					for filedata in files:
+						if i >= 5:
+							break
+
+						self.privmsg (replyto, '- %s: \'%s\' by \'%s\', rating: %s: %s' % \
+							(filedata['filename'], filedata['title'], filedata['author'], filedata['rating'], filedata['url']))
+
+						i += 1
+					self.privmsg (replyto, "(%d / %d results posted)" % (i, len(files)))
+				elif 'warning' in data and 'message' in data['warning']:
+					raise logical_exception (data['warning']['message'])
+				elif 'error' in data and 'message' in data['error']:
+					raise logical_exception (data['error']['message'])
+				else:
+					raise logical_exception ("Incomplete JSON response from doomworld.com/idgames")
+			except logical_exception as e:
+				raise e
+			except Exception as e:
+				raise logical_exception ('Search failed: %s' % `e`)
 		else:
 			raise logical_exception ("unknown command `.%s`" % command)