51 lines
1.0 KiB
Python
51 lines
1.0 KiB
Python
from flask.cli import FlaskGroup
|
|
|
|
from src import create_app, db
|
|
from src.api.models.users import Roles, User
|
|
|
|
app = create_app()
|
|
cli = FlaskGroup(create_app=create_app)
|
|
|
|
|
|
@cli.command("recreate_db")
|
|
def recreate_db():
|
|
db.drop_all()
|
|
db.create_all()
|
|
db.session.commit()
|
|
|
|
|
|
@cli.command("seed_db")
|
|
def seed_db():
|
|
db.session.add(
|
|
User(
|
|
username="lufthansa",
|
|
email="info@lufthansa.com",
|
|
password="password1234",
|
|
role=Roles.airline,
|
|
)
|
|
)
|
|
db.session.add(
|
|
User(
|
|
username="ryanair",
|
|
email="info@ryanair.com",
|
|
password="password1234",
|
|
role=Roles.airline,
|
|
)
|
|
)
|
|
db.session.add(
|
|
User(
|
|
username="admin",
|
|
email="admin",
|
|
password="password1234",
|
|
role=Roles.admin,
|
|
)
|
|
)
|
|
db.session.add(
|
|
User(username="messi", email="messi@gmail.com", password="password1234")
|
|
)
|
|
db.session.commit()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|