from django.db import models from django.contrib.auth.models import User import datetime import random from django.core.cache import cache from apps.rss_feeds.models import Feed, Story from utils import feedparser, object_manager, json from apps.analyzer.models import ClassifierFeed, ClassifierAuthor, ClassifierTag, ClassifierTitle from apps.analyzer.models import apply_classifier_titles, apply_classifier_feeds, apply_classifier_authors, apply_classifier_tags DAYS_OF_UNREAD = 14 class UserSubscription(models.Model): user = models.ForeignKey(User) feed = models.ForeignKey(Feed) last_read_date = models.DateTimeField(default=datetime.datetime.now() - datetime.timedelta(days=DAYS_OF_UNREAD)) mark_read_date = models.DateTimeField(default=datetime.datetime.now() - datetime.timedelta(days=DAYS_OF_UNREAD)) unread_count_neutral = models.IntegerField(default=0) unread_count_positive = models.IntegerField(default=0) unread_count_negative = models.IntegerField(default=0) unread_count_updated = models.DateTimeField(default=datetime.datetime(2000,1,1)) def __unicode__(self): return '[' + self.feed.feed_title + '] ' def save(self, force_insert=False, force_update=False): self.unread_count_updated = datetime.datetime.now() super(UserSubscription, self).save(force_insert, force_update) def get_user_feeds(self): return Feed.objects.get(user=self.user, feed=feeds) def count_unread(self): if self.unread_count_updated > self.feed.last_update: return self.unread_count count = (self.stories_newer_lastread() + self.stories_between_lastread_allread()) if count == 0: self.mark_read_date = datetime.datetime.now() self.last_read_date = datetime.datetime.now() self.unread_count_updated = datetime.datetime.now() self.unread_count = 0 self.save() else: self.unread_count = count self.unread_count_updated = datetime.datetime.now() self.save() return count def mark_read(self): self.last_read_date = datetime.datetime.now() self.unread_count -= 1 self.unread_count_updated = datetime.datetime.now() self.save() def mark_feed_read(self): self.last_read_date = datetime.datetime.now() self.mark_read_date = datetime.datetime.now() self.unread_count = 0 self.unread_count_updated = datetime.datetime.now() self.save() # readstories = ReadStories.objects.filter(user=self.user, feed=self.feed) # readstories.delete() def stories_newer_lastread(self): return self.feed.new_stories_since_date(self.last_read_date).count() def stories_between_lastread_allread(self): story_count = Story.objects.filter( story_date__gte=self.mark_read_date, story_date__lte=self.last_read_date, story_feed=self.feed ).count() read_count = UserStory.objects.filter( feed=self.feed, read_date__gte=self.mark_read_date, read_date__lte=self.last_read_date ).count() return story_count - read_count def subscribe_to_feed(self, feed_id): feed = Feed.objects.get(id=feed_id) new_subscription = UserSubscription(user=self.user, feed=feed) new_subscription.save() def calculate_feed_scores(self): feed_scores = dict(negative=0, neutral=0, positive=0) date_delta = datetime.datetime.now()-datetime.timedelta(days=DAYS_OF_UNREAD) if date_delta < self.mark_read_date: date_delta = self.mark_read_date read_stories = UserStory.objects.select_related('story')\ .filter(user=self.user, feed=self.feed, story__story_date__gte=date_delta) stories_db = Story.objects.filter(story_date__gte=date_delta, story_feed=self.feed)\ .exclude(id__in=[rs.story.id for rs in read_stories]) stories = self.feed.format_stories(stories_db) classifier_feeds = ClassifierFeed.objects.filter(user=self.user, feed=self.feed) classifier_authors = ClassifierAuthor.objects.filter(user=self.user, feed=self.feed) classifier_titles = ClassifierTitle.objects.filter(user=self.user, feed=self.feed) classifier_tags = ClassifierTag.objects.filter(user=self.user, feed=self.feed) scores = { 'feed': apply_classifier_feeds(classifier_feeds, self.feed), } for story in stories: scores.update({ 'author': apply_classifier_authors(classifier_authors, story), 'tags': apply_classifier_tags(classifier_tags, story), 'title': apply_classifier_titles(classifier_titles, story), }) max_score = max(scores['feed'], scores['author'], scores['tags'], scores['title']) min_score = min(scores['feed'], scores['author'], scores['tags'], scores['title']) if max_score > 0: feed_scores['positive'] += 1 if min_score < 0: feed_scores['negative'] += 1 if max_score == 0 and min_score == 0: feed_scores['neutral'] += 1 self.unread_count_positive = feed_scores['positive'] self.unread_count_neutral = feed_scores['neutral'] self.unread_count_negative = feed_scores['negative'] self.save() return def get_scores(self): scores = json.decode(self.scores) return scores class Meta: unique_together = ("user", "feed") class UserStory(models.Model): user = models.ForeignKey(User) feed = models.ForeignKey(Feed) story = models.ForeignKey(Story) read_date = models.DateTimeField(auto_now=True) opinion = models.IntegerField(default=0) def __unicode__(self): return ('[' + self.feed.feed_title + '] ' + self.story.story_title) class Meta: verbose_name_plural = "user stories" verbose_name = "user story" unique_together = ("user", "feed", "story") class UserSubscriptionFolders(models.Model): user = models.ForeignKey(User) user_sub = models.ForeignKey(UserSubscription) feed = models.ForeignKey(Feed) folder = models.CharField(max_length=255) def __unicode__(self): return ('[' + self.feed.feed_title + '] ' + self.folder) class Meta: verbose_name_plural = "folders" verbose_name = "folder" unique_together = ("user", "user_sub")