NewsBlur/apps/analyzer/models.py

from django.db import models
from django.contrib.auth.models import User
import datetime
from apps.rss_feeds.models import Feed, Story, StoryAuthor, Tag

class FeatureCategory(models.Model):

    user = models.ForeignKey(User)
    feed = models.ForeignKey(Feed)
    feature = models.CharField(max_length=255)
    category = models.CharField(max_length=255)
    count = models.IntegerField(default=0)
    
    def __unicode__(self):
        return '%s - %s (%s)' % (self.feature, self.category, self.count)

class Category(models.Model):
    
    user = models.ForeignKey(User)
    feed = models.ForeignKey(Feed)
    category = models.CharField(max_length=255)
    count = models.IntegerField(default=0)
    
    def __unicode__(self):
        return '%s (%s)' % (self.category, self.count)
        
        
class ClassifierTitle(models.Model):
    user = models.ForeignKey(User)
    score = models.SmallIntegerField()
    title = models.CharField(max_length=255)
    feed = models.ForeignKey(Feed)
    original_story = models.ForeignKey(Story, null=True)
    creation_date = models.DateTimeField(auto_now=True)
    
    def __unicode__(self):
        return '%s: %s (%s)' % (self.user, self.title, self.feed)
        
        
class ClassifierAuthor(models.Model):
    user = models.ForeignKey(User)
    score = models.SmallIntegerField()
    author = models.ForeignKey(StoryAuthor)
    feed = models.ForeignKey(Feed)
    original_story = models.ForeignKey(Story, null=True)
    creation_date = models.DateTimeField(auto_now=True)
    
    def __unicode__(self):
        return '%s: %s (%s)' % (self.user, self.author.author_name, self.feed)
        
    def apply_classifier(self, story):
        if story['author'] == self.author:
            return True
        return False


class ClassifierFeed(models.Model):
    user = models.ForeignKey(User)
    score = models.SmallIntegerField()
    feed = models.ForeignKey(Feed)
    original_story = models.ForeignKey(Story, null=True)
    creation_date = models.DateTimeField(auto_now=True)
    
    def __unicode__(self):
        return '%s: %s' % (self.user, self.feed)
        
    def apply_classifier(self, story):
        if self.feed == story.feed:
            return True
        return False

        
class ClassifierTag(models.Model):
    user = models.ForeignKey(User)
    score = models.SmallIntegerField()
    tag = models.ForeignKey(Tag)
    feed = models.ForeignKey(Feed)
    original_story = models.ForeignKey(Story, null=True)
    creation_date = models.DateTimeField(auto_now=True)
    
    def __unicode__(self):
        return '%s: %s (%s)' % (self.user, self.tag.name, self.feed)
        
def apply_classifier_titles(classifiers, story):
    for classifier in classifiers:
        if classifier.title in story['story_title']:
            # print 'Titles: (%s) %s -- %s' % (classifier.title in story['story_title'], classifier.title, story['story_title'])
            return classifier.score
    return 0
    
def apply_classifier_feeds(classifiers, feed):
    for classifier in classifiers:
        if classifier.feed == feed:
            # print 'Feeds: %s -- %s' % (classifier.feed, feed)
            return classifier.score
    return 0
    
def apply_classifier_authors(classifiers, story):
    for classifier in classifiers:
        if classifier.author.id == story['story_author_id']:
            # print 'Authors: %s -- %s' % (classifier.author.id, story['story_author_id'])
            return classifier.score
    return 0
    
def apply_classifier_tags(classifiers, story):
    for classifier in classifiers:
        if classifier.tag.name in story['story_tags']:
            # print 'Tags: (%s) %s -- %s' % (classifier.tag.name in story['story_tags'], classifier.tag.name, story['story_tags'])
            return classifier.score
    return 0