mirror of
https://github.com/simple-login/app.git
synced 2024-09-20 15:05:59 +08:00
add dummy-data flask command
This commit is contained in:
parent
1384ccc459
commit
4cbbf260d4
15
server.py
15
server.py
|
@ -204,13 +204,7 @@ def create_app() -> Flask:
|
|||
|
||||
def fake_data():
|
||||
LOG.d("create fake data")
|
||||
# Remove db if exist
|
||||
if os.path.exists("db.sqlite"):
|
||||
LOG.d("remove existing db file")
|
||||
os.remove("db.sqlite")
|
||||
|
||||
# Create all tables
|
||||
db.create_all()
|
||||
|
||||
# Create a user
|
||||
user = User.create(
|
||||
|
@ -918,6 +912,15 @@ def register_custom_commands(app):
|
|||
LOG.d("finish trunk %s, update %s email logs", trunk, nb_update)
|
||||
db.session.commit()
|
||||
|
||||
@app.cli.command("dummy-data")
|
||||
def dummy_data():
|
||||
from init_app import add_sl_domains
|
||||
|
||||
LOG.warning("reset db, add fake data")
|
||||
with app.app_context():
|
||||
fake_data()
|
||||
add_sl_domains()
|
||||
|
||||
|
||||
def setup_do_not_track(app):
|
||||
@app.route("/dnt")
|
||||
|
|
Loading…
Reference in a new issue