Fix FeedTask collecting reddit rules

This commit is contained in:
Sonny Bakker 2020-08-12 19:38:16 +02:00
parent e58c5a4559
commit 03ac016dd3

View file

@ -8,6 +8,7 @@ from celery.utils.log import get_task_logger
from newsreader.accounts.models import User from newsreader.accounts.models import User
from newsreader.celery import app from newsreader.celery import app
from newsreader.news.collection.choices import RuleTypeChoices
from newsreader.news.collection.feed import FeedCollector from newsreader.news.collection.feed import FeedCollector
from newsreader.news.collection.utils import post from newsreader.news.collection.utils import post
from newsreader.utils.celery import MemCacheLock from newsreader.utils.celery import MemCacheLock
@ -33,7 +34,7 @@ class FeedTask(app.Task):
if acquired: if acquired:
logger.info(f"Running task for user {user_pk}") logger.info(f"Running task for user {user_pk}")
rules = user.rules.enabled() rules = user.rules.enabled().filter(type=RuleTypeChoices.feed)
collector = FeedCollector() collector = FeedCollector()
collector.collect(rules=rules) collector.collect(rules=rules)