summaryrefslogtreecommitdiff
path: root/psycopg/psycopgmodule.c
diff options
context:
space:
mode:
Diffstat (limited to 'psycopg/psycopgmodule.c')
-rw-r--r--psycopg/psycopgmodule.c6
1 files changed, 6 insertions, 0 deletions
diff --git a/psycopg/psycopgmodule.c b/psycopg/psycopgmodule.c
index 528c592..60c9c04 100644
--- a/psycopg/psycopgmodule.c
+++ b/psycopg/psycopgmodule.c
@@ -32,6 +32,7 @@
#include "psycopg/psycopg.h"
#include "psycopg/connection.h"
#include "psycopg/cursor.h"
+#include "psycopg/green.h"
#include "psycopg/lobject.h"
#include "psycopg/typecast.h"
#include "psycopg/microprotocols.h"
@@ -693,6 +694,11 @@ static PyMethodDef psycopgMethods[] = {
METH_VARARGS, psyco_IntervalFromMx_doc},
#endif
+#ifdef PSYCOPG_EXTENSIONS
+ {"set_wait_callback", (PyCFunction)psyco_set_wait_callback,
+ METH_O, psyco_set_wait_callback_doc},
+#endif
+
{NULL, NULL, 0, NULL} /* Sentinel */
};