You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
485 B
Python
29 lines
485 B
Python
from flask.cli import FlaskGroup
|
|
|
|
from project import create_app, db
|
|
from project.api.models import User
|
|
|
|
app = create_app()
|
|
cli = FlaskGroup(create_app=create_app)
|
|
|
|
|
|
@cli.command()
|
|
def recreate_db():
|
|
db.drop_all()
|
|
db.create_all()
|
|
db.session.commit()
|
|
|
|
|
|
@cli.command()
|
|
def seed_db():
|
|
"""Seeds the database."""
|
|
db.session.add(User(
|
|
username='michael',
|
|
email='michael@notreal.com',
|
|
))
|
|
db.session.commit()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cli()
|