Add support for JSON APIs in feedify

feedify.py

@@ -3,10 +3,12 @@
 from ConfigParser import ConfigParser
 from fnmatch import fnmatch
 import feeds
+import morss
 import re
 
 import urllib2
 import lxml.html
+import json
 import urlparse
 
 def toclass(query):
@@ -29,49 +31,139 @@ def getRule(link):
 def supported(link):
 	return getRule(link) is not False
 
-def getString(html, expr):
-	matches = html.xpath(toclass(expr))
-	if len(matches):
-		out = ''
-		for match in matches:
-			if isinstance(match, basestring):
-				out += match
-			elif isinstance(match, lxml.html.HtmlElement):
-				out += lxml.html.tostring(match)
-		return out
-	else:
-		return ''
+def formatString(string, getter, error=False):
+	out = ""
+	char = string[0]
 
-def build(link, data=None):
-	rule = getRule(link)
-	if rule is False:
-		return False
+	follow = string[1:]
+
+	if char == '"':
+		match = follow.partition('"')
+		out = match[0]
+		if len(match) >= 2:
+			next = match[2]
+		else:
+			next = None
+	elif char == '{':
+		match = follow.partition('}')
+		try:
+			test = formatString(match[0], getter, True)
+		except ValueError, KeyError:
+			pass
+		else:
+			out = test
+
+		next = match[2]
+	elif char == ' ':
+		next = follow
+	elif re.search(r'^([^{}<>" ]+)(?:<"([^>]+)">)?(.*)$', string):
+		match = re.search(r'^([^{}<>" ]+)(?:<"([^>]+)">)?(.*)$', string).groups()
+		rawValue = getter(match[0])
+		print repr(rawValue)
+		if not isinstance(rawValue, basestring):
+			if match[1] is not None:
+				out = match[1].join(rawValue)
+			else:
+				out = ''.join(rawValue)
+		if not out and error:
+			raise ValueError
+		next = match[2]
+	else:
+		raise ValueError('bogus string')
+
+	if next is not None and len(next):
+		return out + formatString(next, getter, error)
+	else:
+		return out
+
+class Builder(object):
+	def __init__(self, link, data=None):
+		self.link = link
 
 		if data is None:
 			data = urllib2.urlopen(link).read()
+		self.data = data
 
-	html = lxml.html.fromstring(data)
-	feed = feeds.FeedParserAtom()
+		self.rule = getRule(link)
 
-	if 'title' in rule:
-		feed.title = getString(html, rule['title'])
+		if self.rule['mode'] == 'xpath':
+			self.data = morss.decodeHTML(self.data)
+			self.doc = lxml.html.fromstring(self.data)
+		elif self.rule['mode'] == 'json':
+			self.doc = json.loads(data)
 
-	if 'items' in rule:
-		for item in html.xpath(toclass(rule['items'])):
+		self.feed = feeds.FeedParserAtom()
+
+	def raw(self, html, expr):
+		if self.rule['mode'] == 'xpath':
+			print 1, toclass(expr)
+			return html.xpath(toclass(expr))
+
+		elif self.rule['mode'] == 'json':
+			a = [html]
+			b = []
+			for x in expr.strip(".").split("."):
+				match = re.search(r'^([^\[]+)(?:\[([0-9]+)\])?$', x).groups()
+				for elem in a:
+					if isinstance(elem, dict):
+						kids = elem.get(match[0])
+						if kids is None:
+							pass
+						elif isinstance(kids, list):
+							[b.append(i) for i in kids]
+						elif isinstance(kids, basestring):
+							b.append(kids.replace('\n', '<br/>'))
+						else:
+							b.append(kids)
+
+				if match[1] is None:
+					a = b
+				else:
+					if len(b)-1 >= int(match[1]):
+						a = [b[int(match[1])]]
+					else:
+						a = []
+				b = []
+			return a
+
+	def strings(self, html, expr):
+		if self.rule['mode'] == 'xpath':
+			out = []
+			for match in self.raw(html, expr):
+				if isinstance(match, basestring):
+					out.append(match)
+				elif isinstance(match, lxml.html.HtmlElement):
+					out.append(lxml.html.tostring(match))
+			return out
+
+		elif self.rule['mode'] == 'json':
+			return self.raw(html, expr)
+
+	def string(self, html, expr):
+		getter = lambda x: self.strings(html, x)
+		return formatString(self.rule[expr], getter)
+
+	def build(self):
+		if 'title' in self.rule:
+			self.feed.title = self.string(self.doc, 'title')
+
+		if 'items' in self.rule:
+			matches = self.raw(self.doc, self.rule['items'])
+			if matches and len(matches):
+				for item in matches:
 					feedItem = {}
 
-			if 'item_title' in rule:
-				feedItem['title'] = getString(item, rule['item_title'])
-			if 'item_link' in rule:
-				url = getString(item, rule['item_link'])
-				url = urlparse.urljoin(link, url)
+					if 'item_title' in self.rule:
+						feedItem['title'] = self.string(item, 'item_title')
+					if 'item_link' in self.rule:
+						url = self.string(item, 'item_link')
+						url = urlparse.urljoin(self.link, url)
 						feedItem['link'] = url
-			if 'item_desc' in rule:
-				feedItem['desc'] = getString(item, rule['item_desc'])
-			if 'item_content' in rule:
-				feedItem['content'] = getString(item, rule['item_content'])
-			if 'item_time' in rule:
-				feedItem['updated'] = getString(item, rule['item_time'])
+					if 'item_desc' in self.rule:
+						feedItem['desc'] = self.string(item, 'item_desc')
+					if 'item_content' in self.rule:
+						feedItem['content'] = self.string(item, 'item_content')
+					if 'item_time' in self.rule:
+						feedItem['updated'] = self.string(item, 'item_time')
 
-			feed.items.append(feedItem)
-	return feed
+					self.feed.items.append(feedItem)

morss.py

@@ -410,8 +410,9 @@ def Gather(url, cachePath, options):
 	if style == 'normal':
 		rss = feeds.parse(xml)
 	elif style == 'feedify':
-		xml = decodeHTML(xml)
-		rss = feedify.build(url, xml)
+		feed = feedify.Builder(url, xml)
+		feed.build()
+		rss = feed.feed
 	elif style == 'html':
 		match = lxml.html.fromstring(xml).xpath("//link[@rel='alternate'][@type='application/rss+xml' or @type='application/atom+xml']/@href")
 		if len(match):