41 lines
956 B
Python
41 lines
956 B
Python
|
from datetime import datetime
|
||
|
import pandas as pd
|
||
|
import numpy as np
|
||
|
from flask_script import Command
|
||
|
|
||
|
|
||
|
from app import db
|
||
|
|
||
|
|
||
|
def seed_widgets():
|
||
|
from app.widget import Widget
|
||
|
widgets = [
|
||
|
{
|
||
|
'name': 'Pizza Slicer',
|
||
|
'purpose': 'Cut delicious pizza',
|
||
|
},
|
||
|
{
|
||
|
'name': 'Rolling Pin',
|
||
|
'purpose': 'Roll delicious pizza',
|
||
|
},
|
||
|
{
|
||
|
'name': 'Pizza Oven',
|
||
|
'purpose': 'Bake delicious pizza',
|
||
|
},
|
||
|
]
|
||
|
db.session.bulk_insert_mappings(Widget, widgets)
|
||
|
|
||
|
|
||
|
class SeedCommand(Command):
|
||
|
""" Seed the DB."""
|
||
|
|
||
|
def run(self):
|
||
|
if input('ARE YOU SURE YOU WANT TO DROP ALL TABLES AND RECREATE? (Y/N)\n'
|
||
|
).lower() == 'y':
|
||
|
print('Dropping tables...')
|
||
|
db.drop_all()
|
||
|
db.create_all()
|
||
|
seed_widgets()
|
||
|
db.session.commit()
|
||
|
print('Widgets successfully seeded.')
|