ataxya/CICD
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Cleanup display messages

Browse Source
This commit is contained in:
Alan Pryor 2019-08-03 12:50:36 -04:00
parent fadafbdee9
commit 0e8ee9a58f
2 changed files with 2 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
commands/seed_command.py
View File

@ -34,30 +34,12 @@ def seed_thing(cls):
},
]
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'
if input('Are you sure you want to drop all tables and recreate? (y/N)\n'
).lower() == 'y':
print('Dropping tables...')
db.drop_all()

2
manage.py
View File

@ -26,7 +26,7 @@ def init_db():
@manager.command
def drop_all():
if input('ARE YOU SURE YOU WANT TO DROP ALL TABLES? (Y/N)\n').lower() == 'y':
if input('Are you sure you want to drop all tables? (y/N)\n').lower() == 'y':
print('Dropping tables...')
db.drop_all()