diff --git a/create_app.py b/create_app.py index 913fee1..ec0d08d 100644 --- a/create_app.py +++ b/create_app.py @@ -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 diff --git a/forms.py b/forms.py index 1d05b27..6867c83 100644 --- a/forms.py +++ b/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__) diff --git a/models.py b/models.py index 4c84f6c..c25d893 100644 --- a/models.py +++ b/models.py @@ -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)