49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
|
from werkzeug.security import generate_password_hash
|
||
|
|
from models import db, User, News, Comment
|
||
|
|
from faker import Factory
|
||
|
|
from random import randint
|
||
|
|
|
||
|
|
# Spanish
|
||
|
|
fake = Factory.create('es_ES')
|
||
|
|
|
||
|
|
# Reload tables
|
||
|
|
db.drop_all()
|
||
|
|
db.create_all()
|
||
|
|
|
||
|
|
# Make 100 fake users
|
||
|
|
for num in range(100):
|
||
|
|
profile = fake.simple_profile()
|
||
|
|
username = profile['username']
|
||
|
|
mail = profile['mail']
|
||
|
|
password = generate_password_hash('123')
|
||
|
|
# Save in database
|
||
|
|
my_user = User(username=username, mail=mail, password=password)
|
||
|
|
db.session.add(my_user)
|
||
|
|
|
||
|
|
print('Users created')
|
||
|
|
|
||
|
|
# Make 1000 fake news
|
||
|
|
for num in range(1000):
|
||
|
|
title = fake.sentence()
|
||
|
|
link = fake.uri()
|
||
|
|
user_id = randint(1, 100)
|
||
|
|
# Save in database
|
||
|
|
my_news = News(title=title, link=link, user_id=user_id)
|
||
|
|
db.session.add(my_news)
|
||
|
|
|
||
|
|
print('News created')
|
||
|
|
|
||
|
|
# Make 10000 fake comments
|
||
|
|
for num in range(10000):
|
||
|
|
text = fake.text()
|
||
|
|
news_id = randint(1, 1000)
|
||
|
|
user_id = randint(1, 100)
|
||
|
|
# Save in database
|
||
|
|
my_comment = Comment(text=text, news_id=news_id, user_id=user_id)
|
||
|
|
db.session.add(my_comment)
|
||
|
|
|
||
|
|
print('Comments created')
|
||
|
|
|
||
|
|
# Save
|
||
|
|
db.session.commit()
|