mirror of
https://github.com/Findus23/se-simulator.git
synced 2024-09-19 15:53:45 +02:00
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
|
from slugify import slugify
|
||
|
|
||
|
from models import *
|
||
|
from text_generator import get_chain, generate_text
|
||
|
|
||
|
|
||
|
def add_username(site, count=100):
|
||
|
"""
|
||
|
|
||
|
:type site: Site
|
||
|
"""
|
||
|
chain = get_chain(site.url, "Usernames")
|
||
|
for _ in range(count):
|
||
|
username = generate_text(chain, "Usernames")
|
||
|
User.create(username=username, site=site)
|
||
|
|
||
|
|
||
|
def add_title(site, count=100):
|
||
|
chain = get_chain(site.url, "Titles")
|
||
|
for _ in range(count):
|
||
|
title = generate_text(chain, "Titles")
|
||
|
slug = slugify(title, max_length=70, word_boundary=True)
|
||
|
Title.create(text=title, slug=slug, site=site)
|
||
|
|
||
|
|
||
|
def add_question(site, count=100):
|
||
|
users = User.select().where(User.site == site).limit(count)
|
||
|
titles = Title.select().where(Title.site == site).limit(count)
|
||
|
chain = get_chain(site.url, "Questions")
|
||
|
|
||
|
for i in range(count):
|
||
|
text = generate_text(chain, "Questions")
|
||
|
title = titles[i]
|
||
|
print(title.text)
|
||
|
user = users[i]
|
||
|
print(user.username)
|
||
|
Question.create(text=text, title_id=title, user_id=user, site_id=site)
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
query = Site.select().where(Site.last_download.is_null(False)).limit(1)
|
||
|
s = query.get()
|
||
|
add_username(s)
|
||
|
add_title(s)
|
||
|
add_question(s)
|