seed.py

"""Seed database with sample data from CSV Files."""

from csv import DictReader
from app import db
from models import User, Message, Follows, Likes, DEFAULT_HEADER_IMAGE_URL, DEFAULT_IMAGE_URL

db.drop_all()
db.create_all()

with open('generator/users.csv') as users:
    db.session.bulk_insert_mappings(User, DictReader(users))

with open('generator/messages.csv') as messages:
    db.session.bulk_insert_mappings(Message, DictReader(messages))

with open('generator/follows.csv') as follows:
    db.session.bulk_insert_mappings(Follows, DictReader(follows))

user1 = User.signup(username='Vaughn', email='vseekamp@gmail.com', image_url=DEFAULT_IMAGE_URL, password='password')
user2 = User.signup(username='Madelyn', email='madelyn@gmail.com', image_url=DEFAULT_IMAGE_URL, password='password')

db.session.add_all([user1, user2])


db.session.commit()