mirror of
https://github.com/viq/NewsBlur.git
synced 2025-04-13 09:38:09 +00:00
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
|
from django.core.management.base import BaseCommand
|
||
|
|
from django.core.handlers.wsgi import WSGIHandler
|
||
|
|
from apps.rss_feeds.models import Feed, Story
|
||
|
|
from django.core.cache import cache
|
||
|
|
from apps.reader.models import UserSubscription, UserSubscriptionFolders, UserStory
|
||
|
|
from optparse import OptionParser, make_option
|
||
|
|
import os
|
||
|
|
import logging
|
||
|
|
import errno
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
option_list = BaseCommand.option_list + (
|
||
|
|
make_option("-f", "--feed", dest="feed", default=None),
|
||
|
|
make_option("-d", "--daemon", dest="daemonize", action="store_true"),
|
||
|
|
)
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
if options['daemonize']:
|
||
|
|
daemonize()
|
||
|
|
|
||
|
|
feeds = Feed.objects.all()
|
||
|
|
self._refresh_feeds(feeds)
|
||
|
|
|
||
|
|
def _refresh_feeds(self, feeds):
|
||
|
|
for feed in feeds:
|
||
|
|
feed.update(True)
|
||
|
|
usersubs = UserSubscription.objects.filter(
|
||
|
|
feed=feed.id
|
||
|
|
)
|
||
|
|
for us in usersubs:
|
||
|
|
us.count_unread()
|
||
|
|
cache.delete('usersub:%s' % us.user_id)
|
||
|
|
|
||
|
|
def daemonize():
|
||
|
|
"""
|
||
|
|
Detach from the terminal and continue as a daemon.
|
||
|
|
"""
|
||
|
|
# swiped from twisted/scripts/twistd.py
|
||
|
|
# See http://www.erlenstar.demon.co.uk/unix/faq_toc.html#TOC16
|
||
|
|
if os.fork(): # launch child and...
|
||
|
|
os._exit(0) # kill off parent
|
||
|
|
os.setsid()
|
||
|
|
if os.fork(): # launch child and...
|
||
|
|
os._exit(0) # kill off parent again.
|
||
|
|
os.umask(077)
|
||
|
|
null = os.open("/dev/null", os.O_RDWR)
|
||
|
|
for i in range(3):
|
||
|
|
try:
|
||
|
|
os.dup2(null, i)
|
||
|
|
except OSError, e:
|
||
|
|
if e.errno != errno.EBADF:
|
||
|
|
raise
|
||
|
|
os.close(null)
|