SQLAlchemy create_all()不创建表

时间:2013-12-23 13:09:19

标签: python postgresql sqlalchemy

我正在尝试集成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

我该如何解决这个问题?

3 个答案:

答案 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”函数。

this thread

@SuperShoot 的回答更好地解释了这个概念