使用SQLAlChemy从PostgreSQL行对象获取单个值

Question

我将Python与SQLAlchemy（在我的特殊情况下是GeoAlchemy）一起使用，并且我有一个查询，该查询将生成一列行对象。我希望从这些行对象中提取特定的值，但是SQLAlchemy将行对象解释为单个实体（如果我正确的话，将它们作为字符串返回）。如何在不使用Python解析的情况下更清晰地返回这些单独的值？

我的真实生活用例：PostgreSQL的PostGIS扩展提供了一个名为ST_IsValidDetail的函数。此函数旨在返回一个valid_detail行，该行由布尔值有效、字符串原因和发生无效的几何位置组成。我已经放弃了PostGIS标签，因为我觉得这个问题比那个更普遍。我的查询类似于SELECTST_IsValidDetail（'POINT（1 1）'::GEOMETRY）；（当然是更复杂的几何）。

Answer 1

我从ST_IsValidDetail页面上的示例中找到了一种方法。显然，以下语法是有效的：

SELECT gid, reason(ST_IsValidDetail(the_geom)), ST_AsText(location(ST_IsValidDetail(the_geom)))

注意围绕函数调用的原因和位置“调用”；ST_IsValidDetail返回的行中的列名称几乎被视为函数。事实证明，你可以欺骗SQLAlchemy做同样的事情。（假设session是先前设置的session对象，而db_geom是一个地质炼金术几何体对象。）

from sqlalchemy import func as sqlfunc
result = session.query(sqlfunc.reason(sqlfunc.ST_IsValidDetail(db_geom)), sqlfunc.ST_AsText(sqlfunc.location(sqlfunc.ST_IsValidDetail(db_geom)))).one()

result[0]将包含原因，result[1]将包含位置的WKT。（我们可以使用label给出列的实际名称。）

在不使用PostGIS功能的情况下对其进行修剪：

from sqlalchemy import func as sqlfunc
result = session.query(sqlfunc.columnname(sqlfunc.myrowfunc('some input string')).label('mylabel')).one()

这使得SQLAlChemy认为列名是一个函数，并以表单将SQL发送到数据库

SELECT columnname(myrowfunc('some input string')) AS mylabel;

我还没有尝试过它，但是如果有办法让SqLalCimy考虑我们的行是表，我们选择<代码>从< /代码>，这也可以工作。（请参见ST_IsValidDetail页面的最底部。）

Answer 2

您可以使用：

SELECT (ST_IsValidDetail(the_value)).* FROM the_table;

... 但不幸的是，PostgreSQL实际上对每行执行一次ST_IsValidDetail函数。作为一种解决方法，您可以稍微修改查询，通过公共表表达式具体化，然后在第二次过程中提取元组：

WITH interim_result(v) AS (
    SELECT ST_IsValidDetail(the_value) FROM the_table
)
SELECT (v).* FROM interim_result;

（v）周围的参数需要告诉解析器您引用的是值，而不是表名。

演示：

CREATE OR REPLACE FUNCTION multirows(x IN integer, a OUT integer, b OUT integer, c OUT integer) AS
$$
BEGIN
    RAISE NOTICE 'multirows(%) invoked', x;
    a := x;
    b := x+1;
    c := x+2;
    RETURN;
END;
$$ LANGUAGE plpgsql;

craig=> SELECT multirows(x) FROM generate_series(1,2) x;
NOTICE:  multirows(1) invoked
NOTICE:  multirows(2) invoked
 multirows 
-----------
 (1,2,3)
 (2,3,4)
(2 rows)

craig=> SELECT (multirows(x)).* FROM generate_series(1,2) x;
NOTICE:  multirows(1) invoked
NOTICE:  multirows(1) invoked
NOTICE:  multirows(1) invoked
NOTICE:  multirows(2) invoked
NOTICE:  multirows(2) invoked
NOTICE:  multirows(2) invoked
 a | b | c 
---+---+---
 1 | 2 | 3
 2 | 3 | 4
(2 rows)




craig=> WITH interim(v) AS (SELECT multirows(x) FROM generate_series(1,2) x)
SELECT (v).* FROM interim;
NOTICE:  multirows(1) invoked
NOTICE:  multirows(2) invoked
 a | b | c 
---+---+---
 1 | 2 | 3
 2 | 3 | 4
(2 rows)