Nhận tin nhắn cảnh báo qua psycopg2

Question

Tôi muốn gọi hàm plpgsql thông qua psycopg2 và xem các thông báo cảnh báo. tức là, tôi có chức năng này:

create or replace function test_warning() returns void as $$ 
begin 
raise warning 'this is only a test'; 
end; 
$$ 
language plpgsql; 

và gọi nó như vậy trong python:

import psycopg2 
conn = psycopg2.connect(conn_string) 
cursor = conn.cursor() 
cursor.callproc("test_warning") 
# or so: 
cursor.execute('SELECT test_warning()') 

Thật không may là thông điệp cảnh báo theo quy định tại plpgsql không xuất hiện bất cứ nơi nào trong đầu ra python. Có cách nào để nhận thông điệp cảnh báo được in trong đầu ra python không?

Answer 1

Các notices thành viên của connection là một danh sách các thông điệp của phiên gửi cho khách hàng đến thời điểm đó:

for notice in conn.notices: 
    print notice 

http://initd.org/psycopg/docs/connection.html#connection.notices

Để có được thông báo lần cuối:

print conn.notices[-1] 

Nếu ngoại lệ được nâng lên bên trong một hàm, và không bị bắt, sẽ không nhận được cảnh báo nào. Đó là bởi vì một hàm kết thúc một giao dịch tiềm ẩn và mọi thứ bên trong giao dịch đó được khôi phục bao gồm các cảnh báo.