我正在尝试集成PostgreSQL和SQLAlchemy,但SQLAlchemy.create_all()没有从我的模型中创建任何表。
我的代码:
from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql+psycopg2://login:pass@localhost/flask_app'
db = SQLAlchemy(app)
db.create_all()
db.session.commit()
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True)
email = db.Column(db.String(120), unique=True)
def __init__(self, username, email):
self.username = username
self.email = email
def __repr__(self):
return '<User %r>' % self.username
admin = User('admin', 'admin@example.com')
guest = User('guest', 'guest@example.com')
db.session.add(admin)
db.session.add(guest)
db.session.commit()
users = User.query.all()
print users
但是我收到了这个错误:sqlalchemy.exc.ProgrammingError: (ProgrammingError) relation "user" does not exist
我该如何解决这个问题?
答案 0 :(得分:74)
您应该在create_all()
调用之前放置模型类,如下所示:
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql+psycopg2://login:pass@localhost/flask_app'
db = SQLAlchemy(app)
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True)
email = db.Column(db.String(120), unique=True)
def __init__(self, username, email):
self.username = username
self.email = email
def __repr__(self):
return '<User %r>' % self.username
db.create_all()
db.session.commit()
admin = User('admin', 'admin@example.com')
guest = User('guest', 'guest@example.com')
db.session.add(admin)
db.session.add(guest)
db.session.commit()
users = User.query.all()
print users
如果您的模型在单独的模块中声明,请在调用create_all()
之前导入它们。
说,User
模型位于名为models.py
的文件中,
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql+psycopg2://login:pass@localhost/flask_app'
db = SQLAlchemy(app)
# See important note below
from models import User
db.create_all()
db.session.commit()
admin = User('admin', 'admin@example.com')
guest = User('guest', 'guest@example.com')
db.session.add(admin)
db.session.add(guest)
db.session.commit()
users = User.query.all()
print users
重要说明:在初始化db
对象后导入模型非常重要,因为在models.py
_中您还需要导入db
来自这个模块的对象。
答案 1 :(得分:2)
这可能不是 create_all()
方法调用对人们不起作用的主要原因,但对我来说,来自各种教程的拼凑说明有它,因此我在请求上下文中创建了我的数据库,这意味着我有类似的东西:
# lib/db.py
from flask import g, current_app
from flask_sqlalchemy import SQLAlchemy
def get_db():
if 'db' not in g:
g.db = SQLAlchemy(current_app)
return g.db
我还有一个单独的 cli 命令也可以执行 create_all:
# tasks/db.py
from lib.db import get_db
@current_app.cli.command('init-db')
def init_db():
db = get_db()
db.create_all()
我也在使用应用程序工厂。
运行 cli 命令时,会使用新的应用程序上下文,这意味着使用新的 db。此外,在这个世界中,init_db 方法中的导入模型不会做任何事情,因为可能您的模型文件已经加载(并且与单独的数据库相关联)。
我想到的解决办法是确保数据库是一个单一的全局引用:
# lib/db.py
from flask import g, current_app
from flask_sqlalchemy import SQLAlchemy
db = None
def get_db():
global db
if not db:
db = SQLAlchemy(current_app)
return db
我对flask、sqlalchemy 或flask-sqlalchemy 的研究不够深入,无法理解这是否意味着从多个线程向数据库发出的请求是安全的,但是如果您正在阅读本文,您可能会陷入婴儿阶段理解这些概念。
答案 2 :(得分:0)
如果有人在使用专用于每个模型的文件创建表时遇到问题,请注意从与声明该函数的文件不同的文件运行“create_all”函数。 所以,如果文件系统是这样的:
Root
--app.py <-- file from which app will be run
--models
----user.py <-- file with "User" model
----order.py <-- file with "Order" model
----database.py <-- file with database and "create_all" function declaration
注意从 app.py 调用“create_all”函数。
对 @SuperShoot 的回答更好地解释了这个概念