CICD/commands/seed_command.py

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)
# 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_things()
            db.session.commit()
            print('DB successfully seeded.')