Early fake threads generator utility

misago/faker/management/commands/createfakethreads.py

@@ -0,0 +1,112 @@
+import random
+
+from faker import Factory
+
+from django.contrib.auth import get_user_model
+from django.core.management.base import BaseCommand
+from django.db.transaction import atomic
+from django.template.defaultfilters import linebreaks_filter
+from django.utils import timezone
+
+from misago.core.management.progressbar import show_progress
+from misago.forums.models import Forum
+from misago.threads.checksums import update_post_checksum
+from misago.threads.models import Thread, Post
+
+
+class Command(BaseCommand):
+    help = 'Adds random threads and posts for testing purposes'
+
+    def handle(self, *args, **options):
+        try:
+            fake_threads_to_create = int(args[0])
+        except IndexError:
+            fake_threads_to_create = 5
+        except ValueError:
+            self.stderr.write("\nOptional argument should be integer.")
+            sys.exit(1)
+
+        forums = [f for f in Forum.objects.all_forums().filter(role='forum')]
+
+        fake = Factory.create()
+
+        User = get_user_model()
+        total_users = User.objects.count()
+
+        self.stdout.write('Creating fake threads...\n')
+
+        message = '\nSuccessfully created %s fake threads'
+
+        created_threads = 0
+        show_progress(self, created_threads, fake_threads_to_create)
+        for i in xrange(fake_threads_to_create):
+            with atomic():
+                time = timezone.now()
+                forum = random.choice(forums)
+                user = User.objects.order_by('?')[:1][0]
+
+                thread_is_moderated = random.randint(0, 100) > 90
+                thread_is_hidden = random.randint(0, 100) > 90
+                thread_is_closed = random.randint(0, 100) > 90
+
+                thread = Thread(
+                    forum=forum,
+                    weight=0,
+                    started_on=time,
+                    starter_name='-',
+                    starter_slug='-',
+                    last_post_on=time,
+                    last_poster_name='-',
+                    last_poster_slug='-',
+                    replies=random.randint(0, 2000),
+                    is_moderated=thread_is_moderated,
+                    is_hidden=thread_is_hidden,
+                    is_closed=thread_is_closed)
+                thread.set_title(fake.sentence())
+                thread.save()
+
+                fake_message = "\n\n".join(fake.paragraphs())
+                post = Post.objects.create(
+                    forum=forum,
+                    thread=thread,
+                    poster=user,
+                    poster_name=user.username,
+                    poster_ip=fake.ipv4(),
+                    original=fake_message,
+                    parsed=linebreaks_filter(fake_message),
+                    posted_on=time,
+                    updated_on=time)
+                update_post_checksum(post)
+                post.save(update_fields=['checksum'])
+
+                thread.set_first_post(post)
+                thread.set_last_post(post)
+                thread.save()
+
+                forum.threads += 1
+                forum.posts += 1
+                forum.set_last_thread(thread)
+                forum.save()
+
+                user.threads += 1
+                user.posts += 1
+                user.save()
+
+                created_threads += 1
+                show_progress(self, created_threads, fake_threads_to_create)
+
+        pinned_threads = random.randint(0, int(created_threads * 0.05)) or 1
+        self.stdout.write('\Pinning %s threads...' % pinned_threads)
+        for i in xrange(0, pinned_threads):
+            thread = Thread.objects.order_by('?')[:1][0]
+            thread.weight = 1
+            thread.save()
+
+        announcements = random.randint(0, int(created_threads * 0.01)) or 1
+        self.stdout.write('\Making %s announcements...' % announcements)
+        for i in xrange(0, announcements):
+            thread = Thread.objects.order_by('?')[:1][0]
+            thread.weight = 2
+            thread.save()
+
+        self.stdout.write(message % created_threads)