blob: e78e1d1f4b2c6a233e8bfaf076c8ed0ffcfadb8c [file] [log] [blame]
Fred Drakec37b65e2001-11-28 07:26:15 +00001\chapter{Extending Python with C or \Cpp \label{intro}}
Fred Drakecc8f44b2001-08-20 19:30:29 +00002
3
4It is quite easy to add new built-in modules to Python, if you know
5how to program in C. Such \dfn{extension modules} can do two things
6that can't be done directly in Python: they can implement new built-in
7object types, and they can call C library functions and system calls.
8
9To support extensions, the Python API (Application Programmers
10Interface) defines a set of functions, macros and variables that
11provide access to most aspects of the Python run-time system. The
12Python API is incorporated in a C source file by including the header
13\code{"Python.h"}.
14
15The compilation of an extension module depends on its intended use as
16well as on your system setup; details are given in later chapters.
17
18
19\section{A Simple Example
20 \label{simpleExample}}
21
22Let's create an extension module called \samp{spam} (the favorite food
23of Monty Python fans...) and let's say we want to create a Python
24interface to the C library function \cfunction{system()}.\footnote{An
25interface for this function already exists in the standard module
26\module{os} --- it was chosen as a simple and straightfoward example.}
27This function takes a null-terminated character string as argument and
28returns an integer. We want this function to be callable from Python
29as follows:
30
31\begin{verbatim}
32>>> import spam
33>>> status = spam.system("ls -l")
34\end{verbatim}
35
36Begin by creating a file \file{spammodule.c}. (Historically, if a
37module is called \samp{spam}, the C file containing its implementation
38is called \file{spammodule.c}; if the module name is very long, like
39\samp{spammify}, the module name can be just \file{spammify.c}.)
40
41The first line of our file can be:
42
43\begin{verbatim}
44#include <Python.h>
45\end{verbatim}
46
47which pulls in the Python API (you can add a comment describing the
48purpose of the module and a copyright notice if you like).
Fred Drake396ca572001-09-06 16:30:30 +000049Since Python may define some pre-processor definitions which affect
50the standard headers on some systems, you must include \file{Python.h}
51before any standard headers are included.
Fred Drakecc8f44b2001-08-20 19:30:29 +000052
Fred Drake396ca572001-09-06 16:30:30 +000053All user-visible symbols defined by \file{Python.h} have a prefix of
Fred Drakecc8f44b2001-08-20 19:30:29 +000054\samp{Py} or \samp{PY}, except those defined in standard header files.
55For convenience, and since they are used extensively by the Python
56interpreter, \code{"Python.h"} includes a few standard header files:
57\code{<stdio.h>}, \code{<string.h>}, \code{<errno.h>}, and
58\code{<stdlib.h>}. If the latter header file does not exist on your
59system, it declares the functions \cfunction{malloc()},
60\cfunction{free()} and \cfunction{realloc()} directly.
61
62The next thing we add to our module file is the C function that will
63be called when the Python expression \samp{spam.system(\var{string})}
64is evaluated (we'll see shortly how it ends up being called):
65
66\begin{verbatim}
67static PyObject *
68spam_system(self, args)
69 PyObject *self;
70 PyObject *args;
71{
72 char *command;
73 int sts;
74
75 if (!PyArg_ParseTuple(args, "s", &command))
76 return NULL;
77 sts = system(command);
78 return Py_BuildValue("i", sts);
79}
80\end{verbatim}
81
82There is a straightforward translation from the argument list in
83Python (for example, the single expression \code{"ls -l"}) to the
84arguments passed to the C function. The C function always has two
85arguments, conventionally named \var{self} and \var{args}.
86
87The \var{self} argument is only used when the C function implements a
88built-in method, not a function. In the example, \var{self} will
89always be a \NULL{} pointer, since we are defining a function, not a
90method. (This is done so that the interpreter doesn't have to
91understand two different types of C functions.)
92
93The \var{args} argument will be a pointer to a Python tuple object
94containing the arguments. Each item of the tuple corresponds to an
95argument in the call's argument list. The arguments are Python
96objects --- in order to do anything with them in our C function we have
97to convert them to C values. The function \cfunction{PyArg_ParseTuple()}
98in the Python API checks the argument types and converts them to C
99values. It uses a template string to determine the required types of
100the arguments as well as the types of the C variables into which to
101store the converted values. More about this later.
102
103\cfunction{PyArg_ParseTuple()} returns true (nonzero) if all arguments have
104the right type and its components have been stored in the variables
105whose addresses are passed. It returns false (zero) if an invalid
106argument list was passed. In the latter case it also raises an
107appropriate exception so the calling function can return
108\NULL{} immediately (as we saw in the example).
109
110
111\section{Intermezzo: Errors and Exceptions
112 \label{errors}}
113
114An important convention throughout the Python interpreter is the
115following: when a function fails, it should set an exception condition
116and return an error value (usually a \NULL{} pointer). Exceptions
117are stored in a static global variable inside the interpreter; if this
118variable is \NULL{} no exception has occurred. A second global
119variable stores the ``associated value'' of the exception (the second
120argument to \keyword{raise}). A third variable contains the stack
121traceback in case the error originated in Python code. These three
122variables are the C equivalents of the Python variables
123\code{sys.exc_type}, \code{sys.exc_value} and \code{sys.exc_traceback} (see
124the section on module \module{sys} in the
125\citetitle[../lib/lib.html]{Python Library Reference}). It is
126important to know about them to understand how errors are passed
127around.
128
129The Python API defines a number of functions to set various types of
130exceptions.
131
132The most common one is \cfunction{PyErr_SetString()}. Its arguments
133are an exception object and a C string. The exception object is
134usually a predefined object like \cdata{PyExc_ZeroDivisionError}. The
135C string indicates the cause of the error and is converted to a
136Python string object and stored as the ``associated value'' of the
137exception.
138
139Another useful function is \cfunction{PyErr_SetFromErrno()}, which only
140takes an exception argument and constructs the associated value by
141inspection of the global variable \cdata{errno}. The most
142general function is \cfunction{PyErr_SetObject()}, which takes two object
143arguments, the exception and its associated value. You don't need to
144\cfunction{Py_INCREF()} the objects passed to any of these functions.
145
146You can test non-destructively whether an exception has been set with
147\cfunction{PyErr_Occurred()}. This returns the current exception object,
148or \NULL{} if no exception has occurred. You normally don't need
149to call \cfunction{PyErr_Occurred()} to see whether an error occurred in a
150function call, since you should be able to tell from the return value.
151
152When a function \var{f} that calls another function \var{g} detects
153that the latter fails, \var{f} should itself return an error value
154(usually \NULL{} or \code{-1}). It should \emph{not} call one of the
155\cfunction{PyErr_*()} functions --- one has already been called by \var{g}.
156\var{f}'s caller is then supposed to also return an error indication
157to \emph{its} caller, again \emph{without} calling \cfunction{PyErr_*()},
158and so on --- the most detailed cause of the error was already
159reported by the function that first detected it. Once the error
160reaches the Python interpreter's main loop, this aborts the currently
161executing Python code and tries to find an exception handler specified
162by the Python programmer.
163
164(There are situations where a module can actually give a more detailed
165error message by calling another \cfunction{PyErr_*()} function, and in
166such cases it is fine to do so. As a general rule, however, this is
167not necessary, and can cause information about the cause of the error
168to be lost: most operations can fail for a variety of reasons.)
169
170To ignore an exception set by a function call that failed, the exception
171condition must be cleared explicitly by calling \cfunction{PyErr_Clear()}.
172The only time C code should call \cfunction{PyErr_Clear()} is if it doesn't
173want to pass the error on to the interpreter but wants to handle it
174completely by itself (possibly by trying something else, or pretending
175nothing went wrong).
176
177Every failing \cfunction{malloc()} call must be turned into an
178exception --- the direct caller of \cfunction{malloc()} (or
179\cfunction{realloc()}) must call \cfunction{PyErr_NoMemory()} and
180return a failure indicator itself. All the object-creating functions
181(for example, \cfunction{PyInt_FromLong()}) already do this, so this
182note is only relevant to those who call \cfunction{malloc()} directly.
183
184Also note that, with the important exception of
185\cfunction{PyArg_ParseTuple()} and friends, functions that return an
186integer status usually return a positive value or zero for success and
187\code{-1} for failure, like \UNIX{} system calls.
188
189Finally, be careful to clean up garbage (by making
190\cfunction{Py_XDECREF()} or \cfunction{Py_DECREF()} calls for objects
191you have already created) when you return an error indicator!
192
193The choice of which exception to raise is entirely yours. There are
194predeclared C objects corresponding to all built-in Python exceptions,
195such as \cdata{PyExc_ZeroDivisionError}, which you can use directly.
196Of course, you should choose exceptions wisely --- don't use
197\cdata{PyExc_TypeError} to mean that a file couldn't be opened (that
198should probably be \cdata{PyExc_IOError}). If something's wrong with
199the argument list, the \cfunction{PyArg_ParseTuple()} function usually
200raises \cdata{PyExc_TypeError}. If you have an argument whose value
201must be in a particular range or must satisfy other conditions,
202\cdata{PyExc_ValueError} is appropriate.
203
204You can also define a new exception that is unique to your module.
205For this, you usually declare a static object variable at the
206beginning of your file:
207
208\begin{verbatim}
209static PyObject *SpamError;
210\end{verbatim}
211
212and initialize it in your module's initialization function
213(\cfunction{initspam()}) with an exception object (leaving out
214the error checking for now):
215
216\begin{verbatim}
217void
Fred Drakeef6373a2001-11-17 06:50:42 +0000218initspam(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +0000219{
220 PyObject *m, *d;
221
222 m = Py_InitModule("spam", SpamMethods);
223 d = PyModule_GetDict(m);
224 SpamError = PyErr_NewException("spam.error", NULL, NULL);
225 PyDict_SetItemString(d, "error", SpamError);
226}
227\end{verbatim}
228
229Note that the Python name for the exception object is
230\exception{spam.error}. The \cfunction{PyErr_NewException()} function
231may create a class with the base class being \exception{Exception}
232(unless another class is passed in instead of \NULL), described in the
233\citetitle[../lib/lib.html]{Python Library Reference} under ``Built-in
234Exceptions.''
235
236Note also that the \cdata{SpamError} variable retains a reference to
237the newly created exception class; this is intentional! Since the
238exception could be removed from the module by external code, an owned
239reference to the class is needed to ensure that it will not be
240discarded, causing \cdata{SpamError} to become a dangling pointer.
241Should it become a dangling pointer, C code which raises the exception
242could cause a core dump or other unintended side effects.
243
244
245\section{Back to the Example
246 \label{backToExample}}
247
248Going back to our example function, you should now be able to
249understand this statement:
250
251\begin{verbatim}
252 if (!PyArg_ParseTuple(args, "s", &command))
253 return NULL;
254\end{verbatim}
255
256It returns \NULL{} (the error indicator for functions returning
257object pointers) if an error is detected in the argument list, relying
258on the exception set by \cfunction{PyArg_ParseTuple()}. Otherwise the
259string value of the argument has been copied to the local variable
260\cdata{command}. This is a pointer assignment and you are not supposed
261to modify the string to which it points (so in Standard C, the variable
262\cdata{command} should properly be declared as \samp{const char
263*command}).
264
265The next statement is a call to the \UNIX{} function
266\cfunction{system()}, passing it the string we just got from
267\cfunction{PyArg_ParseTuple()}:
268
269\begin{verbatim}
270 sts = system(command);
271\end{verbatim}
272
273Our \function{spam.system()} function must return the value of
274\cdata{sts} as a Python object. This is done using the function
275\cfunction{Py_BuildValue()}, which is something like the inverse of
276\cfunction{PyArg_ParseTuple()}: it takes a format string and an
277arbitrary number of C values, and returns a new Python object.
278More info on \cfunction{Py_BuildValue()} is given later.
279
280\begin{verbatim}
281 return Py_BuildValue("i", sts);
282\end{verbatim}
283
284In this case, it will return an integer object. (Yes, even integers
285are objects on the heap in Python!)
286
287If you have a C function that returns no useful argument (a function
288returning \ctype{void}), the corresponding Python function must return
289\code{None}. You need this idiom to do so:
290
291\begin{verbatim}
292 Py_INCREF(Py_None);
293 return Py_None;
294\end{verbatim}
295
296\cdata{Py_None} is the C name for the special Python object
297\code{None}. It is a genuine Python object rather than a \NULL{}
298pointer, which means ``error'' in most contexts, as we have seen.
299
300
301\section{The Module's Method Table and Initialization Function
302 \label{methodTable}}
303
304I promised to show how \cfunction{spam_system()} is called from Python
305programs. First, we need to list its name and address in a ``method
306table'':
307
308\begin{verbatim}
309static PyMethodDef SpamMethods[] = {
310 ...
Fred Drakeef6373a2001-11-17 06:50:42 +0000311 {"system", spam_system, METH_VARARGS,
312 "Execute a shell command."},
Fred Drakecc8f44b2001-08-20 19:30:29 +0000313 ...
Fred Drakeef6373a2001-11-17 06:50:42 +0000314 {NULL, NULL, 0, NULL} /* Sentinel */
Fred Drakecc8f44b2001-08-20 19:30:29 +0000315};
316\end{verbatim}
317
318Note the third entry (\samp{METH_VARARGS}). This is a flag telling
319the interpreter the calling convention to be used for the C
320function. It should normally always be \samp{METH_VARARGS} or
321\samp{METH_VARARGS | METH_KEYWORDS}; a value of \code{0} means that an
322obsolete variant of \cfunction{PyArg_ParseTuple()} is used.
323
324When using only \samp{METH_VARARGS}, the function should expect
325the Python-level parameters to be passed in as a tuple acceptable for
326parsing via \cfunction{PyArg_ParseTuple()}; more information on this
327function is provided below.
328
329The \constant{METH_KEYWORDS} bit may be set in the third field if
330keyword arguments should be passed to the function. In this case, the
331C function should accept a third \samp{PyObject *} parameter which
332will be a dictionary of keywords. Use
333\cfunction{PyArg_ParseTupleAndKeywords()} to parse the arguments to
334such a function.
335
336The method table must be passed to the interpreter in the module's
337initialization function. The initialization function must be named
338\cfunction{init\var{name}()}, where \var{name} is the name of the
339module, and should be the only non-\keyword{static} item defined in
340the module file:
341
342\begin{verbatim}
343void
Fred Drakeef6373a2001-11-17 06:50:42 +0000344initspam(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +0000345{
346 (void) Py_InitModule("spam", SpamMethods);
347}
348\end{verbatim}
349
350Note that for \Cpp, this method must be declared \code{extern "C"}.
351
352When the Python program imports module \module{spam} for the first
353time, \cfunction{initspam()} is called. (See below for comments about
354embedding Python.) It calls
355\cfunction{Py_InitModule()}, which creates a ``module object'' (which
356is inserted in the dictionary \code{sys.modules} under the key
357\code{"spam"}), and inserts built-in function objects into the newly
358created module based upon the table (an array of \ctype{PyMethodDef}
359structures) that was passed as its second argument.
360\cfunction{Py_InitModule()} returns a pointer to the module object
361that it creates (which is unused here). It aborts with a fatal error
362if the module could not be initialized satisfactorily, so the caller
363doesn't need to check for errors.
364
365When embedding Python, the \cfunction{initspam()} function is not
366called automatically unless there's an entry in the
367\cdata{_PyImport_Inittab} table. The easiest way to handle this is to
368statically initialize your statically-linked modules by directly
369calling \cfunction{initspam()} after the call to
370\cfunction{Py_Initialize()} or \cfunction{PyMac_Initialize()}:
371
372\begin{verbatim}
373int main(int argc, char **argv)
374{
375 /* Pass argv[0] to the Python interpreter */
376 Py_SetProgramName(argv[0]);
377
378 /* Initialize the Python interpreter. Required. */
379 Py_Initialize();
380
381 /* Add a static module */
382 initspam();
383\end{verbatim}
384
385An example may be found in the file \file{Demo/embed/demo.c} in the
386Python source distribution.
387
Fred Drake0aa811c2001-10-20 04:24:09 +0000388\note{Removing entries from \code{sys.modules} or importing
Fred Drakecc8f44b2001-08-20 19:30:29 +0000389compiled modules into multiple interpreters within a process (or
390following a \cfunction{fork()} without an intervening
391\cfunction{exec()}) can create problems for some extension modules.
392Extension module authors should exercise caution when initializing
393internal data structures.
394Note also that the \function{reload()} function can be used with
395extension modules, and will call the module initialization function
396(\cfunction{initspam()} in the example), but will not load the module
397again if it was loaded from a dynamically loadable object file
Fred Drake0aa811c2001-10-20 04:24:09 +0000398(\file{.so} on \UNIX, \file{.dll} on Windows).}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000399
400A more substantial example module is included in the Python source
401distribution as \file{Modules/xxmodule.c}. This file may be used as a
402template or simply read as an example. The \program{modulator.py}
403script included in the source distribution or Windows install provides
404a simple graphical user interface for declaring the functions and
405objects which a module should implement, and can generate a template
406which can be filled in. The script lives in the
407\file{Tools/modulator/} directory; see the \file{README} file there
408for more information.
409
410
411\section{Compilation and Linkage
412 \label{compilation}}
413
414There are two more things to do before you can use your new extension:
415compiling and linking it with the Python system. If you use dynamic
416loading, the details depend on the style of dynamic loading your
417system uses; see the chapters about building extension modules on
418\UNIX{} (chapter \ref{building-on-unix}) and Windows (chapter
419\ref{building-on-windows}) for more information about this.
420% XXX Add information about MacOS
421
422If you can't use dynamic loading, or if you want to make your module a
423permanent part of the Python interpreter, you will have to change the
424configuration setup and rebuild the interpreter. Luckily, this is
425very simple: just place your file (\file{spammodule.c} for example) in
426the \file{Modules/} directory of an unpacked source distribution, add
427a line to the file \file{Modules/Setup.local} describing your file:
428
429\begin{verbatim}
430spam spammodule.o
431\end{verbatim}
432
433and rebuild the interpreter by running \program{make} in the toplevel
434directory. You can also run \program{make} in the \file{Modules/}
435subdirectory, but then you must first rebuild \file{Makefile}
436there by running `\program{make} Makefile'. (This is necessary each
437time you change the \file{Setup} file.)
438
439If your module requires additional libraries to link with, these can
440be listed on the line in the configuration file as well, for instance:
441
442\begin{verbatim}
443spam spammodule.o -lX11
444\end{verbatim}
445
446\section{Calling Python Functions from C
447 \label{callingPython}}
448
449So far we have concentrated on making C functions callable from
450Python. The reverse is also useful: calling Python functions from C.
451This is especially the case for libraries that support so-called
452``callback'' functions. If a C interface makes use of callbacks, the
453equivalent Python often needs to provide a callback mechanism to the
454Python programmer; the implementation will require calling the Python
455callback functions from a C callback. Other uses are also imaginable.
456
457Fortunately, the Python interpreter is easily called recursively, and
458there is a standard interface to call a Python function. (I won't
459dwell on how to call the Python parser with a particular string as
460input --- if you're interested, have a look at the implementation of
461the \programopt{-c} command line option in \file{Python/pythonmain.c}
462from the Python source code.)
463
464Calling a Python function is easy. First, the Python program must
465somehow pass you the Python function object. You should provide a
466function (or some other interface) to do this. When this function is
467called, save a pointer to the Python function object (be careful to
468\cfunction{Py_INCREF()} it!) in a global variable --- or wherever you
469see fit. For example, the following function might be part of a module
470definition:
471
472\begin{verbatim}
473static PyObject *my_callback = NULL;
474
475static PyObject *
476my_set_callback(dummy, args)
477 PyObject *dummy, *args;
478{
479 PyObject *result = NULL;
480 PyObject *temp;
481
482 if (PyArg_ParseTuple(args, "O:set_callback", &temp)) {
483 if (!PyCallable_Check(temp)) {
484 PyErr_SetString(PyExc_TypeError, "parameter must be callable");
485 return NULL;
486 }
487 Py_XINCREF(temp); /* Add a reference to new callback */
488 Py_XDECREF(my_callback); /* Dispose of previous callback */
489 my_callback = temp; /* Remember new callback */
490 /* Boilerplate to return "None" */
491 Py_INCREF(Py_None);
492 result = Py_None;
493 }
494 return result;
495}
496\end{verbatim}
497
498This function must be registered with the interpreter using the
499\constant{METH_VARARGS} flag; this is described in section
500\ref{methodTable}, ``The Module's Method Table and Initialization
501Function.'' The \cfunction{PyArg_ParseTuple()} function and its
Fred Drakec37b65e2001-11-28 07:26:15 +0000502arguments are documented in section~\ref{parseTuple}, ``Extracting
Fred Drakecc8f44b2001-08-20 19:30:29 +0000503Parameters in Extension Functions.''
504
505The macros \cfunction{Py_XINCREF()} and \cfunction{Py_XDECREF()}
506increment/decrement the reference count of an object and are safe in
507the presence of \NULL{} pointers (but note that \var{temp} will not be
Fred Drakec37b65e2001-11-28 07:26:15 +0000508\NULL{} in this context). More info on them in
509section~\ref{refcounts}, ``Reference Counts.''
Fred Drakecc8f44b2001-08-20 19:30:29 +0000510
511Later, when it is time to call the function, you call the C function
Fred Drake99181ac2001-11-29 05:02:34 +0000512\cfunction{PyEval_CallObject()}.\ttindex{PyEval_CallObject()} This
513function has two arguments, both pointers to arbitrary Python objects:
514the Python function, and the argument list. The argument list must
515always be a tuple object, whose length is the number of arguments. To
516call the Python function with no arguments, pass an empty tuple; to
517call it with one argument, pass a singleton tuple.
518\cfunction{Py_BuildValue()} returns a tuple when its format string
519consists of zero or more format codes between parentheses. For
520example:
Fred Drakecc8f44b2001-08-20 19:30:29 +0000521
522\begin{verbatim}
523 int arg;
524 PyObject *arglist;
525 PyObject *result;
526 ...
527 arg = 123;
528 ...
529 /* Time to call the callback */
530 arglist = Py_BuildValue("(i)", arg);
531 result = PyEval_CallObject(my_callback, arglist);
532 Py_DECREF(arglist);
533\end{verbatim}
534
535\cfunction{PyEval_CallObject()} returns a Python object pointer: this is
536the return value of the Python function. \cfunction{PyEval_CallObject()} is
537``reference-count-neutral'' with respect to its arguments. In the
538example a new tuple was created to serve as the argument list, which
539is \cfunction{Py_DECREF()}-ed immediately after the call.
540
541The return value of \cfunction{PyEval_CallObject()} is ``new'': either it
542is a brand new object, or it is an existing object whose reference
543count has been incremented. So, unless you want to save it in a
544global variable, you should somehow \cfunction{Py_DECREF()} the result,
545even (especially!) if you are not interested in its value.
546
547Before you do this, however, it is important to check that the return
Fred Drakec37b65e2001-11-28 07:26:15 +0000548value isn't \NULL. If it is, the Python function terminated by
Fred Drakecc8f44b2001-08-20 19:30:29 +0000549raising an exception. If the C code that called
550\cfunction{PyEval_CallObject()} is called from Python, it should now
551return an error indication to its Python caller, so the interpreter
552can print a stack trace, or the calling Python code can handle the
553exception. If this is not possible or desirable, the exception should
554be cleared by calling \cfunction{PyErr_Clear()}. For example:
555
556\begin{verbatim}
557 if (result == NULL)
558 return NULL; /* Pass error back */
559 ...use result...
560 Py_DECREF(result);
561\end{verbatim}
562
563Depending on the desired interface to the Python callback function,
564you may also have to provide an argument list to
565\cfunction{PyEval_CallObject()}. In some cases the argument list is
566also provided by the Python program, through the same interface that
567specified the callback function. It can then be saved and used in the
568same manner as the function object. In other cases, you may have to
569construct a new tuple to pass as the argument list. The simplest way
570to do this is to call \cfunction{Py_BuildValue()}. For example, if
571you want to pass an integral event code, you might use the following
572code:
573
574\begin{verbatim}
575 PyObject *arglist;
576 ...
577 arglist = Py_BuildValue("(l)", eventcode);
578 result = PyEval_CallObject(my_callback, arglist);
579 Py_DECREF(arglist);
580 if (result == NULL)
581 return NULL; /* Pass error back */
582 /* Here maybe use the result */
583 Py_DECREF(result);
584\end{verbatim}
585
586Note the placement of \samp{Py_DECREF(arglist)} immediately after the
587call, before the error check! Also note that strictly spoken this
588code is not complete: \cfunction{Py_BuildValue()} may run out of
589memory, and this should be checked.
590
591
592\section{Extracting Parameters in Extension Functions
593 \label{parseTuple}}
594
595The \cfunction{PyArg_ParseTuple()} function is declared as follows:
596
597\begin{verbatim}
598int PyArg_ParseTuple(PyObject *arg, char *format, ...);
599\end{verbatim}
600
601The \var{arg} argument must be a tuple object containing an argument
602list passed from Python to a C function. The \var{format} argument
603must be a format string, whose syntax is explained below. The
604remaining arguments must be addresses of variables whose type is
605determined by the format string. For the conversion to succeed, the
606\var{arg} object must match the format and the format must be
607exhausted. On success, \cfunction{PyArg_ParseTuple()} returns true,
608otherwise it returns false and raises an appropriate exception.
609
610Note that while \cfunction{PyArg_ParseTuple()} checks that the Python
611arguments have the required types, it cannot check the validity of the
612addresses of C variables passed to the call: if you make mistakes
613there, your code will probably crash or at least overwrite random bits
614in memory. So be careful!
615
616A format string consists of zero or more ``format units''. A format
617unit describes one Python object; it is usually a single character or
618a parenthesized sequence of format units. With a few exceptions, a
619format unit that is not a parenthesized sequence normally corresponds
620to a single address argument to \cfunction{PyArg_ParseTuple()}. In the
621following description, the quoted form is the format unit; the entry
622in (round) parentheses is the Python object type that matches the
623format unit; and the entry in [square] brackets is the type of the C
624variable(s) whose address should be passed. (Use the \samp{\&}
625operator to pass a variable's address.)
626
627Note that any Python object references which are provided to the
628caller are \emph{borrowed} references; do not decrement their
629reference count!
630
631\begin{description}
632
633\item[\samp{s} (string or Unicode object) {[char *]}]
634Convert a Python string or Unicode object to a C pointer to a
635character string. You must not provide storage for the string
636itself; a pointer to an existing string is stored into the character
637pointer variable whose address you pass. The C string is
638null-terminated. The Python string must not contain embedded null
639bytes; if it does, a \exception{TypeError} exception is raised.
640Unicode objects are converted to C strings using the default
641encoding. If this conversion fails, an \exception{UnicodeError} is
642raised.
643
644\item[\samp{s\#} (string, Unicode or any read buffer compatible object)
645{[char *, int]}]
646This variant on \samp{s} stores into two C variables, the first one a
647pointer to a character string, the second one its length. In this
648case the Python string may contain embedded null bytes. Unicode
649objects pass back a pointer to the default encoded string version of the
650object if such a conversion is possible. All other read buffer
651compatible objects pass back a reference to the raw internal data
652representation.
653
654\item[\samp{z} (string or \code{None}) {[char *]}]
655Like \samp{s}, but the Python object may also be \code{None}, in which
Fred Drakec37b65e2001-11-28 07:26:15 +0000656case the C pointer is set to \NULL.
Fred Drakecc8f44b2001-08-20 19:30:29 +0000657
658\item[\samp{z\#} (string or \code{None} or any read buffer compatible object)
659{[char *, int]}]
660This is to \samp{s\#} as \samp{z} is to \samp{s}.
661
662\item[\samp{u} (Unicode object) {[Py_UNICODE *]}]
663Convert a Python Unicode object to a C pointer to a null-terminated
664buffer of 16-bit Unicode (UTF-16) data. As with \samp{s}, there is no need
665to provide storage for the Unicode data buffer; a pointer to the
666existing Unicode data is stored into the Py_UNICODE pointer variable whose
667address you pass.
668
669\item[\samp{u\#} (Unicode object) {[Py_UNICODE *, int]}]
670This variant on \samp{u} stores into two C variables, the first one
671a pointer to a Unicode data buffer, the second one its length.
672
673\item[\samp{es} (string, Unicode object or character buffer compatible
674object) {[const char *encoding, char **buffer]}]
675This variant on \samp{s} is used for encoding Unicode and objects
676convertible to Unicode into a character buffer. It only works for
677encoded data without embedded \NULL{} bytes.
678
679The variant reads one C variable and stores into two C variables, the
680first one a pointer to an encoding name string (\var{encoding}), and the
681second a pointer to a pointer to a character buffer (\var{**buffer},
682the buffer used for storing the encoded data).
683
Fred Drakec37b65e2001-11-28 07:26:15 +0000684The encoding name must map to a registered codec. If set to \NULL,
Fred Drakecc8f44b2001-08-20 19:30:29 +0000685the default encoding is used.
686
687\cfunction{PyArg_ParseTuple()} will allocate a buffer of the needed
688size using \cfunction{PyMem_NEW()}, copy the encoded data into this
689buffer and adjust \var{*buffer} to reference the newly allocated
690storage. The caller is responsible for calling
691\cfunction{PyMem_Free()} to free the allocated buffer after usage.
692
693\item[\samp{et} (string, Unicode object or character buffer compatible
694object) {[const char *encoding, char **buffer]}]
695Same as \samp{es} except that string objects are passed through without
696recoding them. Instead, the implementation assumes that the string
697object uses the encoding passed in as parameter.
698
699\item[\samp{es\#} (string, Unicode object or character buffer compatible
700object) {[const char *encoding, char **buffer, int *buffer_length]}]
701This variant on \samp{s\#} is used for encoding Unicode and objects
702convertible to Unicode into a character buffer. It reads one C
703variable and stores into three C variables, the first one a pointer to
704an encoding name string (\var{encoding}), the second a pointer to a
705pointer to a character buffer (\var{**buffer}, the buffer used for
706storing the encoded data) and the third one a pointer to an integer
707(\var{*buffer_length}, the buffer length).
708
Fred Drakec37b65e2001-11-28 07:26:15 +0000709The encoding name must map to a registered codec. If set to \NULL,
Fred Drakecc8f44b2001-08-20 19:30:29 +0000710the default encoding is used.
711
712There are two modes of operation:
713
714If \var{*buffer} points a \NULL{} pointer,
715\cfunction{PyArg_ParseTuple()} will allocate a buffer of the needed
716size using \cfunction{PyMem_NEW()}, copy the encoded data into this
717buffer and adjust \var{*buffer} to reference the newly allocated
718storage. The caller is responsible for calling
719\cfunction{PyMem_Free()} to free the allocated buffer after usage.
720
721If \var{*buffer} points to a non-\NULL{} pointer (an already allocated
722buffer), \cfunction{PyArg_ParseTuple()} will use this location as
723buffer and interpret \var{*buffer_length} as buffer size. It will then
724copy the encoded data into the buffer and 0-terminate it. Buffer
725overflow is signalled with an exception.
726
727In both cases, \var{*buffer_length} is set to the length of the
728encoded data without the trailing 0-byte.
729
730\item[\samp{et\#} (string, Unicode object or character buffer compatible
731object) {[const char *encoding, char **buffer]}]
732Same as \samp{es\#} except that string objects are passed through without
733recoding them. Instead, the implementation assumes that the string
734object uses the encoding passed in as parameter.
735
736\item[\samp{b} (integer) {[char]}]
737Convert a Python integer to a tiny int, stored in a C \ctype{char}.
738
739\item[\samp{h} (integer) {[short int]}]
740Convert a Python integer to a C \ctype{short int}.
741
742\item[\samp{i} (integer) {[int]}]
743Convert a Python integer to a plain C \ctype{int}.
744
745\item[\samp{l} (integer) {[long int]}]
746Convert a Python integer to a C \ctype{long int}.
747
Tim Petersd38b1c72001-09-30 05:09:37 +0000748\item[\samp{L} (integer) {[LONG_LONG]}]
749Convert a Python integer to a C \ctype{long long}. This format is only
750available on platforms that support \ctype{long long} (or \ctype{_int64}
751on Windows).
752
Fred Drakecc8f44b2001-08-20 19:30:29 +0000753\item[\samp{c} (string of length 1) {[char]}]
754Convert a Python character, represented as a string of length 1, to a
755C \ctype{char}.
756
757\item[\samp{f} (float) {[float]}]
758Convert a Python floating point number to a C \ctype{float}.
759
760\item[\samp{d} (float) {[double]}]
761Convert a Python floating point number to a C \ctype{double}.
762
763\item[\samp{D} (complex) {[Py_complex]}]
764Convert a Python complex number to a C \ctype{Py_complex} structure.
765
766\item[\samp{O} (object) {[PyObject *]}]
767Store a Python object (without any conversion) in a C object pointer.
768The C program thus receives the actual object that was passed. The
769object's reference count is not increased. The pointer stored is not
Fred Drakec37b65e2001-11-28 07:26:15 +0000770\NULL.
Fred Drakecc8f44b2001-08-20 19:30:29 +0000771
772\item[\samp{O!} (object) {[\var{typeobject}, PyObject *]}]
773Store a Python object in a C object pointer. This is similar to
774\samp{O}, but takes two C arguments: the first is the address of a
775Python type object, the second is the address of the C variable (of
776type \ctype{PyObject *}) into which the object pointer is stored.
777If the Python object does not have the required type,
778\exception{TypeError} is raised.
779
780\item[\samp{O\&} (object) {[\var{converter}, \var{anything}]}]
781Convert a Python object to a C variable through a \var{converter}
782function. This takes two arguments: the first is a function, the
783second is the address of a C variable (of arbitrary type), converted
784to \ctype{void *}. The \var{converter} function in turn is called as
785follows:
786
787\var{status}\code{ = }\var{converter}\code{(}\var{object}, \var{address}\code{);}
788
789where \var{object} is the Python object to be converted and
790\var{address} is the \ctype{void *} argument that was passed to
791\cfunction{PyArg_ConvertTuple()}. The returned \var{status} should be
792\code{1} for a successful conversion and \code{0} if the conversion
793has failed. When the conversion fails, the \var{converter} function
794should raise an exception.
795
796\item[\samp{S} (string) {[PyStringObject *]}]
797Like \samp{O} but requires that the Python object is a string object.
798Raises \exception{TypeError} if the object is not a string object.
799The C variable may also be declared as \ctype{PyObject *}.
800
801\item[\samp{U} (Unicode string) {[PyUnicodeObject *]}]
802Like \samp{O} but requires that the Python object is a Unicode object.
803Raises \exception{TypeError} if the object is not a Unicode object.
804The C variable may also be declared as \ctype{PyObject *}.
805
806\item[\samp{t\#} (read-only character buffer) {[char *, int]}]
807Like \samp{s\#}, but accepts any object which implements the read-only
808buffer interface. The \ctype{char *} variable is set to point to the
809first byte of the buffer, and the \ctype{int} is set to the length of
810the buffer. Only single-segment buffer objects are accepted;
811\exception{TypeError} is raised for all others.
812
813\item[\samp{w} (read-write character buffer) {[char *]}]
814Similar to \samp{s}, but accepts any object which implements the
815read-write buffer interface. The caller must determine the length of
816the buffer by other means, or use \samp{w\#} instead. Only
817single-segment buffer objects are accepted; \exception{TypeError} is
818raised for all others.
819
820\item[\samp{w\#} (read-write character buffer) {[char *, int]}]
821Like \samp{s\#}, but accepts any object which implements the
822read-write buffer interface. The \ctype{char *} variable is set to
823point to the first byte of the buffer, and the \ctype{int} is set to
824the length of the buffer. Only single-segment buffer objects are
825accepted; \exception{TypeError} is raised for all others.
826
827\item[\samp{(\var{items})} (tuple) {[\var{matching-items}]}]
828The object must be a Python sequence whose length is the number of
829format units in \var{items}. The C arguments must correspond to the
830individual format units in \var{items}. Format units for sequences
831may be nested.
832
Fred Drake0aa811c2001-10-20 04:24:09 +0000833\note{Prior to Python version 1.5.2, this format specifier
Fred Drakecc8f44b2001-08-20 19:30:29 +0000834only accepted a tuple containing the individual parameters, not an
835arbitrary sequence. Code which previously caused
836\exception{TypeError} to be raised here may now proceed without an
Fred Drake0aa811c2001-10-20 04:24:09 +0000837exception. This is not expected to be a problem for existing code.}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000838
839\end{description}
840
841It is possible to pass Python long integers where integers are
842requested; however no proper range checking is done --- the most
843significant bits are silently truncated when the receiving field is
844too small to receive the value (actually, the semantics are inherited
845from downcasts in C --- your mileage may vary).
846
847A few other characters have a meaning in a format string. These may
848not occur inside nested parentheses. They are:
849
850\begin{description}
851
852\item[\samp{|}]
853Indicates that the remaining arguments in the Python argument list are
854optional. The C variables corresponding to optional arguments should
855be initialized to their default value --- when an optional argument is
856not specified, \cfunction{PyArg_ParseTuple()} does not touch the contents
857of the corresponding C variable(s).
858
859\item[\samp{:}]
860The list of format units ends here; the string after the colon is used
861as the function name in error messages (the ``associated value'' of
862the exception that \cfunction{PyArg_ParseTuple()} raises).
863
864\item[\samp{;}]
865The list of format units ends here; the string after the semicolon is
866used as the error message \emph{instead} of the default error message.
867Clearly, \samp{:} and \samp{;} mutually exclude each other.
868
869\end{description}
870
871Some example calls:
872
873\begin{verbatim}
874 int ok;
875 int i, j;
876 long k, l;
877 char *s;
878 int size;
879
880 ok = PyArg_ParseTuple(args, ""); /* No arguments */
881 /* Python call: f() */
882\end{verbatim}
883
884\begin{verbatim}
885 ok = PyArg_ParseTuple(args, "s", &s); /* A string */
886 /* Possible Python call: f('whoops!') */
887\end{verbatim}
888
889\begin{verbatim}
890 ok = PyArg_ParseTuple(args, "lls", &k, &l, &s); /* Two longs and a string */
891 /* Possible Python call: f(1, 2, 'three') */
892\end{verbatim}
893
894\begin{verbatim}
895 ok = PyArg_ParseTuple(args, "(ii)s#", &i, &j, &s, &size);
896 /* A pair of ints and a string, whose size is also returned */
897 /* Possible Python call: f((1, 2), 'three') */
898\end{verbatim}
899
900\begin{verbatim}
901 {
902 char *file;
903 char *mode = "r";
904 int bufsize = 0;
905 ok = PyArg_ParseTuple(args, "s|si", &file, &mode, &bufsize);
906 /* A string, and optionally another string and an integer */
907 /* Possible Python calls:
908 f('spam')
909 f('spam', 'w')
910 f('spam', 'wb', 100000) */
911 }
912\end{verbatim}
913
914\begin{verbatim}
915 {
916 int left, top, right, bottom, h, v;
917 ok = PyArg_ParseTuple(args, "((ii)(ii))(ii)",
918 &left, &top, &right, &bottom, &h, &v);
919 /* A rectangle and a point */
920 /* Possible Python call:
921 f(((0, 0), (400, 300)), (10, 10)) */
922 }
923\end{verbatim}
924
925\begin{verbatim}
926 {
927 Py_complex c;
928 ok = PyArg_ParseTuple(args, "D:myfunction", &c);
929 /* a complex, also providing a function name for errors */
930 /* Possible Python call: myfunction(1+2j) */
931 }
932\end{verbatim}
933
934
935\section{Keyword Parameters for Extension Functions
936 \label{parseTupleAndKeywords}}
937
938The \cfunction{PyArg_ParseTupleAndKeywords()} function is declared as
939follows:
940
941\begin{verbatim}
942int PyArg_ParseTupleAndKeywords(PyObject *arg, PyObject *kwdict,
943 char *format, char **kwlist, ...);
944\end{verbatim}
945
946The \var{arg} and \var{format} parameters are identical to those of the
947\cfunction{PyArg_ParseTuple()} function. The \var{kwdict} parameter
948is the dictionary of keywords received as the third parameter from the
Fred Drakec37b65e2001-11-28 07:26:15 +0000949Python runtime. The \var{kwlist} parameter is a \NULL-terminated
Fred Drakecc8f44b2001-08-20 19:30:29 +0000950list of strings which identify the parameters; the names are matched
951with the type information from \var{format} from left to right. On
952success, \cfunction{PyArg_ParseTupleAndKeywords()} returns true,
953otherwise it returns false and raises an appropriate exception.
954
Fred Drake0aa811c2001-10-20 04:24:09 +0000955\note{Nested tuples cannot be parsed when using keyword
Fred Drakecc8f44b2001-08-20 19:30:29 +0000956arguments! Keyword parameters passed in which are not present in the
Fred Drake0aa811c2001-10-20 04:24:09 +0000957\var{kwlist} will cause \exception{TypeError} to be raised.}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000958
959Here is an example module which uses keywords, based on an example by
960Geoff Philbrick (\email{philbrick@hks.com}):%
961\index{Philbrick, Geoff}
962
963\begin{verbatim}
Fred Drakecc8f44b2001-08-20 19:30:29 +0000964#include "Python.h"
965
966static PyObject *
967keywdarg_parrot(self, args, keywds)
968 PyObject *self;
969 PyObject *args;
970 PyObject *keywds;
971{
972 int voltage;
973 char *state = "a stiff";
974 char *action = "voom";
975 char *type = "Norwegian Blue";
976
977 static char *kwlist[] = {"voltage", "state", "action", "type", NULL};
978
979 if (!PyArg_ParseTupleAndKeywords(args, keywds, "i|sss", kwlist,
980 &voltage, &state, &action, &type))
981 return NULL;
982
983 printf("-- This parrot wouldn't %s if you put %i Volts through it.\n",
984 action, voltage);
985 printf("-- Lovely plumage, the %s -- It's %s!\n", type, state);
986
987 Py_INCREF(Py_None);
988
989 return Py_None;
990}
991
992static PyMethodDef keywdarg_methods[] = {
993 /* The cast of the function is necessary since PyCFunction values
994 * only take two PyObject* parameters, and keywdarg_parrot() takes
995 * three.
996 */
Fred Drakeef6373a2001-11-17 06:50:42 +0000997 {"parrot", (PyCFunction)keywdarg_parrot, METH_VARARGS|METH_KEYWORDS,
998 "Print a lovely skit to standard output."},
999 {NULL, NULL, 0, NULL} /* sentinel */
Fred Drakecc8f44b2001-08-20 19:30:29 +00001000};
1001
1002void
Fred Drakeef6373a2001-11-17 06:50:42 +00001003initkeywdarg(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +00001004{
1005 /* Create the module and add the functions */
1006 Py_InitModule("keywdarg", keywdarg_methods);
1007}
1008\end{verbatim}
1009
1010
1011\section{Building Arbitrary Values
1012 \label{buildValue}}
1013
1014This function is the counterpart to \cfunction{PyArg_ParseTuple()}. It is
1015declared as follows:
1016
1017\begin{verbatim}
1018PyObject *Py_BuildValue(char *format, ...);
1019\end{verbatim}
1020
1021It recognizes a set of format units similar to the ones recognized by
1022\cfunction{PyArg_ParseTuple()}, but the arguments (which are input to the
1023function, not output) must not be pointers, just values. It returns a
1024new Python object, suitable for returning from a C function called
1025from Python.
1026
1027One difference with \cfunction{PyArg_ParseTuple()}: while the latter
1028requires its first argument to be a tuple (since Python argument lists
1029are always represented as tuples internally),
1030\cfunction{Py_BuildValue()} does not always build a tuple. It builds
1031a tuple only if its format string contains two or more format units.
1032If the format string is empty, it returns \code{None}; if it contains
1033exactly one format unit, it returns whatever object is described by
1034that format unit. To force it to return a tuple of size 0 or one,
1035parenthesize the format string.
1036
1037When memory buffers are passed as parameters to supply data to build
1038objects, as for the \samp{s} and \samp{s\#} formats, the required data
1039is copied. Buffers provided by the caller are never referenced by the
1040objects created by \cfunction{Py_BuildValue()}. In other words, if
1041your code invokes \cfunction{malloc()} and passes the allocated memory
1042to \cfunction{Py_BuildValue()}, your code is responsible for
1043calling \cfunction{free()} for that memory once
1044\cfunction{Py_BuildValue()} returns.
1045
1046In the following description, the quoted form is the format unit; the
1047entry in (round) parentheses is the Python object type that the format
1048unit will return; and the entry in [square] brackets is the type of
1049the C value(s) to be passed.
1050
1051The characters space, tab, colon and comma are ignored in format
1052strings (but not within format units such as \samp{s\#}). This can be
1053used to make long format strings a tad more readable.
1054
1055\begin{description}
1056
1057\item[\samp{s} (string) {[char *]}]
1058Convert a null-terminated C string to a Python object. If the C
Fred Drakec37b65e2001-11-28 07:26:15 +00001059string pointer is \NULL, \code{None} is used.
Fred Drakecc8f44b2001-08-20 19:30:29 +00001060
1061\item[\samp{s\#} (string) {[char *, int]}]
1062Convert a C string and its length to a Python object. If the C string
Fred Drakec37b65e2001-11-28 07:26:15 +00001063pointer is \NULL, the length is ignored and \code{None} is
Fred Drakecc8f44b2001-08-20 19:30:29 +00001064returned.
1065
1066\item[\samp{z} (string or \code{None}) {[char *]}]
1067Same as \samp{s}.
1068
1069\item[\samp{z\#} (string or \code{None}) {[char *, int]}]
1070Same as \samp{s\#}.
1071
1072\item[\samp{u} (Unicode string) {[Py_UNICODE *]}]
1073Convert a null-terminated buffer of Unicode (UCS-2) data to a Python
1074Unicode object. If the Unicode buffer pointer is \NULL,
1075\code{None} is returned.
1076
1077\item[\samp{u\#} (Unicode string) {[Py_UNICODE *, int]}]
1078Convert a Unicode (UCS-2) data buffer and its length to a Python
1079Unicode object. If the Unicode buffer pointer is \NULL, the length
1080is ignored and \code{None} is returned.
1081
1082\item[\samp{i} (integer) {[int]}]
1083Convert a plain C \ctype{int} to a Python integer object.
1084
1085\item[\samp{b} (integer) {[char]}]
1086Same as \samp{i}.
1087
1088\item[\samp{h} (integer) {[short int]}]
1089Same as \samp{i}.
1090
1091\item[\samp{l} (integer) {[long int]}]
1092Convert a C \ctype{long int} to a Python integer object.
1093
1094\item[\samp{c} (string of length 1) {[char]}]
1095Convert a C \ctype{int} representing a character to a Python string of
1096length 1.
1097
1098\item[\samp{d} (float) {[double]}]
1099Convert a C \ctype{double} to a Python floating point number.
1100
1101\item[\samp{f} (float) {[float]}]
1102Same as \samp{d}.
1103
1104\item[\samp{D} (complex) {[Py_complex *]}]
1105Convert a C \ctype{Py_complex} structure to a Python complex number.
1106
1107\item[\samp{O} (object) {[PyObject *]}]
1108Pass a Python object untouched (except for its reference count, which
1109is incremented by one). If the object passed in is a \NULL{}
1110pointer, it is assumed that this was caused because the call producing
1111the argument found an error and set an exception. Therefore,
1112\cfunction{Py_BuildValue()} will return \NULL{} but won't raise an
1113exception. If no exception has been raised yet,
1114\cdata{PyExc_SystemError} is set.
1115
1116\item[\samp{S} (object) {[PyObject *]}]
1117Same as \samp{O}.
1118
1119\item[\samp{U} (object) {[PyObject *]}]
1120Same as \samp{O}.
1121
1122\item[\samp{N} (object) {[PyObject *]}]
1123Same as \samp{O}, except it doesn't increment the reference count on
1124the object. Useful when the object is created by a call to an object
1125constructor in the argument list.
1126
1127\item[\samp{O\&} (object) {[\var{converter}, \var{anything}]}]
1128Convert \var{anything} to a Python object through a \var{converter}
1129function. The function is called with \var{anything} (which should be
1130compatible with \ctype{void *}) as its argument and should return a
1131``new'' Python object, or \NULL{} if an error occurred.
1132
1133\item[\samp{(\var{items})} (tuple) {[\var{matching-items}]}]
1134Convert a sequence of C values to a Python tuple with the same number
1135of items.
1136
1137\item[\samp{[\var{items}]} (list) {[\var{matching-items}]}]
1138Convert a sequence of C values to a Python list with the same number
1139of items.
1140
1141\item[\samp{\{\var{items}\}} (dictionary) {[\var{matching-items}]}]
1142Convert a sequence of C values to a Python dictionary. Each pair of
1143consecutive C values adds one item to the dictionary, serving as key
1144and value, respectively.
1145
1146\end{description}
1147
1148If there is an error in the format string, the
1149\cdata{PyExc_SystemError} exception is raised and \NULL{} returned.
1150
1151Examples (to the left the call, to the right the resulting Python value):
1152
1153\begin{verbatim}
1154 Py_BuildValue("") None
1155 Py_BuildValue("i", 123) 123
1156 Py_BuildValue("iii", 123, 456, 789) (123, 456, 789)
1157 Py_BuildValue("s", "hello") 'hello'
1158 Py_BuildValue("ss", "hello", "world") ('hello', 'world')
1159 Py_BuildValue("s#", "hello", 4) 'hell'
1160 Py_BuildValue("()") ()
1161 Py_BuildValue("(i)", 123) (123,)
1162 Py_BuildValue("(ii)", 123, 456) (123, 456)
1163 Py_BuildValue("(i,i)", 123, 456) (123, 456)
1164 Py_BuildValue("[i,i]", 123, 456) [123, 456]
1165 Py_BuildValue("{s:i,s:i}",
1166 "abc", 123, "def", 456) {'abc': 123, 'def': 456}
1167 Py_BuildValue("((ii)(ii)) (ii)",
1168 1, 2, 3, 4, 5, 6) (((1, 2), (3, 4)), (5, 6))
1169\end{verbatim}
1170
1171
1172\section{Reference Counts
1173 \label{refcounts}}
1174
Fred Drakec37b65e2001-11-28 07:26:15 +00001175In languages like C or \Cpp, the programmer is responsible for
Fred Drakecc8f44b2001-08-20 19:30:29 +00001176dynamic allocation and deallocation of memory on the heap. In C,
1177this is done using the functions \cfunction{malloc()} and
Fred Drakec37b65e2001-11-28 07:26:15 +00001178\cfunction{free()}. In \Cpp, the operators \keyword{new} and
Fred Drakecc8f44b2001-08-20 19:30:29 +00001179\keyword{delete} are used with essentially the same meaning; they are
1180actually implemented using \cfunction{malloc()} and
1181\cfunction{free()}, so we'll restrict the following discussion to the
1182latter.
1183
1184Every block of memory allocated with \cfunction{malloc()} should
1185eventually be returned to the pool of available memory by exactly one
1186call to \cfunction{free()}. It is important to call
1187\cfunction{free()} at the right time. If a block's address is
1188forgotten but \cfunction{free()} is not called for it, the memory it
1189occupies cannot be reused until the program terminates. This is
1190called a \dfn{memory leak}. On the other hand, if a program calls
1191\cfunction{free()} for a block and then continues to use the block, it
1192creates a conflict with re-use of the block through another
1193\cfunction{malloc()} call. This is called \dfn{using freed memory}.
1194It has the same bad consequences as referencing uninitialized data ---
1195core dumps, wrong results, mysterious crashes.
1196
1197Common causes of memory leaks are unusual paths through the code. For
1198instance, a function may allocate a block of memory, do some
1199calculation, and then free the block again. Now a change in the
1200requirements for the function may add a test to the calculation that
1201detects an error condition and can return prematurely from the
1202function. It's easy to forget to free the allocated memory block when
1203taking this premature exit, especially when it is added later to the
1204code. Such leaks, once introduced, often go undetected for a long
1205time: the error exit is taken only in a small fraction of all calls,
1206and most modern machines have plenty of virtual memory, so the leak
1207only becomes apparent in a long-running process that uses the leaking
1208function frequently. Therefore, it's important to prevent leaks from
1209happening by having a coding convention or strategy that minimizes
1210this kind of errors.
1211
1212Since Python makes heavy use of \cfunction{malloc()} and
1213\cfunction{free()}, it needs a strategy to avoid memory leaks as well
1214as the use of freed memory. The chosen method is called
1215\dfn{reference counting}. The principle is simple: every object
1216contains a counter, which is incremented when a reference to the
1217object is stored somewhere, and which is decremented when a reference
1218to it is deleted. When the counter reaches zero, the last reference
1219to the object has been deleted and the object is freed.
1220
1221An alternative strategy is called \dfn{automatic garbage collection}.
1222(Sometimes, reference counting is also referred to as a garbage
1223collection strategy, hence my use of ``automatic'' to distinguish the
1224two.) The big advantage of automatic garbage collection is that the
1225user doesn't need to call \cfunction{free()} explicitly. (Another claimed
1226advantage is an improvement in speed or memory usage --- this is no
1227hard fact however.) The disadvantage is that for C, there is no
1228truly portable automatic garbage collector, while reference counting
1229can be implemented portably (as long as the functions \cfunction{malloc()}
1230and \cfunction{free()} are available --- which the C Standard guarantees).
1231Maybe some day a sufficiently portable automatic garbage collector
1232will be available for C. Until then, we'll have to live with
1233reference counts.
1234
1235\subsection{Reference Counting in Python
1236 \label{refcountsInPython}}
1237
1238There are two macros, \code{Py_INCREF(x)} and \code{Py_DECREF(x)},
1239which handle the incrementing and decrementing of the reference count.
1240\cfunction{Py_DECREF()} also frees the object when the count reaches zero.
1241For flexibility, it doesn't call \cfunction{free()} directly --- rather, it
1242makes a call through a function pointer in the object's \dfn{type
1243object}. For this purpose (and others), every object also contains a
1244pointer to its type object.
1245
1246The big question now remains: when to use \code{Py_INCREF(x)} and
1247\code{Py_DECREF(x)}? Let's first introduce some terms. Nobody
1248``owns'' an object; however, you can \dfn{own a reference} to an
1249object. An object's reference count is now defined as the number of
1250owned references to it. The owner of a reference is responsible for
1251calling \cfunction{Py_DECREF()} when the reference is no longer
1252needed. Ownership of a reference can be transferred. There are three
1253ways to dispose of an owned reference: pass it on, store it, or call
1254\cfunction{Py_DECREF()}. Forgetting to dispose of an owned reference
1255creates a memory leak.
1256
1257It is also possible to \dfn{borrow}\footnote{The metaphor of
1258``borrowing'' a reference is not completely correct: the owner still
1259has a copy of the reference.} a reference to an object. The borrower
1260of a reference should not call \cfunction{Py_DECREF()}. The borrower must
1261not hold on to the object longer than the owner from which it was
1262borrowed. Using a borrowed reference after the owner has disposed of
1263it risks using freed memory and should be avoided
1264completely.\footnote{Checking that the reference count is at least 1
1265\strong{does not work} --- the reference count itself could be in
1266freed memory and may thus be reused for another object!}
1267
1268The advantage of borrowing over owning a reference is that you don't
1269need to take care of disposing of the reference on all possible paths
1270through the code --- in other words, with a borrowed reference you
1271don't run the risk of leaking when a premature exit is taken. The
1272disadvantage of borrowing over leaking is that there are some subtle
1273situations where in seemingly correct code a borrowed reference can be
1274used after the owner from which it was borrowed has in fact disposed
1275of it.
1276
1277A borrowed reference can be changed into an owned reference by calling
1278\cfunction{Py_INCREF()}. This does not affect the status of the owner from
1279which the reference was borrowed --- it creates a new owned reference,
1280and gives full owner responsibilities (the new owner must
1281dispose of the reference properly, as well as the previous owner).
1282
1283
1284\subsection{Ownership Rules
1285 \label{ownershipRules}}
1286
1287Whenever an object reference is passed into or out of a function, it
1288is part of the function's interface specification whether ownership is
1289transferred with the reference or not.
1290
1291Most functions that return a reference to an object pass on ownership
1292with the reference. In particular, all functions whose function it is
1293to create a new object, such as \cfunction{PyInt_FromLong()} and
Fred Drake92024d12001-11-29 07:16:19 +00001294\cfunction{Py_BuildValue()}, pass ownership to the receiver. Even if
1295the object is not actually new, you still receive ownership of a new
1296reference to that object. For instance, \cfunction{PyInt_FromLong()}
1297maintains a cache of popular values and can return a reference to a
1298cached item.
Fred Drakecc8f44b2001-08-20 19:30:29 +00001299
1300Many functions that extract objects from other objects also transfer
1301ownership with the reference, for instance
1302\cfunction{PyObject_GetAttrString()}. The picture is less clear, here,
1303however, since a few common routines are exceptions:
1304\cfunction{PyTuple_GetItem()}, \cfunction{PyList_GetItem()},
1305\cfunction{PyDict_GetItem()}, and \cfunction{PyDict_GetItemString()}
1306all return references that you borrow from the tuple, list or
1307dictionary.
1308
1309The function \cfunction{PyImport_AddModule()} also returns a borrowed
1310reference, even though it may actually create the object it returns:
1311this is possible because an owned reference to the object is stored in
1312\code{sys.modules}.
1313
1314When you pass an object reference into another function, in general,
1315the function borrows the reference from you --- if it needs to store
1316it, it will use \cfunction{Py_INCREF()} to become an independent
1317owner. There are exactly two important exceptions to this rule:
1318\cfunction{PyTuple_SetItem()} and \cfunction{PyList_SetItem()}. These
1319functions take over ownership of the item passed to them --- even if
1320they fail! (Note that \cfunction{PyDict_SetItem()} and friends don't
1321take over ownership --- they are ``normal.'')
1322
1323When a C function is called from Python, it borrows references to its
1324arguments from the caller. The caller owns a reference to the object,
1325so the borrowed reference's lifetime is guaranteed until the function
1326returns. Only when such a borrowed reference must be stored or passed
1327on, it must be turned into an owned reference by calling
1328\cfunction{Py_INCREF()}.
1329
1330The object reference returned from a C function that is called from
1331Python must be an owned reference --- ownership is tranferred from the
1332function to its caller.
1333
1334
1335\subsection{Thin Ice
1336 \label{thinIce}}
1337
1338There are a few situations where seemingly harmless use of a borrowed
1339reference can lead to problems. These all have to do with implicit
1340invocations of the interpreter, which can cause the owner of a
1341reference to dispose of it.
1342
1343The first and most important case to know about is using
1344\cfunction{Py_DECREF()} on an unrelated object while borrowing a
1345reference to a list item. For instance:
1346
1347\begin{verbatim}
1348bug(PyObject *list) {
1349 PyObject *item = PyList_GetItem(list, 0);
1350
1351 PyList_SetItem(list, 1, PyInt_FromLong(0L));
1352 PyObject_Print(item, stdout, 0); /* BUG! */
1353}
1354\end{verbatim}
1355
1356This function first borrows a reference to \code{list[0]}, then
1357replaces \code{list[1]} with the value \code{0}, and finally prints
1358the borrowed reference. Looks harmless, right? But it's not!
1359
1360Let's follow the control flow into \cfunction{PyList_SetItem()}. The list
1361owns references to all its items, so when item 1 is replaced, it has
1362to dispose of the original item 1. Now let's suppose the original
1363item 1 was an instance of a user-defined class, and let's further
1364suppose that the class defined a \method{__del__()} method. If this
1365class instance has a reference count of 1, disposing of it will call
1366its \method{__del__()} method.
1367
1368Since it is written in Python, the \method{__del__()} method can execute
1369arbitrary Python code. Could it perhaps do something to invalidate
1370the reference to \code{item} in \cfunction{bug()}? You bet! Assuming
1371that the list passed into \cfunction{bug()} is accessible to the
1372\method{__del__()} method, it could execute a statement to the effect of
1373\samp{del list[0]}, and assuming this was the last reference to that
1374object, it would free the memory associated with it, thereby
1375invalidating \code{item}.
1376
1377The solution, once you know the source of the problem, is easy:
1378temporarily increment the reference count. The correct version of the
1379function reads:
1380
1381\begin{verbatim}
1382no_bug(PyObject *list) {
1383 PyObject *item = PyList_GetItem(list, 0);
1384
1385 Py_INCREF(item);
1386 PyList_SetItem(list, 1, PyInt_FromLong(0L));
1387 PyObject_Print(item, stdout, 0);
1388 Py_DECREF(item);
1389}
1390\end{verbatim}
1391
1392This is a true story. An older version of Python contained variants
1393of this bug and someone spent a considerable amount of time in a C
1394debugger to figure out why his \method{__del__()} methods would fail...
1395
1396The second case of problems with a borrowed reference is a variant
1397involving threads. Normally, multiple threads in the Python
1398interpreter can't get in each other's way, because there is a global
1399lock protecting Python's entire object space. However, it is possible
1400to temporarily release this lock using the macro
1401\code{Py_BEGIN_ALLOW_THREADS}, and to re-acquire it using
1402\code{Py_END_ALLOW_THREADS}. This is common around blocking I/O
1403calls, to let other threads use the processor while waiting for the I/O to
1404complete. Obviously, the following function has the same problem as
1405the previous one:
1406
1407\begin{verbatim}
1408bug(PyObject *list) {
1409 PyObject *item = PyList_GetItem(list, 0);
1410 Py_BEGIN_ALLOW_THREADS
1411 ...some blocking I/O call...
1412 Py_END_ALLOW_THREADS
1413 PyObject_Print(item, stdout, 0); /* BUG! */
1414}
1415\end{verbatim}
1416
1417
1418\subsection{NULL Pointers
1419 \label{nullPointers}}
1420
1421In general, functions that take object references as arguments do not
1422expect you to pass them \NULL{} pointers, and will dump core (or
1423cause later core dumps) if you do so. Functions that return object
1424references generally return \NULL{} only to indicate that an
1425exception occurred. The reason for not testing for \NULL{}
1426arguments is that functions often pass the objects they receive on to
Fred Drakec37b65e2001-11-28 07:26:15 +00001427other function --- if each function were to test for \NULL,
Fred Drakecc8f44b2001-08-20 19:30:29 +00001428there would be a lot of redundant tests and the code would run more
1429slowly.
1430
1431It is better to test for \NULL{} only at the ``source:'' when a
1432pointer that may be \NULL{} is received, for example, from
1433\cfunction{malloc()} or from a function that may raise an exception.
1434
1435The macros \cfunction{Py_INCREF()} and \cfunction{Py_DECREF()}
1436do not check for \NULL{} pointers --- however, their variants
1437\cfunction{Py_XINCREF()} and \cfunction{Py_XDECREF()} do.
1438
1439The macros for checking for a particular object type
1440(\code{Py\var{type}_Check()}) don't check for \NULL{} pointers ---
1441again, there is much code that calls several of these in a row to test
1442an object against various different expected types, and this would
1443generate redundant tests. There are no variants with \NULL{}
1444checking.
1445
1446The C function calling mechanism guarantees that the argument list
1447passed to C functions (\code{args} in the examples) is never
1448\NULL{} --- in fact it guarantees that it is always a tuple.\footnote{
1449These guarantees don't hold when you use the ``old'' style
1450calling convention --- this is still found in much existing code.}
1451
1452It is a severe error to ever let a \NULL{} pointer ``escape'' to
1453the Python user.
1454
1455% Frank Stajano:
1456% A pedagogically buggy example, along the lines of the previous listing,
1457% would be helpful here -- showing in more concrete terms what sort of
1458% actions could cause the problem. I can't very well imagine it from the
1459% description.
1460
1461
Fred Drakec37b65e2001-11-28 07:26:15 +00001462\section{Writing Extensions in \Cpp
Fred Drakecc8f44b2001-08-20 19:30:29 +00001463 \label{cplusplus}}
1464
Fred Drakec37b65e2001-11-28 07:26:15 +00001465It is possible to write extension modules in \Cpp. Some restrictions
Fred Drakecc8f44b2001-08-20 19:30:29 +00001466apply. If the main program (the Python interpreter) is compiled and
1467linked by the C compiler, global or static objects with constructors
1468cannot be used. This is not a problem if the main program is linked
1469by the \Cpp{} compiler. Functions that will be called by the
1470Python interpreter (in particular, module initalization functions)
1471have to be declared using \code{extern "C"}.
1472It is unnecessary to enclose the Python header files in
1473\code{extern "C" \{...\}} --- they use this form already if the symbol
1474\samp{__cplusplus} is defined (all recent \Cpp{} compilers define this
1475symbol).
1476
1477
1478\section{Providing a C API for an Extension Module
1479 \label{using-cobjects}}
1480\sectionauthor{Konrad Hinsen}{hinsen@cnrs-orleans.fr}
1481
1482Many extension modules just provide new functions and types to be
1483used from Python, but sometimes the code in an extension module can
1484be useful for other extension modules. For example, an extension
1485module could implement a type ``collection'' which works like lists
1486without order. Just like the standard Python list type has a C API
1487which permits extension modules to create and manipulate lists, this
1488new collection type should have a set of C functions for direct
1489manipulation from other extension modules.
1490
1491At first sight this seems easy: just write the functions (without
1492declaring them \keyword{static}, of course), provide an appropriate
1493header file, and document the C API. And in fact this would work if
1494all extension modules were always linked statically with the Python
1495interpreter. When modules are used as shared libraries, however, the
1496symbols defined in one module may not be visible to another module.
1497The details of visibility depend on the operating system; some systems
1498use one global namespace for the Python interpreter and all extension
1499modules (Windows, for example), whereas others require an explicit
1500list of imported symbols at module link time (AIX is one example), or
1501offer a choice of different strategies (most Unices). And even if
1502symbols are globally visible, the module whose functions one wishes to
1503call might not have been loaded yet!
1504
1505Portability therefore requires not to make any assumptions about
1506symbol visibility. This means that all symbols in extension modules
1507should be declared \keyword{static}, except for the module's
1508initialization function, in order to avoid name clashes with other
1509extension modules (as discussed in section~\ref{methodTable}). And it
1510means that symbols that \emph{should} be accessible from other
1511extension modules must be exported in a different way.
1512
1513Python provides a special mechanism to pass C-level information
1514(pointers) from one extension module to another one: CObjects.
1515A CObject is a Python data type which stores a pointer (\ctype{void
1516*}). CObjects can only be created and accessed via their C API, but
1517they can be passed around like any other Python object. In particular,
1518they can be assigned to a name in an extension module's namespace.
1519Other extension modules can then import this module, retrieve the
1520value of this name, and then retrieve the pointer from the CObject.
1521
1522There are many ways in which CObjects can be used to export the C API
1523of an extension module. Each name could get its own CObject, or all C
1524API pointers could be stored in an array whose address is published in
1525a CObject. And the various tasks of storing and retrieving the pointers
1526can be distributed in different ways between the module providing the
1527code and the client modules.
1528
1529The following example demonstrates an approach that puts most of the
1530burden on the writer of the exporting module, which is appropriate
1531for commonly used library modules. It stores all C API pointers
1532(just one in the example!) in an array of \ctype{void} pointers which
1533becomes the value of a CObject. The header file corresponding to
1534the module provides a macro that takes care of importing the module
1535and retrieving its C API pointers; client modules only have to call
1536this macro before accessing the C API.
1537
1538The exporting module is a modification of the \module{spam} module from
1539section~\ref{simpleExample}. The function \function{spam.system()}
1540does not call the C library function \cfunction{system()} directly,
1541but a function \cfunction{PySpam_System()}, which would of course do
1542something more complicated in reality (such as adding ``spam'' to
1543every command). This function \cfunction{PySpam_System()} is also
1544exported to other extension modules.
1545
1546The function \cfunction{PySpam_System()} is a plain C function,
1547declared \keyword{static} like everything else:
1548
1549\begin{verbatim}
1550static int
1551PySpam_System(command)
1552 char *command;
1553{
1554 return system(command);
1555}
1556\end{verbatim}
1557
1558The function \cfunction{spam_system()} is modified in a trivial way:
1559
1560\begin{verbatim}
1561static PyObject *
1562spam_system(self, args)
1563 PyObject *self;
1564 PyObject *args;
1565{
1566 char *command;
1567 int sts;
1568
1569 if (!PyArg_ParseTuple(args, "s", &command))
1570 return NULL;
1571 sts = PySpam_System(command);
1572 return Py_BuildValue("i", sts);
1573}
1574\end{verbatim}
1575
1576In the beginning of the module, right after the line
1577
1578\begin{verbatim}
1579#include "Python.h"
1580\end{verbatim}
1581
1582two more lines must be added:
1583
1584\begin{verbatim}
1585#define SPAM_MODULE
1586#include "spammodule.h"
1587\end{verbatim}
1588
1589The \code{\#define} is used to tell the header file that it is being
1590included in the exporting module, not a client module. Finally,
1591the module's initialization function must take care of initializing
1592the C API pointer array:
1593
1594\begin{verbatim}
1595void
Fred Drakeef6373a2001-11-17 06:50:42 +00001596initspam(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +00001597{
1598 PyObject *m;
1599 static void *PySpam_API[PySpam_API_pointers];
1600 PyObject *c_api_object;
1601
1602 m = Py_InitModule("spam", SpamMethods);
1603
1604 /* Initialize the C API pointer array */
1605 PySpam_API[PySpam_System_NUM] = (void *)PySpam_System;
1606
1607 /* Create a CObject containing the API pointer array's address */
1608 c_api_object = PyCObject_FromVoidPtr((void *)PySpam_API, NULL);
1609
1610 if (c_api_object != NULL) {
1611 /* Create a name for this object in the module's namespace */
1612 PyObject *d = PyModule_GetDict(m);
1613
1614 PyDict_SetItemString(d, "_C_API", c_api_object);
1615 Py_DECREF(c_api_object);
1616 }
1617}
1618\end{verbatim}
1619
Fred Drakeef6373a2001-11-17 06:50:42 +00001620Note that \code{PySpam_API} is declared \keyword{static}; otherwise
1621the pointer array would disappear when \function{initspam()} terminates!
Fred Drakecc8f44b2001-08-20 19:30:29 +00001622
1623The bulk of the work is in the header file \file{spammodule.h},
1624which looks like this:
1625
1626\begin{verbatim}
1627#ifndef Py_SPAMMODULE_H
1628#define Py_SPAMMODULE_H
1629#ifdef __cplusplus
1630extern "C" {
1631#endif
1632
1633/* Header file for spammodule */
1634
1635/* C API functions */
1636#define PySpam_System_NUM 0
1637#define PySpam_System_RETURN int
1638#define PySpam_System_PROTO (char *command)
1639
1640/* Total number of C API pointers */
1641#define PySpam_API_pointers 1
1642
1643
1644#ifdef SPAM_MODULE
1645/* This section is used when compiling spammodule.c */
1646
1647static PySpam_System_RETURN PySpam_System PySpam_System_PROTO;
1648
1649#else
1650/* This section is used in modules that use spammodule's API */
1651
1652static void **PySpam_API;
1653
1654#define PySpam_System \
1655 (*(PySpam_System_RETURN (*)PySpam_System_PROTO) PySpam_API[PySpam_System_NUM])
1656
1657#define import_spam() \
1658{ \
1659 PyObject *module = PyImport_ImportModule("spam"); \
1660 if (module != NULL) { \
1661 PyObject *module_dict = PyModule_GetDict(module); \
1662 PyObject *c_api_object = PyDict_GetItemString(module_dict, "_C_API"); \
1663 if (PyCObject_Check(c_api_object)) { \
1664 PySpam_API = (void **)PyCObject_AsVoidPtr(c_api_object); \
1665 } \
1666 } \
1667}
1668
1669#endif
1670
1671#ifdef __cplusplus
1672}
1673#endif
1674
1675#endif /* !defined(Py_SPAMMODULE_H */
1676\end{verbatim}
1677
1678All that a client module must do in order to have access to the
1679function \cfunction{PySpam_System()} is to call the function (or
1680rather macro) \cfunction{import_spam()} in its initialization
1681function:
1682
1683\begin{verbatim}
1684void
Fred Drakeef6373a2001-11-17 06:50:42 +00001685initclient(void)
Fred Drakecc8f44b2001-08-20 19:30:29 +00001686{
1687 PyObject *m;
1688
1689 Py_InitModule("client", ClientMethods);
1690 import_spam();
1691}
1692\end{verbatim}
1693
1694The main disadvantage of this approach is that the file
1695\file{spammodule.h} is rather complicated. However, the
1696basic structure is the same for each function that is
1697exported, so it has to be learned only once.
1698
1699Finally it should be mentioned that CObjects offer additional
1700functionality, which is especially useful for memory allocation and
1701deallocation of the pointer stored in a CObject. The details
1702are described in the \citetitle[../api/api.html]{Python/C API
1703Reference Manual} in the section ``CObjects'' and in the
1704implementation of CObjects (files \file{Include/cobject.h} and
1705\file{Objects/cobject.c} in the Python source code distribution).