def handle(self, *args, **options):
if options["flush"]:
call_command("flush", "--noinput")
fake = Faker("hu_HU")
users = []
if not User.objects.filter(username="******").exists():
admin = User()
admin.username = "******"
admin.is_superuser = True
admin.is_staff = True
admin.first_name = "Admin"
admin.last_name = "Adminsson"
admin.set_password("admin")
admin.save()
Profile.objects.create(user=User.objects.get(username="******"))
users.append("admin")
if not User.objects.filter(username="******").exists():
User.objects.create(username="******",
is_staff=True,
first_name="Mod",
last_name="Mod")
Profile.objects.create(user=User.objects.get(username="******"))
users.append("mod")
if not User.objects.filter(username="******").exists():
User.objects.create(username="******",
first_name="User",
last_name="User")
Profile.objects.create(user=User.objects.get(username="******"))
users.append("user")
# Accounts
for i in range(random.choice(range(3, 9))):
# Users
fakedata = fake.simple_profile()
u = User()
u.username = fakedata["username"]
name = fakedata["name"].split(" ")
u.first_name = name[0]
u.last_name = name[1]
u.save()
# Profiles
Profile.objects.create(user=u)
users.append(u.username)
# Tags
tag_choices = []
for t in range(random.choice(range(5, 11))):
tag_choices.append(fake.word())
# Questions
for i in range(random.choice(range(3, 11))):
ov_q = self.generate_owner_votes(users)
q = Question()
q.title = fake.text(max_nb_chars=50)
q.text = fake.paragraph(nb_sentences=4)
q.owner = Profile.objects.get(user__username=ov_q["owner"])
q.show_username = bool(random.getrandbits(1))
q.votes = ov_q["votes"]
q.created_at = datetime.datetime.now()
q.save()
for t in random.sample(tag_choices,
random.randint(0, len(tag_choices))):
q.tags.add(t)
# Comments
for j in range(random.choice(range(0, 4))):
ov_cq = self.generate_owner_votes(users)
Comment.objects.create(
parent_question=q,
text=fake.paragraph(nb_sentences=4),
owner=Profile.objects.get(user__username=ov_cq["owner"]),
show_username=bool(random.getrandbits(1)),
votes=ov_cq["votes"],
)
# Answers
accepted_answer = -1
answer_count = random.choice(range(0, 6))
has_accepted = bool(random.getrandbits(1))
if has_accepted and answer_count > 0:
accepted_answer = random.choice(range(answer_count))
for k in range(answer_count):
ov_a = self.generate_owner_votes(users)
a = Answer()
a.text = fake.paragraph(nb_sentences=4)
a.owner = Profile.objects.get(user__username=ov_a["owner"])
a.show_username = bool(random.getrandbits(1))
if k == accepted_answer:
a.is_accepted = True
else:
a.is_accepted = False
a.votes = ov_a["votes"]
a.parent = q
a.save()
# Comments to answers
for l in range(random.choice(range(0, 4))):
ov_ca = self.generate_owner_votes(users)
Comment.objects.create(
parent_answer=a,
text=fake.paragraph(nb_sentences=4),
owner=Profile.objects.get(
user__username=ov_ca["owner"]),
show_username=bool(random.getrandbits(1)),
votes=ov_ca["votes"],
)
