You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
1012 B
37 lines
1012 B
from random import randint
|
|
from sqlalchemy.exc import IntegrityError
|
|
from faker import Faker
|
|
from . import db
|
|
from .models import User, Post
|
|
|
|
|
|
def users(count=100):
|
|
fake = Faker()
|
|
i = 0
|
|
while i < count:
|
|
u = User(email=fake.email(),
|
|
username=fake.user_name(),
|
|
password='password',
|
|
confirmed=True,
|
|
name=fake.name(),
|
|
location=fake.city(),
|
|
about_me=fake.text(),
|
|
member_since=fake.past_date())
|
|
db.session.add(u)
|
|
try:
|
|
db.session.commit()
|
|
i += 1
|
|
except IntegrityError:
|
|
db.session.rollback()
|
|
|
|
|
|
def posts(count=100):
|
|
fake = Faker()
|
|
user_count = User.query.count()
|
|
for i in range(count):
|
|
u = User.query.offset(randint(0, user_count - 1)).first()
|
|
p = Post(body=fake.text(),
|
|
timestamp=fake.past_date(),
|
|
author=u)
|
|
db.session.add(p)
|
|
db.session.commit()
|