blob: 5a7df0d98056bc371aaaf4e4fdadf616527803e9 [file] [log] [blame]
Fred Drake53765752001-08-04 01:58:36 +00001#include <Python.h>
2
3int
4main(int argc, char *argv[])
5{
6 PyObject *pName, *pModule, *pDict, *pFunc;
7 PyObject *pArgs, *pValue;
Neal Norwitzae6bf4a2002-12-02 04:40:21 +00008 int i;
Fred Drake53765752001-08-04 01:58:36 +00009
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 Norwitz0f30dbd2002-12-17 00:40:38 +000020 Py_DECREF(pName);
21
Fred Drake53765752001-08-04 01:58:36 +000022 if (pModule != NULL) {
Fred Drake44829292006-02-15 06:55:57 +000023 pFunc = PyObject_GetAttrString(pModule, argv[2]);
Georg Brandl9abfa902005-07-12 13:17:59 +000024 /* pFunc is a new reference */
Fred Drake53765752001-08-04 01:58:36 +000025
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 Norwitz0f30dbd2002-12-17 00:40:38 +000031 Py_DECREF(pArgs);
32 Py_DECREF(pModule);
Fred Drake53765752001-08-04 01:58:36 +000033 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 Norwitz0f30dbd2002-12-17 00:40:38 +000040 Py_DECREF(pArgs);
Fred Drake53765752001-08-04 01:58:36 +000041 if (pValue != NULL) {
42 printf("Result of call: %ld\n", PyInt_AsLong(pValue));
43 Py_DECREF(pValue);
44 }
45 else {
Georg Brandl9abfa902005-07-12 13:17:59 +000046 Py_DECREF(pFunc);
Neal Norwitz0f30dbd2002-12-17 00:40:38 +000047 Py_DECREF(pModule);
Fred Drake53765752001-08-04 01:58:36 +000048 PyErr_Print();
49 fprintf(stderr,"Call failed\n");
50 return 1;
51 }
Fred Drake53765752001-08-04 01:58:36 +000052 }
53 else {
Neal Norwitz0f30dbd2002-12-17 00:40:38 +000054 if (PyErr_Occurred())
55 PyErr_Print();
Fred Drake53765752001-08-04 01:58:36 +000056 fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]);
57 }
Georg Brandl9abfa902005-07-12 13:17:59 +000058 Py_XDECREF(pFunc);
Fred Drake53765752001-08-04 01:58:36 +000059 Py_DECREF(pModule);
60 }
61 else {
62 PyErr_Print();
63 fprintf(stderr, "Failed to load \"%s\"\n", argv[1]);
64 return 1;
65 }
Fred Drake53765752001-08-04 01:58:36 +000066 Py_Finalize();
67 return 0;
68}