Management command for creating fake news items

Author: joemull

src/utils/management/commands/create_fake_news_items.py

@@ -0,0 +1,69 @@
+from django.core.management.base import BaseCommand
+from django.contrib.contenttypes.models import ContentType
+from faker import Faker
+from faker.providers import DynamicProvider
+import secrets
+
+from utils.logic import get_aware_datetime
+from utils.logger import get_logger
+from press import models as press_models
+from journal import models as journal_models
+from core import models as core_models
+from utils.testing.helpers import create_news_item
+
+fake = Faker()
+logger = get_logger(__name__)
+
+
+class Command(BaseCommand):
+    help = "Generates fake news items at the press and journal levels"
+
+    press = press_models.Press.objects.first()
+    press_content_type = ContentType.objects.get_for_model(press_models.Press)
+    journals = journal_models.Journal.objects.all()
+    journal_content_type = ContentType.objects.get_for_model(journal_models.Journal)
+
+    @classmethod
+    def build_fake_news_item(cls, content_type, obj_pk, posted_by):
+        title = fake.sentence()
+        body = ""
+        for num in range(0, 10):
+            body += f"<p>{fake.text()}</p>"
+        date = get_aware_datetime(fake.date())
+        tags = [fake.word() for i in range(0, 3)]
+        create_news_item(
+            content_type,
+            obj_pk,
+            title=title,
+            body=body,
+            posted=date,
+            posted_by=posted_by,
+            start_display=date,
+            tags=tags,
+        )
+
+    def handle(self, *args, **options):
+        confirmation = input(
+            "News items will be public if this installation is live. Create fake items? (y/n)"
+        )
+        if confirmation.lower() != "y":
+            return
+
+        for each in range(0, 30):
+            posted_by = core_models.Account.objects.filter(is_staff=True).first()
+            if not posted_by:
+                logger.warn("Staff member required")
+                return
+            self.build_fake_news_item(self.press_content_type, self.press.pk, posted_by)
+        for journal in self.journals:
+            for each in range(0, 30):
+                posted_by = core_models.Account.objects.filter(
+                    accountrole__role__slug="editor", accountrole__journal=journal
+                ).first()
+                if not posted_by:
+                    logger.warn(f"Editor required for journal {journal}")
+                    return
+                self.build_fake_news_item(
+                    self.journal_content_type, journal.pk, posted_by
+                )
+        logger.debug(self.style.SUCCESS("Successfully created news items"))