示例#1
0
def seed_db():
    from models.Book import Book  # Import the book model
    from models.User import User  # User model
    from main import bcrypt  # Hashing module
    from faker import Faker  # Import the faker module
    import random

    faker = Faker()  # Create an instance of faker
    users = []  # Initializing an empty list

    # Creating 5 users and appending them to the users list
    for i in range(5):
        user = User()
        user.email = f"test{i}@test.com"
        user.password = bcrypt.generate_password_hash("123456").decode("utf-8")
        db.session.add(user)
        users.append(user)

    db.session.commit()

    for i in range(20):  # 20
        book = Book()  # New instance of book
        book.title = faker.catch_phrase()  # Add a title
        book.user_id = random.choice(
            users).id  # Choosing a random user to assign the book to
        db.session.add(book)  # add the book to the db session

    db.session.commit()  # Commit all the books to the db
    print("Tables seeded")
示例#2
0
def seed_db():
    from models.Book import Book
    from models.User import User
    from main import bcrypt
    from faker import Faker
    import random

    faker = Faker()
    users = []

    for i in range(5):
        user = User()
        user.email = f"test{i}@test.com"
        user.password = bcrypt.generate_password_hash("123456").decode("utf-8")
        db.session.add(user)
        users.append(user)

    db.session.commit()

    for i in range(20):
        book = Book()
        book.title = faker.catch_phrase()
        book.user_id = random.choice(users).id
        db.session.add(book)

    db.session.commit()
    print("Tables seeded")
示例#3
0
def seed_db():
    from models.Book import Book
    from models.User import User
    from models.Page import Page
    from main import bcrypt
    from faker import Faker
    import random

    faker = Faker()
    users = []

    for i in range(5):
        user = User()
        user.email = f"test{i}@test.com"
        user.password = bcrypt.generate_password_hash("123456").decode("utf-8")
        db.session.add(user)
        users.append(user)

    db.session.commit()

    for i in range(20):
        book = Book()
        book.title = faker.catch_phrase()
        book.user_id = random.choice(users).id
        book.author = faker.name()
        db.session.add(book)

    db.session.commit()

    for i in range(100):
        page = Page()
        page.page_content = faker.text(1500)
        page.page_number = random.randint(1, 300)
        page.book_id = random.randint(1, 20)
        db.session.add(page)
    db.session.commit()

    print("Tables seeded")