X-Git-Url: http://git.ithinksw.org/~jspiros/reader.git/blobdiff_plain/9b96433a50883314722dd38216dc38e8915f2a4f..HEAD:/utils.py

diff --git a/utils.py b/utils.py
index e58dbdb..b5f6850 100644
--- a/utils.py
+++ b/utils.py
@@ -1,9 +1,73 @@
-from .models import Feed, Entry
+from .models import Feed, Entry, Subscription, UserEntry
 import datetime
 import feedparser
 
 
-def refresh_feed(feed, save=True):
+def _rate_content(content):
+	if content.type == 'application/xhtml+xml':
+		return 0
+	elif content.type == 'text/html':
+		return 1
+	elif content.type == 'text/plain':
+		return 2
+	else:
+		return 3
+
+
+def _choose_content(contents):
+	limited_contents = [content for content in contents if content.type in ('application/xhtml+xml', 'text/html', 'text/plain')]
+	limited_contents.sort(key=_rate_content)
+	return limited_contents[0] if len(limited_contents) > 0 else None
+
+
+def _parse_date(date):
+	try:
+		return datetime.datetime(*(date[0:6]))
+	except:
+		return None
+
+
+def _add_entry(feed, parsed_entry):
+	title = parsed_entry.get('title', 'Untitled')
+	link = parsed_entry.get('link', feed.link)
+	published = _parse_date(parsed_entry.get('published_parsed', parsed_entry.get('created_parsed', parsed_entry.get('updated_parsed', None))))
+	if not published:
+		published = datetime.datetime.now()
+	updated = _parse_date(parsed_entry.get('updated_parsed', None))
+	if not updated:
+		updated = published
+	contents = parsed_entry.get('content', None)
+	if contents:
+		content = _choose_content(contents).value
+	else:
+		content = None
+	summary = parsed_entry.get('summary', None)
+	
+	if summary or content:
+		entry, created = feed.entries.get_or_create(uri=parsed_entry.id, defaults={
+			'title': title,
+			'link': link,
+			'published': published,
+			'updated': updated,
+			'summary': summary,
+			'content': content
+		})
+		if not created:
+			entry.title = title
+			entry.link = link
+			entry.published = published
+			entry.updated = updated
+			entry.summary = summary
+			entry.content = content
+			entry.save()
+
+
+def _add_entries(feed, parsed):
+	for parsed_entry in parsed.entries:
+		_add_entry(feed, parsed_entry)
+
+
+def refresh_feed(feed):
 	if feed.alive:
 		parsed = feedparser.parse(feed.url, etag=feed.etag, modified=(feed.modified.timetuple() if feed.modified else None))
 		if parsed.get('status', None) == 304:
@@ -17,12 +81,35 @@ def refresh_feed(feed, save=True):
 		if parsed.has_key('modified'):
 			feed.modified = datetime.datetime(*(parsed.modified[0:6]))
 		feed.title = parsed.feed.get('title', feed.url)
-		feed.updated = datetime.datetime(*(parsed.feed.get('updated_parsed', datetime.datetime.now().timetuple())[0:6]))
-		if save:
-			feed.save()
+		feed.updated = _parse_date(parsed.feed.get('updated_parsed', datetime.datetime.now().timetuple())[0:6])
+		feed.link = parsed.feed.get('link', feed.url)
+		
+		feed.save()
+		_add_entries(feed, parsed)
+
 
+def refresh_all_feeds():
+	for feed in Feed.objects.all():
+		refresh_feed(feed)
 
-def add_feed(url):
-	feed = Feed(url=url)
+
+def add_feed(url, user=None):
+	try:
+		feed = Feed.objects.get(url=url)
+	except Feed.DoesNotExist:
+		feed = Feed(url=url)
 	refresh_feed(feed)
-	return feed
\ No newline at end of file
+	if user is not None:
+		sub, _ = Subscription.objects.get_or_create(user=user, feed=feed)
+	return feed
+
+
+def unread_count(user, feed=None):
+	if feed:
+		entries = feed.entries
+	else:
+		entries = Entry.objects.filter(feed__subscriptions__user=user)
+	total_entries = entries.count()
+	read_entries = entries.filter(userentries__user=user, userentries__read=True).count()
+	return total_entries - read_entries
+	
\ No newline at end of file