mirror of
https://github.com/Xevion/runnerspace.git
synced 2026-01-31 12:25:51 -06:00
Develop shell command for instantiating all DB tables fast, import all models
This commit is contained in:
+5
-1
@@ -25,7 +25,7 @@ def create_app():
|
||||
login_manager.login_view = 'auth.login'
|
||||
login_manager.init_app(app)
|
||||
|
||||
from .models import User
|
||||
from .models import User, Post, Comment
|
||||
|
||||
@login_manager.user_loader
|
||||
def load_user(user_id):
|
||||
@@ -82,4 +82,8 @@ def create_app():
|
||||
print(f'Committing {len(users)} users into DB.')
|
||||
db.session.commit()
|
||||
|
||||
@app.cli.command("create_all")
|
||||
def db_create_all() -> None:
|
||||
db.create_all(app=app)
|
||||
|
||||
return app
|
||||
|
||||
Reference in New Issue
Block a user