Question

我正在使用PostgreSQL + Psycopg2，SQLAlchemy。我已经使用pgAdminIII工具创建了我的数据库“new_db”，其中的新模式为“new_db_schema”。在这个模式下，我有我需要的所有表。我的代码看起来像这样。

from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy import Column, String, Integer, Boolean

engine_text = 'postgresql+psycopg2://root:12345@localhost:5432/db_name'
database_engine = create_engine(engine_text, echo = True)

Base = declarative_base(database_engine)

class Some_table(Base):
    __tablename__ = 'some_table' # This table exist in database.
    __table_args__ = {'autoload':True}

    col1 = Column(String, primary_key=True)
    col2 = Column(Boolean)

    def __init__(self, col1_name, col2_name):
        self.col1 = col1_name
        self.col2 = col2_name

if __name__ == "__main__":
    a = Some_table('blah', 'blah')

现在，当我尝试运行以下代码时，我得到sqlalchemy.exc.NoSuchTableError: some_table。

由于我已经拥有所有表的数据库设置，我想在创建类时自动加载。我在这里错过了什么吗？我需要为数据库中的所有表编写这样的类。任何帮助将不胜感激。

由于

Answer 1

你可以：

使用模式限定的表名：__tablename__ = 'new_db_schema.some_table'。你必须在任何地方使用它们：在ForeignKey等的字符串参数中。
更改数据库中的SEARCH_PATH：SET search_path TO new_db_schema;。此SQL命令具有会话范围，因此您必须在使用SQLAlchemy事件系统的每个连接的开头发出它。

像这样：

from sqlalchemy import event

def init_search_path(connection, conn_record):
    cursor = connection.cursor()
    try:
        cursor.execute('SET search_path TO new_db_schema;')
    finally:
        cursor.close()

engine = create_engine('postgresql+psycopg2://...')
event.listen(engine, 'connect', init_search_path)

当数据库中存在表时，sqlalchemy NoSuchTableError

1 个答案: