Make some methods raise an exception in asynchronous mode

The methods changed are connection.commit(), rollback(), reset(),
set_isolation_level(), set_client_encoding(), lobject(), cursor(str)
as well as cursor.execute() and cursor.callproc() if another query is
in progress and cursor.executemany(), cursor.copy_{from,to,expert)().

1 files changed, 11 insertions, 0 deletions

diff --git a/psycopg/cursor.h b/psycopg/cursor.h
index bcfc81f..a2293a0 100644
--- a/psycopg/cursor.h
+++ b/psycopg/cursor.h
@@ -106,6 +106,17 @@ if ((self)->mark != (self)->conn->mark) {                                  \
     PyErr_SetString(ProgrammingError, "named cursor isn't valid anymore"); \
     return NULL; }
 
+#define EXC_IF_CURS_ASYNC(self, cmd) if ((self)->conn->async == 1) { \
+    PyErr_SetString(ProgrammingError, #cmd " cannot be used "        \
+    "in asynchronous mode");                                         \
+    return NULL; }
+
+#define EXC_IF_ASYNC_IN_PROGRESS(self, cmd) \
+if ((self)->conn->async_cursor != NULL) {   \
+    PyErr_SetString(ProgrammingError, #cmd " cannot be used "        \
+    "while an asynchronous query is underway");                      \
+    return NULL; }
+
 #ifdef __cplusplus
 }
 #endif