来自 mssql 数据库的 sqlalchemy 映射表,带有“前缀命名空间”

joa*_*inn 2 python sql sql-server orm sqlalchemy

我已经为此苦苦挣扎了一段时间,但还没有找到答案,或者也许我已经看到了答案,只是没有得到它 - 但是,我希望我能够描述我的问题。

我有一个 MS SQL 数据库,其中的表按名称空间(或任何名称)分组,由 Prefix.Tablename(带点)表示。因此,请求某些内容的本机 sql 语句如下所示:

    SELECT TOP 100 
     [Value], [ValueDate]
    FROM [FinancialDataBase].[Reporting].[IndexedElements]
Run Code Online (Sandbox Code Playgroud)

如何将其映射到 sqlalchemy?如果“Reporting”前缀不存在,解决方案(或一种方法)如下所示:

from sqlalchemy import *
from sqlalchemy.ext.declarative import declarative_base, declared_attr
from sqlalchemy.orm import sessionmaker

def get_session():
    from urllib.parse import quote_plus as urllib_quote_plus

    server = "FinancialDataBase.sql.local"
    connstr = "DRIVER={SQL Server};SERVER=%s;DATABASE=FinancialDataBase" % server
    params = urllib_quote_plus(connstr)
    base_url = "mssql+pyodbc:///?odbc_connect=%s" % params

    engine  = create_engine(base_url,echo=True)
    Session = sessionmaker(bind=engine)
    session = Session()

    return engine, session

Base = declarative_base()

class IndexedElements(Base):
    __tablename__ = "IndexedElements"

    UniqueID = Column(String,primary_key=True)
    ValueDate = Column(DateTime)
    Value = Column(Float)
Run Code Online (Sandbox Code Playgroud)

然后可以完成请求并将其包装在 Pandas 数据框中,例如如下所示:

import pandas as pd

engine, session = get_session()

query = session.query(IndexedElements.Value,IndexedElements.ValueDate)

data = pd.read_sql(query.statement,query.session.bind)
Run Code Online (Sandbox Code Playgroud)

但是这里编译并实际执行的SQL语句包含这个错误的FROM部分:

FROM [FinancialDataBase].[IndexedElements]
Run Code Online (Sandbox Code Playgroud)

由于命名空间前缀,它必须是

FROM [FinancialDataBase].[Reporting].[IndexedElements]
Run Code Online (Sandbox Code Playgroud)

只需将表名称扩展为

__tablename__ = "Reporting.IndexedElements"
Run Code Online (Sandbox Code Playgroud)

没有修复它,因为它将编译的sql语句更改为

FROM [FinancialDataBase].[Reporting.IndexedElements]
Run Code Online (Sandbox Code Playgroud)

这不能正常工作。

那么如何解决这个问题呢?

joa*_*inn 5

答案在上面 Ilja 的评论中给出:

“命名空间”是所谓的模式,必须在映射对象中声明。鉴于开头文章中的示例,映射表必须如下定义:

class IndexedElements(Base):
    __tablename__ = "IndexedElements"
    __table_args__ = {"schema": "Reporting"}

    UniqueID = Column(String,primary_key=True)
    ValueDate = Column(DateTime)
    Value = Column(Float)
Run Code Online (Sandbox Code Playgroud)

或者定义一个包含不同模式的这些信息的基类。另请检查 sqlalchemy 文档中的“增强基础”: http://docs.sqlalchemy.org/en/latest/orm/extensions/declarative/mixins.html#augmenting-the-base