From 903aaaf4c4d7664246001e1102e3a70c048a6e0c Mon Sep 17 00:00:00 2001
From: Mouse Reeve <mousereeve@riseup.net>
Date: Tue, 16 Nov 2021 10:41:08 -0800
Subject: [PATCH] Adds management and bw-dev commands

---
 .../commands/populate_lists_streams.py        | 34 +++++++++++++++++++
 .../management/commands/populate_streams.py   |  6 ++--
 bw-dev                                        |  3 ++
 3 files changed, 41 insertions(+), 2 deletions(-)
 create mode 100644 bookwyrm/management/commands/populate_lists_streams.py

diff --git a/bookwyrm/management/commands/populate_lists_streams.py b/bookwyrm/management/commands/populate_lists_streams.py
new file mode 100644
index 000000000..1cf488dfc
--- /dev/null
+++ b/bookwyrm/management/commands/populate_lists_streams.py
@@ -0,0 +1,34 @@
+""" Re-create user streams """
+from django.core.management.base import BaseCommand
+from bookwyrm import lists_stream, models
+
+
+def populate_lists_streams():
+    """build all the lists streams for all the users"""
+    print("Populating lists streams")
+    users = models.User.objects.filter(
+        local=True,
+        is_active=True,
+    ).order_by("-last_active_date")
+    print("This may take a long time! Please be patient.")
+    for user in users:
+        print(".", end="")
+        lists_stream.populate_lists_stream_task.delay(user.id)
+
+
+class Command(BaseCommand):
+    """start all over with user streams"""
+
+    help = "Populate streams for all users"
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--stream",
+            default=None,
+            help="Specifies which time of stream to populate",
+        )
+
+    # pylint: disable=no-self-use,unused-argument
+    def handle(self, *args, **options):
+        """run feed builder"""
+        populate_lists_streams()
diff --git a/bookwyrm/management/commands/populate_streams.py b/bookwyrm/management/commands/populate_streams.py
index a04d7f5ad..6b5376500 100644
--- a/bookwyrm/management/commands/populate_streams.py
+++ b/bookwyrm/management/commands/populate_streams.py
@@ -1,18 +1,20 @@
 """ Re-create user streams """
 from django.core.management.base import BaseCommand
-from bookwyrm import activitystreams, models
+from bookwyrm import activitystreams, lists_stream, models
 
 
 def populate_streams(stream=None):
     """build all the streams for all the users"""
     streams = [stream] if stream else activitystreams.streams.keys()
-    print("Populations streams", streams)
+    print("Populating streams", streams)
     users = models.User.objects.filter(
         local=True,
         is_active=True,
     ).order_by("-last_active_date")
     print("This may take a long time! Please be patient.")
     for user in users:
+        print(".", end="")
+        lists_stream.populate_lists_stream_task.delay(user.id)
         for stream_key in streams:
             print(".", end="")
             activitystreams.populate_stream_task.delay(stream_key, user.id)
diff --git a/bw-dev b/bw-dev
index 35f3694ee..c5ae7c9e2 100755
--- a/bw-dev
+++ b/bw-dev
@@ -120,6 +120,9 @@ case "$CMD" in
     populate_streams)
         runweb python manage.py populate_streams $@
         ;;
+    populate_lists_streams)
+        runweb python manage.py populate_lists_streams $@
+        ;;
     populate_suggestions)
         runweb python manage.py populate_suggestions
         ;;