Flask SQLAlchemy不使用create_all()创建表

2024-05-14 13:26:30 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个Flask应用程序,它使用Flask SQLAlchemy。在我的单元测试中,我初始化了应用程序和数据库,然后调用db.create_all(),出于某种原因,它看起来没有接受我的任何模型,所以没有创建任何表。

我在模型中同时使用__tablename____bind_key__,因为我有两个数据库。

我的配置:

SQLALCHEMY_DATABASE_URI = 'sqlite://'

SQLALCHEMY_BINDS = {
    'db1': SQLALCHEMY_DATABASE_URI,
    'db2': SQLALCHEMY_DATABASE_URI
}

我的单元测试中我的setUp()方法的精简版本:

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy

class APITestCase(unittest.TestCase):

    app = Flask(__name__)
    db = SQLAlchemy(app)
    db.create_all()

下面是我的两个模型的示例,每个绑定类型一个:

class Contact(db.Model):

    __tablename__ = 'contact'
    __bind_key__ = 'db1'

    id = db.Column(db.Integer, primary_key=True)
    first_name = db.Column(db.String(255))
    last_name = db.Column(db.String(255))
    ...

    def __init__(first_name, last_name):
        self.first_name = first_name
        self.last_name = last_name
        ...

class BaseUser(db.Model):

    __tablename__ = 'User__GeneralUser'
    __bind_key__ = 'db2'

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column('Username', db.String(100))
    first_name = db.Column('FirstName', db.String(100))
    last_name = db.Column('Surname', db.String(100))
    ...

    def __init__(username, first_name, last_name):
        self.username = username
        self.first_name = first_name
        self.last_name = last_name
        ...

我有没有漏掉一些显而易见的东西?Flask SQLAlchemy如何知道在哪里查找我的模型来创建关联的表?


Tags: keyname模型selfflaskdbstringsqlalchemy
2条回答

在单元测试中,不应创建SQLAlchemy对象('db')的新实例,应导入模型从中派生的实例:

型号.py:

from flask.ext.sqlalchemy import SQLAlchemy
db = SQLAlchemy()

class Contact(db.Model):
    ...

测试.py:

from models import db
from flask import Flask
import unittest

class TestExample(unittest.TestCase):

    def setUp(self):
        self.app = Flask(__name__)
        db.init_app(self.app)
        with self.app.app_context():
            db.create_all()

您的SQLALCHEMY_数据库URI错误,请阅读此处如何配置引擎Sqlite数据库http://docs.sqlalchemy.org/en/rel_0_9/core/engines.html#sqlite

如果您从未体验过,请查看此回购并应用于您的应用:) https://github.com/Fird0s/Banda-Maps/blob/master/models.py

相关问题 更多 >

    热门问题