首页 文章

如果使用Marshmallow,SQLAlchemy不会创建表

提问于
浏览
1

我正在尝试使用 FlaskFlask-SQLAlchemyflask-restplusmarshmallow (也尝试使用 flask-marshmallow )创建项目,但这些工具的集成效果不佳 . 每次我包含一些新的lib时,都会出现新的错误 .

我已经放弃使用 migrationsFlask-SQLAlchemy 因为某些原因,这件事不起作用 . 但现在,问题在于 marshmallow .

我正在尝试使用模块,我认为这是问题的一部分(Flask-SQLAlchmey,flask-restplus,flask-marshmallow等的所有示例都将所有内容放在一个文件中)

这是我的 app.py

from flask import Flask, Blueprint

import settings
from api import api
from database import init_database, reset_database

app = Flask(__name__)


def configure_app(flask_app):
    flask_app.config['SERVER_NAME'] = settings.SERVER_ADDRESS
    flask_app.secret_key = settings.SECRET_KEY


def initialize_app(flask_app):
    configure_app(flask_app)

    blueprint = Blueprint('api', __name__, url_prefix=settings.URL_PREFIX)
    api.init_app(blueprint)
    # api.add_namespace(auth_login_namespace)
    flask_app.register_blueprint(blueprint)

    init_database(flask_app)
    if settings.DEBUG:
        reset_database(flask_app)


def main():
    initialize_app(app)
    app.run(debug=settings.DEBUG)

if __name__ == '__main__':
    main()

api/__init__.py

import settings

from flask_restplus import Api

api = Api(
    version='1.0',
    title='Test',
    description='Some description'
)


@api.errorhandler
def default_error_handler(e):
    message = 'An unhandled exception occurred.'

    if not settings.DEBUG:
        return {'message': message}, 500

database/__init__.py

from flask_sqlalchemy import SQLAlchemy

import settings

db = SQLAlchemy()


def init_database(flask_app):
    flask_app.config['SQLALCHEMY_DATABASE_URI'] = settings.DATABASE_URI
    flask_app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

    global db # I really don't like this! :(
    db = SQLAlchemy(flask_app)


def reset_database(flask_app):
    from database.models import User
    db.drop_all()
    db.create_all()
    db.session.add(User(username='admin', email='abc@def.com', name='admin', password='123', admin=True)
    db.session.commit()

我有我的应用程序,直到现在只有一个 model

database/models/User.py

from marshmallow import Schema, fields
from sqlalchemy import func

from database import db


class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, index=True)
    email = db.Column(db.String(120), unique=True, index=True)
    name = db.Column(db.String(200), nullable=False)
    password = db.Column(db.String(200), nullable=False)
    admin = db.Column(db.Boolean, nullable=False, default=False)
    created_on = db.Column(db.DateTime, nullable=False, server_default=func.now())


class UserSchema(Schema):
    id = fields.Int(dump_only=True)
    username = fields.Str()
    email = fields.Email()
    name = fields.Str()
    password = fields.Str()
    admin = fields.Bool()
    created_on = fields.DateTime()

现在,如果我使用以下代码(每次我的应用程序在调试模式下启动时调用此代码,在函数 reset_database 上,在文件_1295178中):

db.drop_all()
db.create_all()
db.session.add(User(username='admin', email='abc@def.com', name='admin', password='123', admin=True)
db.session.commit()

未创建 User 表,并且未在表上插入 admin ,因为该表不存在( db.create_all() 不创建任何内容) .

sqlite3.OperationalError: no such table: user .

出于某种原因,如果我删除类 UserSchema (在 database/models/User.py 上), db.create_all() 函数会创建表格 .

1 回答

  • 0

    现在就开始工作了:

    database/__init__.py 上,将 init_database 函数更改为:

    def init_database(flask_app):
        flask_app.config['SQLALCHEMY_DATABASE_URI'] = settings.DATABASE_URI
        flask_app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
    
        db.init_app(flask_app)
    

    并在 database/reset_database.py

    def reset_database(flask_app):
        with flask_app.app_context():
            from database.models.user_module import User
            db.drop_all()
            db.create_all()
            db.session.add(User(username='admin', email='abc@def.com', name='admin', password='123', admin=True))
            db.session.commit()
    

    问题是 init_app ,使用 app_context ,我导入了错误的 User 模块(tks Fian)

相关问题