49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
from datetime import datetime
|
|
import pandas as pd
|
|
import numpy as np
|
|
from flask_script import Command
|
|
|
|
from app import db
|
|
from app.widget import Widget
|
|
from app.fizz.fizzbaz import Fizzbaz
|
|
from app.fizz.fizzbar import Fizzbar
|
|
from app.other_api.doodad import Doodad
|
|
from app.other_api.whatsit import Whatsit
|
|
|
|
|
|
def seed_things():
|
|
classes = [Widget, Fizzbaz, Fizzbar, Doodad, Whatsit]
|
|
for klass in classes:
|
|
seed_thing(klass)
|
|
|
|
|
|
def seed_thing(cls):
|
|
things = [
|
|
{
|
|
'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(cls, things)
|
|
|
|
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_things()
|
|
db.session.commit()
|
|
print('DB successfully seeded.')
|