NewsBlur/apps/rss_feeds/management/commands/mark_read.py

from django.core.management.base import BaseCommand
from django.contrib.auth.models import User
from apps.reader.models import UserSubscription
from optparse import make_option
import datetime

class Command(BaseCommand):
    option_list = BaseCommand.option_list + (
        make_option("-d", "--days", dest="days", nargs=1, default=1, help="Days of unread"),
        make_option("-u", "--username", dest="username", nargs=1, help="Specify user id or username"),
        make_option("-U", "--userid", dest="userid", nargs=1, help="Specify user id or username"),
    )

    def handle(self, *args, **options):
        if options['userid']:
            user = User.objects.filter(pk=options['userid'])[0]
        elif options['username']:
            user = User.objects.get(username__icontains=options['username'])
        else:
            raise Exception, "Need username or user id."
        
        user.profile.last_seen_on = datetime.datetime.utcnow()
        user.profile.save()
        feeds = UserSubscription.objects.filter(user=user)
        for sub in feeds:
            if options['days'] == 0:
                sub.mark_feed_read()
            else:
                sub.mark_read_date = datetime.datetime.utcnow() - datetime.timedelta(days=int(options['days']))
                sub.needs_unread_recalc = True
                sub.save()