mirror of
https://github.com/Xevion/runnerspace.git
synced 2025-12-11 23:11:13 -06:00
Develop shell command for instantiating all DB tables fast, import all models
This commit is contained in:
@@ -25,7 +25,7 @@ def create_app():
|
|||||||
login_manager.login_view = 'auth.login'
|
login_manager.login_view = 'auth.login'
|
||||||
login_manager.init_app(app)
|
login_manager.init_app(app)
|
||||||
|
|
||||||
from .models import User
|
from .models import User, Post, Comment
|
||||||
|
|
||||||
@login_manager.user_loader
|
@login_manager.user_loader
|
||||||
def load_user(user_id):
|
def load_user(user_id):
|
||||||
@@ -82,4 +82,8 @@ def create_app():
|
|||||||
print(f'Committing {len(users)} users into DB.')
|
print(f'Committing {len(users)} users into DB.')
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|
||||||
|
@app.cli.command("create_all")
|
||||||
|
def db_create_all() -> None:
|
||||||
|
db.create_all(app=app)
|
||||||
|
|
||||||
return app
|
return app
|
||||||
|
|||||||
2
forms.py
2
forms.py
@@ -1,7 +1,7 @@
|
|||||||
from flask import Blueprint, flash, redirect, request, url_for
|
from flask import Blueprint, flash, redirect, request, url_for
|
||||||
from flask_login import current_user, login_required
|
from flask_login import current_user, login_required
|
||||||
|
|
||||||
from create_app import db
|
from .create_app import db
|
||||||
from .models import User
|
from .models import User
|
||||||
|
|
||||||
blueprint = Blueprint('forms', __name__)
|
blueprint = Blueprint('forms', __name__)
|
||||||
|
|||||||
@@ -63,4 +63,4 @@ class Comment(db.Model):
|
|||||||
id = db.Column(db.Integer, primary_key=True)
|
id = db.Column(db.Integer, primary_key=True)
|
||||||
text = db.Column(db.Text, nullable=False)
|
text = db.Column(db.Text, nullable=False)
|
||||||
author = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
|
author = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
|
||||||
post = db.Column(db.Integer, db.ForeignKey('post.id'), nulllable=False)
|
post = db.Column(db.Integer, db.ForeignKey('post.id'), nullable=False)
|
||||||
|
|||||||
Reference in New Issue
Block a user