Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 1 | #include <Python.h> |
| 2 | |
| 3 | int |
| 4 | main(int argc, char *argv[]) |
| 5 | { |
| 6 | PyObject *pName, *pModule, *pDict, *pFunc; |
| 7 | PyObject *pArgs, *pValue; |
Neal Norwitz | ae6bf4a | 2002-12-02 04:40:21 +0000 | [diff] [blame] | 8 | int i; |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 9 | |
| 10 | if (argc < 3) { |
| 11 | fprintf(stderr,"Usage: call pythonfile funcname [args]\n"); |
| 12 | return 1; |
| 13 | } |
| 14 | |
| 15 | Py_Initialize(); |
| 16 | pName = PyString_FromString(argv[1]); |
| 17 | /* Error checking of pName left out */ |
| 18 | |
| 19 | pModule = PyImport_Import(pName); |
Neal Norwitz | 0f30dbd | 2002-12-17 00:40:38 +0000 | [diff] [blame] | 20 | Py_DECREF(pName); |
| 21 | |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 22 | if (pModule != NULL) { |
Fred Drake | 4482929 | 2006-02-15 06:55:57 +0000 | [diff] [blame] | 23 | pFunc = PyObject_GetAttrString(pModule, argv[2]); |
Georg Brandl | 9abfa90 | 2005-07-12 13:17:59 +0000 | [diff] [blame] | 24 | /* pFunc is a new reference */ |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 25 | |
| 26 | if (pFunc && PyCallable_Check(pFunc)) { |
| 27 | pArgs = PyTuple_New(argc - 3); |
| 28 | for (i = 0; i < argc - 3; ++i) { |
| 29 | pValue = PyInt_FromLong(atoi(argv[i + 3])); |
| 30 | if (!pValue) { |
Neal Norwitz | 0f30dbd | 2002-12-17 00:40:38 +0000 | [diff] [blame] | 31 | Py_DECREF(pArgs); |
| 32 | Py_DECREF(pModule); |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 33 | fprintf(stderr, "Cannot convert argument\n"); |
| 34 | return 1; |
| 35 | } |
| 36 | /* pValue reference stolen here: */ |
| 37 | PyTuple_SetItem(pArgs, i, pValue); |
| 38 | } |
| 39 | pValue = PyObject_CallObject(pFunc, pArgs); |
Neal Norwitz | 0f30dbd | 2002-12-17 00:40:38 +0000 | [diff] [blame] | 40 | Py_DECREF(pArgs); |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 41 | if (pValue != NULL) { |
| 42 | printf("Result of call: %ld\n", PyInt_AsLong(pValue)); |
| 43 | Py_DECREF(pValue); |
| 44 | } |
| 45 | else { |
Georg Brandl | 9abfa90 | 2005-07-12 13:17:59 +0000 | [diff] [blame] | 46 | Py_DECREF(pFunc); |
Neal Norwitz | 0f30dbd | 2002-12-17 00:40:38 +0000 | [diff] [blame] | 47 | Py_DECREF(pModule); |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 48 | PyErr_Print(); |
| 49 | fprintf(stderr,"Call failed\n"); |
| 50 | return 1; |
| 51 | } |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 52 | } |
| 53 | else { |
Neal Norwitz | 0f30dbd | 2002-12-17 00:40:38 +0000 | [diff] [blame] | 54 | if (PyErr_Occurred()) |
| 55 | PyErr_Print(); |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 56 | fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]); |
| 57 | } |
Georg Brandl | 9abfa90 | 2005-07-12 13:17:59 +0000 | [diff] [blame] | 58 | Py_XDECREF(pFunc); |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 59 | Py_DECREF(pModule); |
| 60 | } |
| 61 | else { |
| 62 | PyErr_Print(); |
| 63 | fprintf(stderr, "Failed to load \"%s\"\n", argv[1]); |
| 64 | return 1; |
| 65 | } |
Fred Drake | 5376575 | 2001-08-04 01:58:36 +0000 | [diff] [blame] | 66 | Py_Finalize(); |
| 67 | return 0; |
| 68 | } |