diff --git a/bookwyrm/management/commands/erase_streams.py b/bookwyrm/management/commands/erase_streams.py new file mode 100644 index 000000000..042e857fc --- /dev/null +++ b/bookwyrm/management/commands/erase_streams.py @@ -0,0 +1,24 @@ +""" Delete user streams """ +from django.core.management.base import BaseCommand +import redis + +from bookwyrm import settings + +r = redis.Redis( + host=settings.REDIS_ACTIVITY_HOST, port=settings.REDIS_ACTIVITY_PORT, db=0 +) + + +def erase_streams(): + """ throw the whole redis away """ + r.flushall() + + +class Command(BaseCommand): + """ delete activity streams for all users """ + + help = "Delete all the user streams" + # pylint: disable=no-self-use,unused-argument + def handle(self, *args, **options): + """ flush all, baby """ + erase_streams() diff --git a/bookwyrm/management/commands/rebuild_feeds.py b/bookwyrm/management/commands/populate_streams.py similarity index 64% rename from bookwyrm/management/commands/rebuild_feeds.py rename to bookwyrm/management/commands/populate_streams.py index 70d4dd981..06ca5f075 100644 --- a/bookwyrm/management/commands/rebuild_feeds.py +++ b/bookwyrm/management/commands/populate_streams.py @@ -1,4 +1,4 @@ -""" Delete and re-create user feeds """ +""" Re-create user streams """ from django.core.management.base import BaseCommand import redis @@ -9,13 +9,8 @@ r = redis.Redis( ) -def erase_feeds(): - """ throw the whole redis away """ - r.flushall() - - -def create_feeds(): - """ build all the fields for all the users """ +def populate_streams(): + """ build all the streams for all the users """ users = models.User.objects.filter( local=True, is_active=True, @@ -26,11 +21,10 @@ def create_feeds(): class Command(BaseCommand): - """ start all over with user feeds """ + """ start all over with user streams """ - help = "Delete and re-create all the user feeds" + help = "Populate streams for all users" # pylint: disable=no-self-use,unused-argument def handle(self, *args, **options): """ run feed builder """ - erase_feeds() - create_feeds() + populate_streams() diff --git a/bw-dev b/bw-dev index 712b80287..b6406f2cb 100755 --- a/bw-dev +++ b/bw-dev @@ -109,7 +109,10 @@ case "$CMD" in black) makeitblack ;; + populate_feeds) + execweb python manage.py populate_streams + ;; *) - echo "Unrecognised command. Try: build, clean, up, initdb, resetdb, makemigrations, migrate, bash, shell, dbshell, restart_celery, test, pytest, test_report, black" + echo "Unrecognised command. Try: build, clean, up, initdb, resetdb, makemigrations, migrate, bash, shell, dbshell, restart_celery, test, pytest, test_report, black, populate_feeds" ;; esac