From 8c2ac0658cf9882c81e90de11c1010467b55ad5e Mon Sep 17 00:00:00 2001
From: Federico Di Gregorio <fog@initd.org>
Date: Mon, 9 May 2005 09:07:07 +0000
Subject: [PATCH] Added error codes to messages.

---
 NEWS             |  2 ++
 psycopg/pqpath.c | 14 ++++++++++++--
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/NEWS b/NEWS
index d9201a1b..69086243 100644
--- a/NEWS
+++ b/NEWS
@@ -11,6 +11,8 @@ What's new in psycopg 2.0 beta 1
 
 * Complete support for BYTEA columns and buffer objects.
 
+* Added error codes to error messages.
+
 * The AsIs adapter is now exported by default (also Decimal objects are
   adapter using the AsIs adapter (when str() is called on them they
   already format themselves using the right precision and scale.)
diff --git a/psycopg/pqpath.c b/psycopg/pqpath.c
index 02923351..c9271871 100644
--- a/psycopg/pqpath.c
+++ b/psycopg/pqpath.c
@@ -47,6 +47,7 @@ void
 pq_raise(connectionObject *conn, cursorObject *curs, PyObject *exc, char *msg)
 {
     char *err = NULL;
+    char *code = NULL;
     
     if ((conn == NULL && curs == NULL) || (curs != NULL && conn == NULL)) {
         PyErr_SetString(Error,
@@ -54,8 +55,14 @@ pq_raise(connectionObject *conn, cursorObject *curs, PyObject *exc, char *msg)
         return;
     }
     
-    if (curs && curs->pgres)
+    if (curs && curs->pgres) {
         err = PQresultErrorMessage(curs->pgres);
+#ifdef HAVE_PQPROTOCOL3
+        if (err != NULL && conn->protocol == 3) {
+            code = PQresultErrorField(curs->pgres, PG_DIAG_SQLSTATE);
+        }
+#endif
+    }
     if (err == NULL)
         err = PQerrorMessage(conn->pgconn);
 
@@ -100,7 +107,10 @@ pq_raise(connectionObject *conn, cursorObject *curs, PyObject *exc, char *msg)
     if (err && strlen(err) > 8) err = &(err[8]);
     
     /* if msg is not NULL, add it to the error message, after a '\n' */
-    if (msg) {
+    if (msg && code) {
+        PyErr_Format(exc, "[%s] %s\n%s", code, err, msg);
+    }
+    else if (msg) {
         PyErr_Format(exc, "%s\n%s", err, msg);
     }
     else {