mirror of
https://github.com/Xevion/runnerspace.git
synced 2025-12-08 14:08:20 -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.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
|
||||
|
||||
2
forms.py
2
forms.py
@@ -1,7 +1,7 @@
|
||||
from flask import Blueprint, flash, redirect, request, url_for
|
||||
from flask_login import current_user, login_required
|
||||
|
||||
from create_app import db
|
||||
from .create_app import db
|
||||
from .models import User
|
||||
|
||||
blueprint = Blueprint('forms', __name__)
|
||||
|
||||
@@ -63,4 +63,4 @@ class Comment(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
text = db.Column(db.Text, 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