Python3 psycopg2:使用RETURNING进行“无结果提取”

时间:2019-03-28 14:55:49

标签: python python-3.x postgresql psycopg2 postgresql-11

我正在使用:

  • psvcopg2
  • python3.6
  • postgresql-11.2
  • kubuntu 18.10

并且我成功插入了一行,但是cursor.fetchall()正在提高

psycopg2.ProgrammingError: no results to fetch

即使我在SQL内部使用RETURNING来检索ID。

我发现了有关cursor.description的信息,但它为空(cursor.description == None)。

在psql终端中,SQL可正确使用,并按要求返回ID。

python代码

import psycopg2
from psycopg2.pool import ThreadedConnectionPool

pool = ThreadedConnectionPool(3, 20,
                              user="user",
                              password='xxxxxxxxx',
                              host="127.0.0.1",
                              port="5432",
                              database="my_database")

query = 'INSERT INTO market.item(item_store_id, title, price, url, image_url, aff_url, store_id) ' \
        'VALUES(%(item_store_id)s, %(title)s, %(price)s, %(url)s, %(image_url)s, %(aff_url)s, %(store_id)s) ' \
        'ON CONFLICT (item_store_id) DO ' \
        'UPDATE SET (price, url, image_url, aff_url) = (excluded.price, excluded.url, excluded.image_url, excluded.aff_url) ' \
        'RETURNING item_id '
args = [{
    'item_store_id': 1,
    'title': 'My title',
    'price': 15,
    'url': 'http://www.url.com',
    'image_url': 'http://www.url.com',
    'aff_url': 'http://www.url.com',
    'store_id': 1,
}]
try:
    result = []
    connection = pool.getconn()
    connection.autocommit = True
    with connection.cursor() as cursor:
        try:
            cursor.executemany(query, args)
            if cursor.rownumber > 0:
                subresult = cursor.fetchall()
                result.append(subresult)

            print(result)
        except (Exception, psycopg2.DatabaseError) as e:
            raise
except (Exception, psycopg2.DatabaseError) as error:
    print(e)
else:
    print(result)
finally:
    pool.putconn(connection)

1 个答案:

答案 0 :(得分:0)

我的坏。我在文档中没有注意到这一点:

The function is mostly useful for commands that update the database: any result set returned by the query is discarded.