Created
September 16, 2022 14:56
-
-
Save chuongmep/843bba1f38df2f85b191a2afd2752a0c to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#define PY_SSIZE_T_CLEAN | |
#include <Python.h> | |
int | |
main(int argc, char *argv[]) | |
{ | |
PyObject *pName, *pModule, *pFunc; | |
PyObject *pArgs, *pValue; | |
int i; | |
if (argc < 3) { | |
fprintf(stderr,"Usage: call pythonfile funcname [args]\n"); | |
return 1; | |
} | |
Py_Initialize(); | |
pName = PyUnicode_DecodeFSDefault(argv[1]); | |
/* Error checking of pName left out */ | |
pModule = PyImport_Import(pName); | |
Py_DECREF(pName); | |
if (pModule != NULL) { | |
pFunc = PyObject_GetAttrString(pModule, argv[2]); | |
/* pFunc is a new reference */ | |
if (pFunc && PyCallable_Check(pFunc)) { | |
pArgs = PyTuple_New(argc - 3); | |
for (i = 0; i < argc - 3; ++i) { | |
pValue = PyLong_FromLong(atoi(argv[i + 3])); | |
if (!pValue) { | |
Py_DECREF(pArgs); | |
Py_DECREF(pModule); | |
fprintf(stderr, "Cannot convert argument\n"); | |
return 1; | |
} | |
/* pValue reference stolen here: */ | |
PyTuple_SetItem(pArgs, i, pValue); | |
} | |
pValue = PyObject_CallObject(pFunc, pArgs); | |
Py_DECREF(pArgs); | |
if (pValue != NULL) { | |
printf("Result of call: %ld\n", PyLong_AsLong(pValue)); | |
Py_DECREF(pValue); | |
} | |
else { | |
Py_DECREF(pFunc); | |
Py_DECREF(pModule); | |
PyErr_Print(); | |
fprintf(stderr,"Call failed\n"); | |
return 1; | |
} | |
} | |
else { | |
if (PyErr_Occurred()) | |
PyErr_Print(); | |
fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]); | |
} | |
Py_XDECREF(pFunc); | |
Py_DECREF(pModule); | |
} | |
else { | |
PyErr_Print(); | |
fprintf(stderr, "Failed to load \"%s\"\n", argv[1]); | |
return 1; | |
} | |
if (Py_FinalizeEx() < 0) { | |
return 120; | |
} | |
return 0; | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment