from flask.ext.script import Manager, prompt_bool
from api import app, db
from api.models import Designer

manager = Manager(app)

@manager.command
def initdb():
  db.create_all()
  print 'All tables created'

@manager.command
def dropdb():
  if prompt_bool("Are you sure you want to lose all your data"):
    db.drop_all()
    print 'All tables dropped'

@manager.command
def seed():
  print 'Adding sample data %s' % db.engine.url
  designer = Designer(name="Martin", path='/martin', username='martin', password='secret')
  db.session.add(designer)
  db.session.commit()


if __name__ == '__main__':
  manager.run()