summaryrefslogtreecommitdiffstats
path: root/Demo/embed/demo.c
blob: dc8bcf04fcbc46b45141a0e22f8b1fe9c0184503 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/* Example of embedding Python in another program */

#include "Python.h"

PyObject* PyInit_xyzzy(void); /* Forward */

main(int argc, char **argv)
{
    /* Ignore passed-in argc/argv. If desired, conversion
       should use mbstowcs to convert them. */
    wchar_t *args[] = {L"embed", L"hello", 0};

    /* Pass argv[0] to the Python interpreter */
    Py_SetProgramName(args[0]);

    /* Add a static module */
    PyImport_AppendInittab("xyzzy", PyInit_xyzzy);

    /* Initialize the Python interpreter.  Required. */
    Py_Initialize();

    /* Define sys.argv.  It is up to the application if you
       want this; you can also leave it undefined (since the Python
       code is generally not a main program it has no business
       touching sys.argv...) 

       If the third argument is true, sys.path is modified to include
       either the directory containing the script named by argv[0], or
       the current working directory.  This can be risky; if you run
       an application embedding Python in a directory controlled by
       someone else, attackers could put a Trojan-horse module in the
       directory (say, a file named os.py) that your application would
       then import and run.
    */
    PySys_SetArgvEx(argc, argv, 0);

    /* Do some application specific code */
    printf("Hello, brave new world\n\n");

    /* Execute some Python statements (in module __main__) */
    PyRun_SimpleString("import sys\n");
    PyRun_SimpleString("print(sys.builtin_module_names)\n");
    PyRun_SimpleString("print(sys.modules.keys())\n");
    PyRun_SimpleString("print(sys.executable)\n");
    PyRun_SimpleString("print(sys.argv)\n");

    /* Note that you can call any public function of the Python
       interpreter here, e.g. call_object(). */

    /* Some more application specific code */
    printf("\nGoodbye, cruel world\n");

    /* Exit, cleaning up the interpreter */
    Py_Exit(0);
    /*NOTREACHED*/
}

/* A static module */

/* 'self' is not used */
static PyObject *
xyzzy_foo(PyObject *self, PyObject* args)
{
    return PyLong_FromLong(42L);
}

static PyMethodDef xyzzy_methods[] = {
    {"foo",             xyzzy_foo,      METH_NOARGS,
     "Return the meaning of everything."},
    {NULL,              NULL}           /* sentinel */
};

static struct PyModuleDef xyzzymodule = {
    {}, /* m_base */
    "xyzzy",  /* m_name */
    0,  /* m_doc */
    0,  /* m_size */
    xyzzy_methods,  /* m_methods */
    0,  /* m_reload */
    0,  /* m_traverse */
    0,  /* m_clear */
    0,  /* m_free */
};

PyObject*
PyInit_xyzzy(void)
{
    return PyModule_Create(&xyzzymodule);
}