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

import datetime

from django.contrib.auth.models import User
from django.core.management.base import BaseCommand

from apps.reader.models import UserSubscription


class Command(BaseCommand):
    def add_arguments(self, parser):
        parser.add_argument("-d", "--days", dest="days", nargs=1, default=1, help="Days of unread")
        parser.add_argument("-u", "--username", dest="username", nargs=1, help="Specify user id or username")
        parser.add_argument("-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()