xref: /vim-8.2.3635/src/if_python.c (revision 2a0f3d3f)
1db913953SBram Moolenaar /* vi:set ts=8 sts=4 sw=4 noet:
2071d4279SBram Moolenaar  *
3071d4279SBram Moolenaar  * VIM - Vi IMproved	by Bram Moolenaar
4071d4279SBram Moolenaar  *
5071d4279SBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
6071d4279SBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
7071d4279SBram Moolenaar  * See README.txt for an overview of the Vim source code.
8071d4279SBram Moolenaar  */
9071d4279SBram Moolenaar /*
10071d4279SBram Moolenaar  * Python extensions by Paul Moore.
11071d4279SBram Moolenaar  * Changes for Unix by David Leonard.
12071d4279SBram Moolenaar  *
13071d4279SBram Moolenaar  * This consists of four parts:
14071d4279SBram Moolenaar  * 1. Python interpreter main program
15071d4279SBram Moolenaar  * 2. Python output stream: writes output via [e]msg().
16071d4279SBram Moolenaar  * 3. Implementation of the Vim module for Python
17071d4279SBram Moolenaar  * 4. Utility functions for handling the interface between Vim and Python.
18071d4279SBram Moolenaar  */
19071d4279SBram Moolenaar 
20071d4279SBram Moolenaar #include "vim.h"
21071d4279SBram Moolenaar 
22071d4279SBram Moolenaar #include <limits.h>
23071d4279SBram Moolenaar 
24071d4279SBram Moolenaar /* Python.h defines _POSIX_THREADS itself (if needed) */
25071d4279SBram Moolenaar #ifdef _POSIX_THREADS
26071d4279SBram Moolenaar # undef _POSIX_THREADS
27071d4279SBram Moolenaar #endif
28071d4279SBram Moolenaar 
29071d4279SBram Moolenaar #if defined(_WIN32) && defined(HAVE_FCNTL_H)
30071d4279SBram Moolenaar # undef HAVE_FCNTL_H
31071d4279SBram Moolenaar #endif
32071d4279SBram Moolenaar 
33071d4279SBram Moolenaar #ifdef _DEBUG
34071d4279SBram Moolenaar # undef _DEBUG
35071d4279SBram Moolenaar #endif
36071d4279SBram Moolenaar 
37071d4279SBram Moolenaar #ifdef HAVE_STDARG_H
38071d4279SBram Moolenaar # undef HAVE_STDARG_H	/* Python's config.h defines it as well. */
39071d4279SBram Moolenaar #endif
40be2c9ae9SBram Moolenaar #ifdef _POSIX_C_SOURCE
41be2c9ae9SBram Moolenaar # undef _POSIX_C_SOURCE	/* pyconfig.h defines it as well. */
42be2c9ae9SBram Moolenaar #endif
43be2c9ae9SBram Moolenaar #ifdef _XOPEN_SOURCE
44be2c9ae9SBram Moolenaar # undef _XOPEN_SOURCE	/* pyconfig.h defines it as well. */
45be2c9ae9SBram Moolenaar #endif
46071d4279SBram Moolenaar 
47071d4279SBram Moolenaar #include <Python.h>
48071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
49071d4279SBram Moolenaar # include "macglue.h"
50071d4279SBram Moolenaar # include <CodeFragments.h>
51071d4279SBram Moolenaar #endif
52071d4279SBram Moolenaar #undef main /* Defined in python.h - aargh */
53071d4279SBram Moolenaar #undef HAVE_FCNTL_H /* Clash with os_win32.h */
54071d4279SBram Moolenaar 
55e8cdcef8SBram Moolenaar #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000
56e8cdcef8SBram Moolenaar # define PY_SSIZE_T_CLEAN
57e8cdcef8SBram Moolenaar #endif
58e8cdcef8SBram Moolenaar 
59db913953SBram Moolenaar #define PyBytes_FromString PyString_FromString
60335e0b69SBram Moolenaar #define PyBytes_Check PyString_Check
61db913953SBram Moolenaar 
6219e60943SBram Moolenaar /* No-op conversion functions, use with care! */
6319e60943SBram Moolenaar #define PyString_AsBytes(obj) (obj)
6419e60943SBram Moolenaar #define PyString_FreeBytes(obj)
6519e60943SBram Moolenaar 
66071d4279SBram Moolenaar #if !defined(FEAT_PYTHON) && defined(PROTO)
67071d4279SBram Moolenaar /* Use this to be able to generate prototypes without python being used. */
68e7cb9cf6SBram Moolenaar # define PyObject Py_ssize_t
69e7cb9cf6SBram Moolenaar # define PyThreadState Py_ssize_t
70e7cb9cf6SBram Moolenaar # define PyTypeObject Py_ssize_t
71e7cb9cf6SBram Moolenaar struct PyMethodDef { Py_ssize_t a; };
72e7cb9cf6SBram Moolenaar # define PySequenceMethods Py_ssize_t
73071d4279SBram Moolenaar #endif
74071d4279SBram Moolenaar 
752afa3238SBram Moolenaar #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02070000
762afa3238SBram Moolenaar # define PY_USE_CAPSULE
772afa3238SBram Moolenaar #endif
782afa3238SBram Moolenaar 
792c45e945SBram Moolenaar #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000
802c45e945SBram Moolenaar # define PyInt Py_ssize_t
812c45e945SBram Moolenaar # define PyInquiry lenfunc
822c45e945SBram Moolenaar # define PyIntArgFunc ssizeargfunc
832c45e945SBram Moolenaar # define PyIntIntArgFunc ssizessizeargfunc
842c45e945SBram Moolenaar # define PyIntObjArgProc ssizeobjargproc
852c45e945SBram Moolenaar # define PyIntIntObjArgProc ssizessizeobjargproc
86e7cb9cf6SBram Moolenaar # define Py_ssize_t_fmt "n"
872c45e945SBram Moolenaar #else
882c45e945SBram Moolenaar # define PyInt int
894d1da49cSBram Moolenaar # define lenfunc inquiry
902c45e945SBram Moolenaar # define PyInquiry inquiry
912c45e945SBram Moolenaar # define PyIntArgFunc intargfunc
922c45e945SBram Moolenaar # define PyIntIntArgFunc intintargfunc
932c45e945SBram Moolenaar # define PyIntObjArgProc intobjargproc
942c45e945SBram Moolenaar # define PyIntIntObjArgProc intintobjargproc
95e7cb9cf6SBram Moolenaar # define Py_ssize_t_fmt "i"
962c45e945SBram Moolenaar #endif
972c45e945SBram Moolenaar 
98071d4279SBram Moolenaar /* Parser flags */
99071d4279SBram Moolenaar #define single_input	256
100071d4279SBram Moolenaar #define file_input	257
101071d4279SBram Moolenaar #define eval_input	258
102071d4279SBram Moolenaar 
103071d4279SBram Moolenaar #if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x020300F0
104071d4279SBram Moolenaar   /* Python 2.3: can invoke ":python" recursively. */
105071d4279SBram Moolenaar # define PY_CAN_RECURSE
106071d4279SBram Moolenaar #endif
107071d4279SBram Moolenaar 
108071d4279SBram Moolenaar # if defined(DYNAMIC_PYTHON) || defined(PROTO)
109071d4279SBram Moolenaar #  ifndef DYNAMIC_PYTHON
110e7cb9cf6SBram Moolenaar #   define HINSTANCE long_u		/* for generating prototypes */
111071d4279SBram Moolenaar #  endif
112071d4279SBram Moolenaar 
113fa5d1e63SBram Moolenaar # ifndef WIN3264
114bd5e15fdSBram Moolenaar #  include <dlfcn.h>
115bd5e15fdSBram Moolenaar #  define FARPROC void*
116bd5e15fdSBram Moolenaar #  define HINSTANCE void*
117644d37b8SBram Moolenaar #  if defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)
118b61f95c3SBram Moolenaar #   define load_dll(n) dlopen((n), RTLD_LAZY)
119b61f95c3SBram Moolenaar #  else
120fa5d1e63SBram Moolenaar #   define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL)
121b61f95c3SBram Moolenaar #  endif
122bd5e15fdSBram Moolenaar #  define close_dll dlclose
123bd5e15fdSBram Moolenaar #  define symbol_from_dll dlsym
124bd5e15fdSBram Moolenaar # else
125ebbcb824SBram Moolenaar #  define load_dll vimLoadLib
126bd5e15fdSBram Moolenaar #  define close_dll FreeLibrary
127bd5e15fdSBram Moolenaar #  define symbol_from_dll GetProcAddress
128bd5e15fdSBram Moolenaar # endif
129bd5e15fdSBram Moolenaar 
130e7cb9cf6SBram Moolenaar /* This makes if_python.c compile without warnings against Python 2.5
131e7cb9cf6SBram Moolenaar  * on Win32 and Win64. */
132e7cb9cf6SBram Moolenaar # undef PyRun_SimpleString
133db913953SBram Moolenaar # undef PyRun_String
134e7cb9cf6SBram Moolenaar # undef PyArg_Parse
135e7cb9cf6SBram Moolenaar # undef PyArg_ParseTuple
136e7cb9cf6SBram Moolenaar # undef Py_BuildValue
137e7cb9cf6SBram Moolenaar # undef Py_InitModule4
138e7cb9cf6SBram Moolenaar # undef Py_InitModule4_64
139db913953SBram Moolenaar # undef PyObject_CallMethod
140e7cb9cf6SBram Moolenaar 
141071d4279SBram Moolenaar /*
142071d4279SBram Moolenaar  * Wrapper defines
143071d4279SBram Moolenaar  */
144071d4279SBram Moolenaar # define PyArg_Parse dll_PyArg_Parse
145071d4279SBram Moolenaar # define PyArg_ParseTuple dll_PyArg_ParseTuple
14619e60943SBram Moolenaar # define PyMem_Free dll_PyMem_Free
147db913953SBram Moolenaar # define PyMem_Malloc dll_PyMem_Malloc
148071d4279SBram Moolenaar # define PyDict_SetItemString dll_PyDict_SetItemString
149071d4279SBram Moolenaar # define PyErr_BadArgument dll_PyErr_BadArgument
150d5f729caSBram Moolenaar # define PyErr_NewException dll_PyErr_NewException
151071d4279SBram Moolenaar # define PyErr_Clear dll_PyErr_Clear
1524d36987cSBram Moolenaar # define PyErr_PrintEx dll_PyErr_PrintEx
153071d4279SBram Moolenaar # define PyErr_NoMemory dll_PyErr_NoMemory
154071d4279SBram Moolenaar # define PyErr_Occurred dll_PyErr_Occurred
155071d4279SBram Moolenaar # define PyErr_SetNone dll_PyErr_SetNone
156071d4279SBram Moolenaar # define PyErr_SetString dll_PyErr_SetString
1574d188da2SBram Moolenaar # define PyErr_SetObject dll_PyErr_SetObject
158071d4279SBram Moolenaar # define PyEval_InitThreads dll_PyEval_InitThreads
159071d4279SBram Moolenaar # define PyEval_RestoreThread dll_PyEval_RestoreThread
160071d4279SBram Moolenaar # define PyEval_SaveThread dll_PyEval_SaveThread
161071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE
162071d4279SBram Moolenaar #  define PyGILState_Ensure dll_PyGILState_Ensure
163071d4279SBram Moolenaar #  define PyGILState_Release dll_PyGILState_Release
164071d4279SBram Moolenaar # endif
165071d4279SBram Moolenaar # define PyInt_AsLong dll_PyInt_AsLong
166071d4279SBram Moolenaar # define PyInt_FromLong dll_PyInt_FromLong
167db913953SBram Moolenaar # define PyLong_AsLong dll_PyLong_AsLong
168db913953SBram Moolenaar # define PyLong_FromLong dll_PyLong_FromLong
16966b7985eSBram Moolenaar # define PyBool_Type (*dll_PyBool_Type)
170071d4279SBram Moolenaar # define PyInt_Type (*dll_PyInt_Type)
171db913953SBram Moolenaar # define PyLong_Type (*dll_PyLong_Type)
172071d4279SBram Moolenaar # define PyList_GetItem dll_PyList_GetItem
1730ac9379aSBram Moolenaar # define PyList_Append dll_PyList_Append
174071d4279SBram Moolenaar # define PyList_New dll_PyList_New
175071d4279SBram Moolenaar # define PyList_SetItem dll_PyList_SetItem
176071d4279SBram Moolenaar # define PyList_Size dll_PyList_Size
177071d4279SBram Moolenaar # define PyList_Type (*dll_PyList_Type)
178db913953SBram Moolenaar # define PySequence_Check dll_PySequence_Check
179db913953SBram Moolenaar # define PySequence_Size dll_PySequence_Size
180db913953SBram Moolenaar # define PySequence_GetItem dll_PySequence_GetItem
181db913953SBram Moolenaar # define PyTuple_Size dll_PyTuple_Size
182db913953SBram Moolenaar # define PyTuple_GetItem dll_PyTuple_GetItem
183db913953SBram Moolenaar # define PyTuple_Type (*dll_PyTuple_Type)
184071d4279SBram Moolenaar # define PyImport_ImportModule dll_PyImport_ImportModule
1850ac9379aSBram Moolenaar # define PyDict_New dll_PyDict_New
186071d4279SBram Moolenaar # define PyDict_GetItemString dll_PyDict_GetItemString
187db913953SBram Moolenaar # define PyDict_Next dll_PyDict_Next
188db913953SBram Moolenaar # ifdef PyMapping_Items
189db913953SBram Moolenaar #  define PY_NO_MAPPING_ITEMS
190db913953SBram Moolenaar # else
191db913953SBram Moolenaar #  define PyMapping_Items dll_PyMapping_Items
192db913953SBram Moolenaar # endif
193db913953SBram Moolenaar # define PyObject_CallMethod dll_PyObject_CallMethod
194db913953SBram Moolenaar # define PyMapping_Check dll_PyMapping_Check
195db913953SBram Moolenaar # define PyIter_Next dll_PyIter_Next
196071d4279SBram Moolenaar # define PyModule_GetDict dll_PyModule_GetDict
197071d4279SBram Moolenaar # define PyRun_SimpleString dll_PyRun_SimpleString
198db913953SBram Moolenaar # define PyRun_String dll_PyRun_String
199d620aa9bSBram Moolenaar # define PyObject_GetAttrString dll_PyObject_GetAttrString
200d620aa9bSBram Moolenaar # define PyObject_SetAttrString dll_PyObject_SetAttrString
201d620aa9bSBram Moolenaar # define PyObject_CallFunctionObjArgs dll_PyObject_CallFunctionObjArgs
202071d4279SBram Moolenaar # define PyString_AsString dll_PyString_AsString
203cdab9051SBram Moolenaar # define PyString_AsStringAndSize dll_PyString_AsStringAndSize
204071d4279SBram Moolenaar # define PyString_FromString dll_PyString_FromString
205071d4279SBram Moolenaar # define PyString_FromStringAndSize dll_PyString_FromStringAndSize
206071d4279SBram Moolenaar # define PyString_Size dll_PyString_Size
207071d4279SBram Moolenaar # define PyString_Type (*dll_PyString_Type)
208db913953SBram Moolenaar # define PyUnicode_Type (*dll_PyUnicode_Type)
209cc3e85f1SBram Moolenaar # undef PyUnicode_AsEncodedString
210cc3e85f1SBram Moolenaar # define PyUnicode_AsEncodedString py_PyUnicode_AsEncodedString
211db913953SBram Moolenaar # define PyFloat_AsDouble dll_PyFloat_AsDouble
212db913953SBram Moolenaar # define PyFloat_FromDouble dll_PyFloat_FromDouble
213db913953SBram Moolenaar # define PyFloat_Type (*dll_PyFloat_Type)
214db913953SBram Moolenaar # define PyImport_AddModule (*dll_PyImport_AddModule)
215071d4279SBram Moolenaar # define PySys_SetObject dll_PySys_SetObject
216071d4279SBram Moolenaar # define PySys_SetArgv dll_PySys_SetArgv
217071d4279SBram Moolenaar # define PyType_Type (*dll_PyType_Type)
21830fec7bcSBram Moolenaar # define PyType_Ready (*dll_PyType_Ready)
219071d4279SBram Moolenaar # define Py_BuildValue dll_Py_BuildValue
220071d4279SBram Moolenaar # define Py_FindMethod dll_Py_FindMethod
221071d4279SBram Moolenaar # define Py_InitModule4 dll_Py_InitModule4
222644d37b8SBram Moolenaar # define Py_SetPythonHome dll_Py_SetPythonHome
223071d4279SBram Moolenaar # define Py_Initialize dll_Py_Initialize
2240e21a3f6SBram Moolenaar # define Py_Finalize dll_Py_Finalize
2250e21a3f6SBram Moolenaar # define Py_IsInitialized dll_Py_IsInitialized
226071d4279SBram Moolenaar # define _PyObject_New dll__PyObject_New
227774267bbSBram Moolenaar # define _PyObject_GC_New dll__PyObject_GC_New
228774267bbSBram Moolenaar # define PyObject_GC_Del dll_PyObject_GC_Del
229774267bbSBram Moolenaar # define PyObject_GC_UnTrack dll_PyObject_GC_UnTrack
230e721122bSBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02070000
231db913953SBram Moolenaar #  define _PyObject_NextNotImplemented (*dll__PyObject_NextNotImplemented)
232e721122bSBram Moolenaar # endif
233071d4279SBram Moolenaar # define _Py_NoneStruct (*dll__Py_NoneStruct)
23466b7985eSBram Moolenaar # define _Py_ZeroStruct (*dll__Py_ZeroStruct)
23566b7985eSBram Moolenaar # define _Py_TrueStruct (*dll__Py_TrueStruct)
236071d4279SBram Moolenaar # define PyObject_Init dll__PyObject_Init
237db913953SBram Moolenaar # define PyObject_GetIter dll_PyObject_GetIter
23803db85b3SBram Moolenaar # define PyObject_IsTrue dll_PyObject_IsTrue
239071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
240071d4279SBram Moolenaar #  define PyType_IsSubtype dll_PyType_IsSubtype
241071d4279SBram Moolenaar # endif
242071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
243071d4279SBram Moolenaar #  define PyObject_Malloc dll_PyObject_Malloc
244071d4279SBram Moolenaar #  define PyObject_Free dll_PyObject_Free
245071d4279SBram Moolenaar # endif
2462afa3238SBram Moolenaar # ifdef PY_USE_CAPSULE
247db913953SBram Moolenaar #  define PyCapsule_New dll_PyCapsule_New
248db913953SBram Moolenaar #  define PyCapsule_GetPointer dll_PyCapsule_GetPointer
2492afa3238SBram Moolenaar # else
2502afa3238SBram Moolenaar #  define PyCObject_FromVoidPtr dll_PyCObject_FromVoidPtr
2512afa3238SBram Moolenaar #  define PyCObject_AsVoidPtr dll_PyCObject_AsVoidPtr
2522afa3238SBram Moolenaar # endif
253071d4279SBram Moolenaar 
254071d4279SBram Moolenaar /*
255071d4279SBram Moolenaar  * Pointers for dynamic link
256071d4279SBram Moolenaar  */
257071d4279SBram Moolenaar static int(*dll_PyArg_Parse)(PyObject *, char *, ...);
258071d4279SBram Moolenaar static int(*dll_PyArg_ParseTuple)(PyObject *, char *, ...);
25919e60943SBram Moolenaar static int(*dll_PyMem_Free)(void *);
260db913953SBram Moolenaar static void* (*dll_PyMem_Malloc)(size_t);
261071d4279SBram Moolenaar static int(*dll_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item);
262071d4279SBram Moolenaar static int(*dll_PyErr_BadArgument)(void);
263d5f729caSBram Moolenaar static PyObject *(*dll_PyErr_NewException)(char *, PyObject *, PyObject *);
264071d4279SBram Moolenaar static void(*dll_PyErr_Clear)(void);
2654d36987cSBram Moolenaar static void(*dll_PyErr_PrintEx)(int);
266071d4279SBram Moolenaar static PyObject*(*dll_PyErr_NoMemory)(void);
267071d4279SBram Moolenaar static PyObject*(*dll_PyErr_Occurred)(void);
268071d4279SBram Moolenaar static void(*dll_PyErr_SetNone)(PyObject *);
269071d4279SBram Moolenaar static void(*dll_PyErr_SetString)(PyObject *, const char *);
2704d188da2SBram Moolenaar static void(*dll_PyErr_SetObject)(PyObject *, PyObject *);
271071d4279SBram Moolenaar static void(*dll_PyEval_InitThreads)(void);
272071d4279SBram Moolenaar static void(*dll_PyEval_RestoreThread)(PyThreadState *);
273071d4279SBram Moolenaar static PyThreadState*(*dll_PyEval_SaveThread)(void);
274071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE
275071d4279SBram Moolenaar static PyGILState_STATE	(*dll_PyGILState_Ensure)(void);
276071d4279SBram Moolenaar static void (*dll_PyGILState_Release)(PyGILState_STATE);
277071d4279SBram Moolenaar # endif
278071d4279SBram Moolenaar static long(*dll_PyInt_AsLong)(PyObject *);
279071d4279SBram Moolenaar static PyObject*(*dll_PyInt_FromLong)(long);
280db913953SBram Moolenaar static long(*dll_PyLong_AsLong)(PyObject *);
281db913953SBram Moolenaar static PyObject*(*dll_PyLong_FromLong)(long);
28266b7985eSBram Moolenaar static PyTypeObject* dll_PyBool_Type;
283071d4279SBram Moolenaar static PyTypeObject* dll_PyInt_Type;
284db913953SBram Moolenaar static PyTypeObject* dll_PyLong_Type;
2852c45e945SBram Moolenaar static PyObject*(*dll_PyList_GetItem)(PyObject *, PyInt);
2860ac9379aSBram Moolenaar static PyObject*(*dll_PyList_Append)(PyObject *, PyObject *);
2872c45e945SBram Moolenaar static PyObject*(*dll_PyList_New)(PyInt size);
2882c45e945SBram Moolenaar static int(*dll_PyList_SetItem)(PyObject *, PyInt, PyObject *);
2892c45e945SBram Moolenaar static PyInt(*dll_PyList_Size)(PyObject *);
290071d4279SBram Moolenaar static PyTypeObject* dll_PyList_Type;
291db913953SBram Moolenaar static int (*dll_PySequence_Check)(PyObject *);
292db913953SBram Moolenaar static PyInt(*dll_PySequence_Size)(PyObject *);
293db913953SBram Moolenaar static PyObject*(*dll_PySequence_GetItem)(PyObject *, PyInt);
294db913953SBram Moolenaar static PyInt(*dll_PyTuple_Size)(PyObject *);
295db913953SBram Moolenaar static PyObject*(*dll_PyTuple_GetItem)(PyObject *, PyInt);
296db913953SBram Moolenaar static PyTypeObject* dll_PyTuple_Type;
297071d4279SBram Moolenaar static PyObject*(*dll_PyImport_ImportModule)(const char *);
2980ac9379aSBram Moolenaar static PyObject*(*dll_PyDict_New)(void);
299071d4279SBram Moolenaar static PyObject*(*dll_PyDict_GetItemString)(PyObject *, const char *);
300db913953SBram Moolenaar static int (*dll_PyDict_Next)(PyObject *, Py_ssize_t *, PyObject **, PyObject **);
301db913953SBram Moolenaar # ifndef PY_NO_MAPPING_ITEMS
302db913953SBram Moolenaar static PyObject* (*dll_PyMapping_Items)(PyObject *);
303db913953SBram Moolenaar # endif
304db913953SBram Moolenaar static PyObject* (*dll_PyObject_CallMethod)(PyObject *, char *, PyObject *);
305db913953SBram Moolenaar static int (*dll_PyMapping_Check)(PyObject *);
306db913953SBram Moolenaar static PyObject* (*dll_PyIter_Next)(PyObject *);
307071d4279SBram Moolenaar static PyObject*(*dll_PyModule_GetDict)(PyObject *);
308071d4279SBram Moolenaar static int(*dll_PyRun_SimpleString)(char *);
309db913953SBram Moolenaar static PyObject *(*dll_PyRun_String)(char *, int, PyObject *, PyObject *);
310d620aa9bSBram Moolenaar static PyObject* (*dll_PyObject_GetAttrString)(PyObject *, const char *);
311d620aa9bSBram Moolenaar static PyObject* (*dll_PyObject_SetAttrString)(PyObject *, const char *, PyObject *);
312d620aa9bSBram Moolenaar static PyObject* (*dll_PyObject_CallFunctionObjArgs)(PyObject *, ...);
313071d4279SBram Moolenaar static char*(*dll_PyString_AsString)(PyObject *);
314cdab9051SBram Moolenaar static int(*dll_PyString_AsStringAndSize)(PyObject *, char **, int *);
315071d4279SBram Moolenaar static PyObject*(*dll_PyString_FromString)(const char *);
3162c45e945SBram Moolenaar static PyObject*(*dll_PyString_FromStringAndSize)(const char *, PyInt);
3172c45e945SBram Moolenaar static PyInt(*dll_PyString_Size)(PyObject *);
318071d4279SBram Moolenaar static PyTypeObject* dll_PyString_Type;
319db913953SBram Moolenaar static PyTypeObject* dll_PyUnicode_Type;
320cc3e85f1SBram Moolenaar static PyObject *(*py_PyUnicode_AsEncodedString)(PyObject *, char *, char *);
321db913953SBram Moolenaar static double(*dll_PyFloat_AsDouble)(PyObject *);
322db913953SBram Moolenaar static PyObject*(*dll_PyFloat_FromDouble)(double);
323db913953SBram Moolenaar static PyTypeObject* dll_PyFloat_Type;
324071d4279SBram Moolenaar static int(*dll_PySys_SetObject)(char *, PyObject *);
325071d4279SBram Moolenaar static int(*dll_PySys_SetArgv)(int, char **);
326071d4279SBram Moolenaar static PyTypeObject* dll_PyType_Type;
32730fec7bcSBram Moolenaar static int (*dll_PyType_Ready)(PyTypeObject *type);
328071d4279SBram Moolenaar static PyObject*(*dll_Py_BuildValue)(char *, ...);
329071d4279SBram Moolenaar static PyObject*(*dll_Py_FindMethod)(struct PyMethodDef[], PyObject *, char *);
330071d4279SBram Moolenaar static PyObject*(*dll_Py_InitModule4)(char *, struct PyMethodDef *, char *, PyObject *, int);
331db913953SBram Moolenaar static PyObject*(*dll_PyImport_AddModule)(char *);
332644d37b8SBram Moolenaar static void(*dll_Py_SetPythonHome)(char *home);
333071d4279SBram Moolenaar static void(*dll_Py_Initialize)(void);
3340e21a3f6SBram Moolenaar static void(*dll_Py_Finalize)(void);
3350e21a3f6SBram Moolenaar static int(*dll_Py_IsInitialized)(void);
336071d4279SBram Moolenaar static PyObject*(*dll__PyObject_New)(PyTypeObject *, PyObject *);
337774267bbSBram Moolenaar static PyObject*(*dll__PyObject_GC_New)(PyTypeObject *);
338774267bbSBram Moolenaar static void(*dll_PyObject_GC_Del)(void *);
339774267bbSBram Moolenaar static void(*dll_PyObject_GC_UnTrack)(void *);
340071d4279SBram Moolenaar static PyObject*(*dll__PyObject_Init)(PyObject *, PyTypeObject *);
341db913953SBram Moolenaar static PyObject* (*dll_PyObject_GetIter)(PyObject *);
34203db85b3SBram Moolenaar static int (*dll_PyObject_IsTrue)(PyObject *);
343e721122bSBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02070000
344db913953SBram Moolenaar static iternextfunc dll__PyObject_NextNotImplemented;
345e721122bSBram Moolenaar # endif
346071d4279SBram Moolenaar static PyObject* dll__Py_NoneStruct;
34766b7985eSBram Moolenaar static PyObject* _Py_ZeroStruct;
34866b7985eSBram Moolenaar static PyObject* dll__Py_TrueStruct;
349071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
350071d4279SBram Moolenaar static int (*dll_PyType_IsSubtype)(PyTypeObject *, PyTypeObject *);
351071d4279SBram Moolenaar # endif
352071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
353071d4279SBram Moolenaar static void* (*dll_PyObject_Malloc)(size_t);
354071d4279SBram Moolenaar static void (*dll_PyObject_Free)(void*);
355071d4279SBram Moolenaar # endif
3562afa3238SBram Moolenaar # ifdef PY_USE_CAPSULE
357db913953SBram Moolenaar static PyObject* (*dll_PyCapsule_New)(void *, char *, PyCapsule_Destructor);
358db913953SBram Moolenaar static void* (*dll_PyCapsule_GetPointer)(PyObject *, char *);
3592afa3238SBram Moolenaar # else
360221d6872SBram Moolenaar static PyObject* (*dll_PyCObject_FromVoidPtr)(void *cobj, void (*destr)(void *));
361221d6872SBram Moolenaar static void* (*dll_PyCObject_AsVoidPtr)(PyObject *);
3622afa3238SBram Moolenaar # endif
363071d4279SBram Moolenaar 
364071d4279SBram Moolenaar static HINSTANCE hinstPython = 0; /* Instance of python.dll */
365071d4279SBram Moolenaar 
366071d4279SBram Moolenaar /* Imported exception objects */
367071d4279SBram Moolenaar static PyObject *imp_PyExc_AttributeError;
368071d4279SBram Moolenaar static PyObject *imp_PyExc_IndexError;
369af6abb9dSBram Moolenaar static PyObject *imp_PyExc_KeyError;
370071d4279SBram Moolenaar static PyObject *imp_PyExc_KeyboardInterrupt;
371071d4279SBram Moolenaar static PyObject *imp_PyExc_TypeError;
372071d4279SBram Moolenaar static PyObject *imp_PyExc_ValueError;
3738661b178SBram Moolenaar static PyObject *imp_PyExc_RuntimeError;
374071d4279SBram Moolenaar 
375071d4279SBram Moolenaar # define PyExc_AttributeError imp_PyExc_AttributeError
376071d4279SBram Moolenaar # define PyExc_IndexError imp_PyExc_IndexError
377af6abb9dSBram Moolenaar # define PyExc_KeyError imp_PyExc_KeyError
378071d4279SBram Moolenaar # define PyExc_KeyboardInterrupt imp_PyExc_KeyboardInterrupt
379071d4279SBram Moolenaar # define PyExc_TypeError imp_PyExc_TypeError
380071d4279SBram Moolenaar # define PyExc_ValueError imp_PyExc_ValueError
3818661b178SBram Moolenaar # define PyExc_RuntimeError imp_PyExc_RuntimeError
382071d4279SBram Moolenaar 
383071d4279SBram Moolenaar /*
384071d4279SBram Moolenaar  * Table of name to function pointer of python.
385071d4279SBram Moolenaar  */
386071d4279SBram Moolenaar # define PYTHON_PROC FARPROC
387071d4279SBram Moolenaar static struct
388071d4279SBram Moolenaar {
389071d4279SBram Moolenaar     char *name;
390071d4279SBram Moolenaar     PYTHON_PROC *ptr;
391071d4279SBram Moolenaar } python_funcname_table[] =
392071d4279SBram Moolenaar {
393e8cdcef8SBram Moolenaar #ifndef PY_SSIZE_T_CLEAN
394071d4279SBram Moolenaar     {"PyArg_Parse", (PYTHON_PROC*)&dll_PyArg_Parse},
395071d4279SBram Moolenaar     {"PyArg_ParseTuple", (PYTHON_PROC*)&dll_PyArg_ParseTuple},
396e8cdcef8SBram Moolenaar     {"Py_BuildValue", (PYTHON_PROC*)&dll_Py_BuildValue},
397e8cdcef8SBram Moolenaar #else
398e8cdcef8SBram Moolenaar     {"_PyArg_Parse_SizeT", (PYTHON_PROC*)&dll_PyArg_Parse},
399e8cdcef8SBram Moolenaar     {"_PyArg_ParseTuple_SizeT", (PYTHON_PROC*)&dll_PyArg_ParseTuple},
400e8cdcef8SBram Moolenaar     {"_Py_BuildValue_SizeT", (PYTHON_PROC*)&dll_Py_BuildValue},
401e8cdcef8SBram Moolenaar #endif
40219e60943SBram Moolenaar     {"PyMem_Free", (PYTHON_PROC*)&dll_PyMem_Free},
403db913953SBram Moolenaar     {"PyMem_Malloc", (PYTHON_PROC*)&dll_PyMem_Malloc},
404071d4279SBram Moolenaar     {"PyDict_SetItemString", (PYTHON_PROC*)&dll_PyDict_SetItemString},
405071d4279SBram Moolenaar     {"PyErr_BadArgument", (PYTHON_PROC*)&dll_PyErr_BadArgument},
406d5f729caSBram Moolenaar     {"PyErr_NewException", (PYTHON_PROC*)&dll_PyErr_NewException},
407071d4279SBram Moolenaar     {"PyErr_Clear", (PYTHON_PROC*)&dll_PyErr_Clear},
4084d36987cSBram Moolenaar     {"PyErr_PrintEx", (PYTHON_PROC*)&dll_PyErr_PrintEx},
409071d4279SBram Moolenaar     {"PyErr_NoMemory", (PYTHON_PROC*)&dll_PyErr_NoMemory},
410071d4279SBram Moolenaar     {"PyErr_Occurred", (PYTHON_PROC*)&dll_PyErr_Occurred},
411071d4279SBram Moolenaar     {"PyErr_SetNone", (PYTHON_PROC*)&dll_PyErr_SetNone},
412071d4279SBram Moolenaar     {"PyErr_SetString", (PYTHON_PROC*)&dll_PyErr_SetString},
4134d188da2SBram Moolenaar     {"PyErr_SetObject", (PYTHON_PROC*)&dll_PyErr_SetObject},
414071d4279SBram Moolenaar     {"PyEval_InitThreads", (PYTHON_PROC*)&dll_PyEval_InitThreads},
415071d4279SBram Moolenaar     {"PyEval_RestoreThread", (PYTHON_PROC*)&dll_PyEval_RestoreThread},
416071d4279SBram Moolenaar     {"PyEval_SaveThread", (PYTHON_PROC*)&dll_PyEval_SaveThread},
417071d4279SBram Moolenaar # ifdef PY_CAN_RECURSE
418071d4279SBram Moolenaar     {"PyGILState_Ensure", (PYTHON_PROC*)&dll_PyGILState_Ensure},
419071d4279SBram Moolenaar     {"PyGILState_Release", (PYTHON_PROC*)&dll_PyGILState_Release},
420071d4279SBram Moolenaar # endif
421071d4279SBram Moolenaar     {"PyInt_AsLong", (PYTHON_PROC*)&dll_PyInt_AsLong},
422071d4279SBram Moolenaar     {"PyInt_FromLong", (PYTHON_PROC*)&dll_PyInt_FromLong},
423db913953SBram Moolenaar     {"PyLong_AsLong", (PYTHON_PROC*)&dll_PyLong_AsLong},
424db913953SBram Moolenaar     {"PyLong_FromLong", (PYTHON_PROC*)&dll_PyLong_FromLong},
42566b7985eSBram Moolenaar     {"PyBool_Type", (PYTHON_PROC*)&dll_PyBool_Type},
426071d4279SBram Moolenaar     {"PyInt_Type", (PYTHON_PROC*)&dll_PyInt_Type},
427db913953SBram Moolenaar     {"PyLong_Type", (PYTHON_PROC*)&dll_PyLong_Type},
428071d4279SBram Moolenaar     {"PyList_GetItem", (PYTHON_PROC*)&dll_PyList_GetItem},
4290ac9379aSBram Moolenaar     {"PyList_Append", (PYTHON_PROC*)&dll_PyList_Append},
430071d4279SBram Moolenaar     {"PyList_New", (PYTHON_PROC*)&dll_PyList_New},
431071d4279SBram Moolenaar     {"PyList_SetItem", (PYTHON_PROC*)&dll_PyList_SetItem},
432071d4279SBram Moolenaar     {"PyList_Size", (PYTHON_PROC*)&dll_PyList_Size},
433071d4279SBram Moolenaar     {"PyList_Type", (PYTHON_PROC*)&dll_PyList_Type},
434db913953SBram Moolenaar     {"PySequence_GetItem", (PYTHON_PROC*)&dll_PySequence_GetItem},
435db913953SBram Moolenaar     {"PySequence_Size", (PYTHON_PROC*)&dll_PySequence_Size},
436db913953SBram Moolenaar     {"PySequence_Check", (PYTHON_PROC*)&dll_PySequence_Check},
437db913953SBram Moolenaar     {"PyTuple_GetItem", (PYTHON_PROC*)&dll_PyTuple_GetItem},
438db913953SBram Moolenaar     {"PyTuple_Size", (PYTHON_PROC*)&dll_PyTuple_Size},
439db913953SBram Moolenaar     {"PyTuple_Type", (PYTHON_PROC*)&dll_PyTuple_Type},
440071d4279SBram Moolenaar     {"PyImport_ImportModule", (PYTHON_PROC*)&dll_PyImport_ImportModule},
441071d4279SBram Moolenaar     {"PyDict_GetItemString", (PYTHON_PROC*)&dll_PyDict_GetItemString},
442db913953SBram Moolenaar     {"PyDict_Next", (PYTHON_PROC*)&dll_PyDict_Next},
4430ac9379aSBram Moolenaar     {"PyDict_New", (PYTHON_PROC*)&dll_PyDict_New},
444db913953SBram Moolenaar # ifndef PY_NO_MAPPING_ITEMS
445db913953SBram Moolenaar     {"PyMapping_Items", (PYTHON_PROC*)&dll_PyMapping_Items},
446db913953SBram Moolenaar # endif
447db913953SBram Moolenaar     {"PyObject_CallMethod", (PYTHON_PROC*)&dll_PyObject_CallMethod},
448db913953SBram Moolenaar     {"PyMapping_Check", (PYTHON_PROC*)&dll_PyMapping_Check},
449db913953SBram Moolenaar     {"PyIter_Next", (PYTHON_PROC*)&dll_PyIter_Next},
450071d4279SBram Moolenaar     {"PyModule_GetDict", (PYTHON_PROC*)&dll_PyModule_GetDict},
451071d4279SBram Moolenaar     {"PyRun_SimpleString", (PYTHON_PROC*)&dll_PyRun_SimpleString},
452db913953SBram Moolenaar     {"PyRun_String", (PYTHON_PROC*)&dll_PyRun_String},
453d620aa9bSBram Moolenaar     {"PyObject_GetAttrString", (PYTHON_PROC*)&dll_PyObject_GetAttrString},
454d620aa9bSBram Moolenaar     {"PyObject_SetAttrString", (PYTHON_PROC*)&dll_PyObject_SetAttrString},
455d620aa9bSBram Moolenaar     {"PyObject_CallFunctionObjArgs", (PYTHON_PROC*)&dll_PyObject_CallFunctionObjArgs},
456071d4279SBram Moolenaar     {"PyString_AsString", (PYTHON_PROC*)&dll_PyString_AsString},
457cdab9051SBram Moolenaar     {"PyString_AsStringAndSize", (PYTHON_PROC*)&dll_PyString_AsStringAndSize},
458071d4279SBram Moolenaar     {"PyString_FromString", (PYTHON_PROC*)&dll_PyString_FromString},
459071d4279SBram Moolenaar     {"PyString_FromStringAndSize", (PYTHON_PROC*)&dll_PyString_FromStringAndSize},
460071d4279SBram Moolenaar     {"PyString_Size", (PYTHON_PROC*)&dll_PyString_Size},
461071d4279SBram Moolenaar     {"PyString_Type", (PYTHON_PROC*)&dll_PyString_Type},
462db913953SBram Moolenaar     {"PyUnicode_Type", (PYTHON_PROC*)&dll_PyUnicode_Type},
463db913953SBram Moolenaar     {"PyFloat_Type", (PYTHON_PROC*)&dll_PyFloat_Type},
464db913953SBram Moolenaar     {"PyFloat_AsDouble", (PYTHON_PROC*)&dll_PyFloat_AsDouble},
465db913953SBram Moolenaar     {"PyFloat_FromDouble", (PYTHON_PROC*)&dll_PyFloat_FromDouble},
466db913953SBram Moolenaar     {"PyImport_AddModule", (PYTHON_PROC*)&dll_PyImport_AddModule},
467071d4279SBram Moolenaar     {"PySys_SetObject", (PYTHON_PROC*)&dll_PySys_SetObject},
468071d4279SBram Moolenaar     {"PySys_SetArgv", (PYTHON_PROC*)&dll_PySys_SetArgv},
469071d4279SBram Moolenaar     {"PyType_Type", (PYTHON_PROC*)&dll_PyType_Type},
47030fec7bcSBram Moolenaar     {"PyType_Ready", (PYTHON_PROC*)&dll_PyType_Ready},
471071d4279SBram Moolenaar     {"Py_FindMethod", (PYTHON_PROC*)&dll_Py_FindMethod},
4722afa3238SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000 \
4732afa3238SBram Moolenaar 	&& SIZEOF_SIZE_T != SIZEOF_INT
474e7cb9cf6SBram Moolenaar     {"Py_InitModule4_64", (PYTHON_PROC*)&dll_Py_InitModule4},
475e7cb9cf6SBram Moolenaar # else
476071d4279SBram Moolenaar     {"Py_InitModule4", (PYTHON_PROC*)&dll_Py_InitModule4},
477e7cb9cf6SBram Moolenaar # endif
478644d37b8SBram Moolenaar     {"Py_SetPythonHome", (PYTHON_PROC*)&dll_Py_SetPythonHome},
479071d4279SBram Moolenaar     {"Py_Initialize", (PYTHON_PROC*)&dll_Py_Initialize},
4800e21a3f6SBram Moolenaar     {"Py_Finalize", (PYTHON_PROC*)&dll_Py_Finalize},
4810e21a3f6SBram Moolenaar     {"Py_IsInitialized", (PYTHON_PROC*)&dll_Py_IsInitialized},
482071d4279SBram Moolenaar     {"_PyObject_New", (PYTHON_PROC*)&dll__PyObject_New},
483774267bbSBram Moolenaar     {"_PyObject_GC_New", (PYTHON_PROC*)&dll__PyObject_GC_New},
484774267bbSBram Moolenaar     {"PyObject_GC_Del", (PYTHON_PROC*)&dll_PyObject_GC_Del},
485774267bbSBram Moolenaar     {"PyObject_GC_UnTrack", (PYTHON_PROC*)&dll_PyObject_GC_UnTrack},
486071d4279SBram Moolenaar     {"PyObject_Init", (PYTHON_PROC*)&dll__PyObject_Init},
487db913953SBram Moolenaar     {"PyObject_GetIter", (PYTHON_PROC*)&dll_PyObject_GetIter},
48803db85b3SBram Moolenaar     {"PyObject_IsTrue", (PYTHON_PROC*)&dll_PyObject_IsTrue},
489e721122bSBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02070000
490db913953SBram Moolenaar     {"_PyObject_NextNotImplemented", (PYTHON_PROC*)&dll__PyObject_NextNotImplemented},
491e721122bSBram Moolenaar # endif
492071d4279SBram Moolenaar     {"_Py_NoneStruct", (PYTHON_PROC*)&dll__Py_NoneStruct},
49366b7985eSBram Moolenaar     {"_Py_ZeroStruct", (PYTHON_PROC*)&dll__Py_ZeroStruct},
49466b7985eSBram Moolenaar     {"_Py_TrueStruct", (PYTHON_PROC*)&dll__Py_TrueStruct},
495071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
496071d4279SBram Moolenaar     {"PyType_IsSubtype", (PYTHON_PROC*)&dll_PyType_IsSubtype},
497071d4279SBram Moolenaar # endif
498071d4279SBram Moolenaar # if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
499071d4279SBram Moolenaar     {"PyObject_Malloc", (PYTHON_PROC*)&dll_PyObject_Malloc},
500071d4279SBram Moolenaar     {"PyObject_Free", (PYTHON_PROC*)&dll_PyObject_Free},
501071d4279SBram Moolenaar # endif
5022afa3238SBram Moolenaar # ifdef PY_USE_CAPSULE
503db913953SBram Moolenaar     {"PyCapsule_New", (PYTHON_PROC*)&dll_PyCapsule_New},
504db913953SBram Moolenaar     {"PyCapsule_GetPointer", (PYTHON_PROC*)&dll_PyCapsule_GetPointer},
5052afa3238SBram Moolenaar # else
5062afa3238SBram Moolenaar     {"PyCObject_FromVoidPtr", (PYTHON_PROC*)&dll_PyCObject_FromVoidPtr},
5072afa3238SBram Moolenaar     {"PyCObject_AsVoidPtr", (PYTHON_PROC*)&dll_PyCObject_AsVoidPtr},
5082afa3238SBram Moolenaar # endif
509071d4279SBram Moolenaar     {"", NULL},
510071d4279SBram Moolenaar };
511071d4279SBram Moolenaar 
512071d4279SBram Moolenaar /*
513071d4279SBram Moolenaar  * Free python.dll
514071d4279SBram Moolenaar  */
515071d4279SBram Moolenaar     static void
516071d4279SBram Moolenaar end_dynamic_python(void)
517071d4279SBram Moolenaar {
518071d4279SBram Moolenaar     if (hinstPython)
519071d4279SBram Moolenaar     {
520bd5e15fdSBram Moolenaar 	close_dll(hinstPython);
521071d4279SBram Moolenaar 	hinstPython = 0;
522071d4279SBram Moolenaar     }
523071d4279SBram Moolenaar }
524071d4279SBram Moolenaar 
525071d4279SBram Moolenaar /*
526071d4279SBram Moolenaar  * Load library and get all pointers.
527071d4279SBram Moolenaar  * Parameter 'libname' provides name of DLL.
528071d4279SBram Moolenaar  * Return OK or FAIL.
529071d4279SBram Moolenaar  */
530071d4279SBram Moolenaar     static int
531071d4279SBram Moolenaar python_runtime_link_init(char *libname, int verbose)
532071d4279SBram Moolenaar {
533071d4279SBram Moolenaar     int i;
534cc3e85f1SBram Moolenaar     void *ucs_as_encoded_string;
535071d4279SBram Moolenaar 
536644d37b8SBram Moolenaar #if !(defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)) && defined(UNIX) && defined(FEAT_PYTHON3)
537b744b2faSBram Moolenaar     /* Can't have Python and Python3 loaded at the same time.
538b744b2faSBram Moolenaar      * It cause a crash, because RTLD_GLOBAL is needed for
539b744b2faSBram Moolenaar      * standard C extension libraries of one or both python versions. */
5404c3a326cSBram Moolenaar     if (python3_loaded())
5414c3a326cSBram Moolenaar     {
5429dc93ae4SBram Moolenaar 	if (verbose)
543b744b2faSBram Moolenaar 	    EMSG(_("E836: This Vim cannot execute :python after using :py3"));
5444c3a326cSBram Moolenaar 	return FAIL;
5454c3a326cSBram Moolenaar     }
5464c3a326cSBram Moolenaar #endif
5474c3a326cSBram Moolenaar 
548071d4279SBram Moolenaar     if (hinstPython)
549071d4279SBram Moolenaar 	return OK;
550bd5e15fdSBram Moolenaar     hinstPython = load_dll(libname);
551071d4279SBram Moolenaar     if (!hinstPython)
552071d4279SBram Moolenaar     {
553071d4279SBram Moolenaar 	if (verbose)
554071d4279SBram Moolenaar 	    EMSG2(_(e_loadlib), libname);
555071d4279SBram Moolenaar 	return FAIL;
556071d4279SBram Moolenaar     }
557071d4279SBram Moolenaar 
558071d4279SBram Moolenaar     for (i = 0; python_funcname_table[i].ptr; ++i)
559071d4279SBram Moolenaar     {
560bd5e15fdSBram Moolenaar 	if ((*python_funcname_table[i].ptr = symbol_from_dll(hinstPython,
561071d4279SBram Moolenaar 			python_funcname_table[i].name)) == NULL)
562071d4279SBram Moolenaar 	{
563bd5e15fdSBram Moolenaar 	    close_dll(hinstPython);
564071d4279SBram Moolenaar 	    hinstPython = 0;
565071d4279SBram Moolenaar 	    if (verbose)
566071d4279SBram Moolenaar 		EMSG2(_(e_loadfunc), python_funcname_table[i].name);
567071d4279SBram Moolenaar 	    return FAIL;
568071d4279SBram Moolenaar 	}
569071d4279SBram Moolenaar     }
570cc3e85f1SBram Moolenaar 
571cc3e85f1SBram Moolenaar     /* Load unicode functions separately as only the ucs2 or the ucs4 functions
572cc3e85f1SBram Moolenaar      * will be present in the library. */
573cc3e85f1SBram Moolenaar     ucs_as_encoded_string = symbol_from_dll(hinstPython,
574cc3e85f1SBram Moolenaar 					     "PyUnicodeUCS2_AsEncodedString");
575cc3e85f1SBram Moolenaar     if (ucs_as_encoded_string == NULL)
576cc3e85f1SBram Moolenaar 	ucs_as_encoded_string = symbol_from_dll(hinstPython,
577cc3e85f1SBram Moolenaar 					     "PyUnicodeUCS4_AsEncodedString");
578cc3e85f1SBram Moolenaar     if (ucs_as_encoded_string != NULL)
579cc3e85f1SBram Moolenaar 	py_PyUnicode_AsEncodedString = ucs_as_encoded_string;
580cc3e85f1SBram Moolenaar     else
581cc3e85f1SBram Moolenaar     {
582cc3e85f1SBram Moolenaar 	close_dll(hinstPython);
583cc3e85f1SBram Moolenaar 	hinstPython = 0;
584cc3e85f1SBram Moolenaar 	if (verbose)
585cc3e85f1SBram Moolenaar 	    EMSG2(_(e_loadfunc), "PyUnicode_UCSX_*");
586cc3e85f1SBram Moolenaar 	return FAIL;
587cc3e85f1SBram Moolenaar     }
588cc3e85f1SBram Moolenaar 
589071d4279SBram Moolenaar     return OK;
590071d4279SBram Moolenaar }
591071d4279SBram Moolenaar 
592071d4279SBram Moolenaar /*
593071d4279SBram Moolenaar  * If python is enabled (there is installed python on Windows system) return
594071d4279SBram Moolenaar  * TRUE, else FALSE.
595071d4279SBram Moolenaar  */
596071d4279SBram Moolenaar     int
597e7cb9cf6SBram Moolenaar python_enabled(int verbose)
598071d4279SBram Moolenaar {
599071d4279SBram Moolenaar     return python_runtime_link_init(DYNAMIC_PYTHON_DLL, verbose) == OK;
600071d4279SBram Moolenaar }
601071d4279SBram Moolenaar 
602ca8a4dfeSBram Moolenaar /*
603ca8a4dfeSBram Moolenaar  * Load the standard Python exceptions - don't import the symbols from the
604071d4279SBram Moolenaar  * DLL, as this can cause errors (importing data symbols is not reliable).
605071d4279SBram Moolenaar  */
606071d4279SBram Moolenaar     static void
607ca8a4dfeSBram Moolenaar get_exceptions(void)
608071d4279SBram Moolenaar {
609071d4279SBram Moolenaar     PyObject *exmod = PyImport_ImportModule("exceptions");
610071d4279SBram Moolenaar     PyObject *exdict = PyModule_GetDict(exmod);
611071d4279SBram Moolenaar     imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError");
612071d4279SBram Moolenaar     imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError");
613af6abb9dSBram Moolenaar     imp_PyExc_KeyError = PyDict_GetItemString(exdict, "KeyError");
614071d4279SBram Moolenaar     imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt");
615071d4279SBram Moolenaar     imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError");
616071d4279SBram Moolenaar     imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError");
6178661b178SBram Moolenaar     imp_PyExc_RuntimeError = PyDict_GetItemString(exdict, "RuntimeError");
618071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_AttributeError);
619071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_IndexError);
620af6abb9dSBram Moolenaar     Py_XINCREF(imp_PyExc_KeyError);
621071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_KeyboardInterrupt);
622071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_TypeError);
623071d4279SBram Moolenaar     Py_XINCREF(imp_PyExc_ValueError);
6248661b178SBram Moolenaar     Py_XINCREF(imp_PyExc_RuntimeError);
625071d4279SBram Moolenaar     Py_XDECREF(exmod);
626071d4279SBram Moolenaar }
627071d4279SBram Moolenaar #endif /* DYNAMIC_PYTHON */
628071d4279SBram Moolenaar 
629db913953SBram Moolenaar static int initialised = 0;
630db913953SBram Moolenaar #define PYINITIALISED initialised
631db913953SBram Moolenaar 
632db913953SBram Moolenaar #define DICTKEY_GET(err) \
633db913953SBram Moolenaar     if (!PyString_Check(keyObject)) \
634db913953SBram Moolenaar     { \
635db913953SBram Moolenaar 	PyErr_SetString(PyExc_TypeError, _("only string keys are allowed")); \
636db913953SBram Moolenaar 	return err; \
637db913953SBram Moolenaar     } \
638cdab9051SBram Moolenaar     if (PyString_AsStringAndSize(keyObject, (char **) &key, NULL) == -1) \
639cdab9051SBram Moolenaar 	return err;
640cdab9051SBram Moolenaar 
641db913953SBram Moolenaar #define DICTKEY_UNREF
642db913953SBram Moolenaar #define DICTKEY_DECL
643db913953SBram Moolenaar 
644*2a0f3d3fSBram Moolenaar #define DESTRUCTOR_FINISH(self) self->ob_type->tp_free((PyObject*)self);
6454d1da49cSBram Moolenaar 
646971db467SBram Moolenaar #define WIN_PYTHON_REF(win) win->w_python_ref
647971db467SBram Moolenaar #define BUF_PYTHON_REF(buf) buf->b_python_ref
6485e538ecdSBram Moolenaar #define TAB_PYTHON_REF(tab) tab->tp_python_ref
649971db467SBram Moolenaar 
6504d1da49cSBram Moolenaar static PyObject *OutputGetattr(PyObject *, char *);
6514d1da49cSBram Moolenaar static PyObject *BufferGetattr(PyObject *, char *);
6524d1da49cSBram Moolenaar static PyObject *WindowGetattr(PyObject *, char *);
6535e538ecdSBram Moolenaar static PyObject *TabPageGetattr(PyObject *, char *);
6544d1da49cSBram Moolenaar static PyObject *RangeGetattr(PyObject *, char *);
6554d1da49cSBram Moolenaar static PyObject *DictionaryGetattr(PyObject *, char*);
6564d1da49cSBram Moolenaar static PyObject *ListGetattr(PyObject *, char *);
6574d1da49cSBram Moolenaar static PyObject *FunctionGetattr(PyObject *, char *);
6584d1da49cSBram Moolenaar 
659*2a0f3d3fSBram Moolenaar #ifndef Py_VISIT
660*2a0f3d3fSBram Moolenaar # define Py_VISIT(obj) visit(obj, arg)
661*2a0f3d3fSBram Moolenaar #endif
662*2a0f3d3fSBram Moolenaar #ifndef Py_CLEAR
663*2a0f3d3fSBram Moolenaar # define Py_CLEAR(obj) \
664*2a0f3d3fSBram Moolenaar     Py_XDECREF(obj); \
665*2a0f3d3fSBram Moolenaar     obj = NULL;
666*2a0f3d3fSBram Moolenaar #endif
667*2a0f3d3fSBram Moolenaar 
668170bf1aeSBram Moolenaar /*
669170bf1aeSBram Moolenaar  * Include the code shared with if_python3.c
670170bf1aeSBram Moolenaar  */
671170bf1aeSBram Moolenaar #include "if_py_both.h"
672170bf1aeSBram Moolenaar 
673170bf1aeSBram Moolenaar 
674071d4279SBram Moolenaar /******************************************************
675071d4279SBram Moolenaar  * Internal function prototypes.
676071d4279SBram Moolenaar  */
677071d4279SBram Moolenaar 
678071d4279SBram Moolenaar static int PythonMod_Init(void);
679071d4279SBram Moolenaar 
680071d4279SBram Moolenaar 
681071d4279SBram Moolenaar /******************************************************
682071d4279SBram Moolenaar  * 1. Python interpreter main program.
683071d4279SBram Moolenaar  */
684071d4279SBram Moolenaar 
685071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */
686071d4279SBram Moolenaar typedef PyObject PyThreadState;
6879ba0eb85SBram Moolenaar #endif
688071d4279SBram Moolenaar 
68971700b89SBram Moolenaar #ifndef PY_CAN_RECURSE
690071d4279SBram Moolenaar static PyThreadState *saved_python_thread = NULL;
691071d4279SBram Moolenaar 
692071d4279SBram Moolenaar /*
693071d4279SBram Moolenaar  * Suspend a thread of the Python interpreter, other threads are allowed to
694071d4279SBram Moolenaar  * run.
695071d4279SBram Moolenaar  */
696293ee4d4SBram Moolenaar     static void
697293ee4d4SBram Moolenaar Python_SaveThread(void)
698071d4279SBram Moolenaar {
699071d4279SBram Moolenaar     saved_python_thread = PyEval_SaveThread();
700071d4279SBram Moolenaar }
701071d4279SBram Moolenaar 
702071d4279SBram Moolenaar /*
703071d4279SBram Moolenaar  * Restore a thread of the Python interpreter, waits for other threads to
704071d4279SBram Moolenaar  * block.
705071d4279SBram Moolenaar  */
706293ee4d4SBram Moolenaar     static void
707293ee4d4SBram Moolenaar Python_RestoreThread(void)
708071d4279SBram Moolenaar {
709071d4279SBram Moolenaar     PyEval_RestoreThread(saved_python_thread);
710071d4279SBram Moolenaar     saved_python_thread = NULL;
7119ba0eb85SBram Moolenaar }
71271700b89SBram Moolenaar #endif
713071d4279SBram Moolenaar 
714071d4279SBram Moolenaar     void
715071d4279SBram Moolenaar python_end()
716071d4279SBram Moolenaar {
717a5792f58SBram Moolenaar     static int recurse = 0;
718a5792f58SBram Moolenaar 
719a5792f58SBram Moolenaar     /* If a crash occurs while doing this, don't try again. */
720a5792f58SBram Moolenaar     if (recurse != 0)
721a5792f58SBram Moolenaar 	return;
722a5792f58SBram Moolenaar 
723a5792f58SBram Moolenaar     ++recurse;
724a5792f58SBram Moolenaar 
725071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
7260e21a3f6SBram Moolenaar     if (hinstPython && Py_IsInitialized())
7279ba0eb85SBram Moolenaar     {
72871700b89SBram Moolenaar # ifdef PY_CAN_RECURSE
72971700b89SBram Moolenaar 	PyGILState_Ensure();
73071700b89SBram Moolenaar # else
7319ba0eb85SBram Moolenaar 	Python_RestoreThread();	    /* enter python */
73271700b89SBram Moolenaar # endif
7330e21a3f6SBram Moolenaar 	Py_Finalize();
7349ba0eb85SBram Moolenaar     }
735071d4279SBram Moolenaar     end_dynamic_python();
7360e21a3f6SBram Moolenaar #else
7370e21a3f6SBram Moolenaar     if (Py_IsInitialized())
7389ba0eb85SBram Moolenaar     {
73971700b89SBram Moolenaar # ifdef PY_CAN_RECURSE
74071700b89SBram Moolenaar 	PyGILState_Ensure();
74171700b89SBram Moolenaar # else
7429ba0eb85SBram Moolenaar 	Python_RestoreThread();	    /* enter python */
74371700b89SBram Moolenaar # endif
7440e21a3f6SBram Moolenaar 	Py_Finalize();
7459ba0eb85SBram Moolenaar     }
746071d4279SBram Moolenaar #endif
747a5792f58SBram Moolenaar 
748a5792f58SBram Moolenaar     --recurse;
749071d4279SBram Moolenaar }
750071d4279SBram Moolenaar 
7514c3a326cSBram Moolenaar #if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
7524c3a326cSBram Moolenaar     int
7534c3a326cSBram Moolenaar python_loaded()
7544c3a326cSBram Moolenaar {
7554c3a326cSBram Moolenaar     return (hinstPython != 0);
7564c3a326cSBram Moolenaar }
7574c3a326cSBram Moolenaar #endif
7584c3a326cSBram Moolenaar 
759071d4279SBram Moolenaar     static int
760071d4279SBram Moolenaar Python_Init(void)
761071d4279SBram Moolenaar {
762071d4279SBram Moolenaar     if (!initialised)
763071d4279SBram Moolenaar     {
764071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
765071d4279SBram Moolenaar 	if (!python_enabled(TRUE))
766071d4279SBram Moolenaar 	{
767071d4279SBram Moolenaar 	    EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded."));
768071d4279SBram Moolenaar 	    goto fail;
769071d4279SBram Moolenaar 	}
770071d4279SBram Moolenaar #endif
771071d4279SBram Moolenaar 
772644d37b8SBram Moolenaar #ifdef PYTHON_HOME
773644d37b8SBram Moolenaar 	Py_SetPythonHome(PYTHON_HOME);
774644d37b8SBram Moolenaar #endif
775644d37b8SBram Moolenaar 
776170bf1aeSBram Moolenaar 	init_structs();
777170bf1aeSBram Moolenaar 
778071d4279SBram Moolenaar #if !defined(MACOS) || defined(MACOS_X_UNIX)
779071d4279SBram Moolenaar 	Py_Initialize();
780071d4279SBram Moolenaar #else
781071d4279SBram Moolenaar 	PyMac_Initialize();
782071d4279SBram Moolenaar #endif
78302366255SBram Moolenaar 	/* Initialise threads, and below save the state using
78476d711c3SBram Moolenaar 	 * PyEval_SaveThread.  Without the call to PyEval_SaveThread, thread
78502366255SBram Moolenaar 	 * specific state (such as the system trace hook), will be lost
78602366255SBram Moolenaar 	 * between invocations of Python code. */
787071d4279SBram Moolenaar 	PyEval_InitThreads();
788071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
789071d4279SBram Moolenaar 	get_exceptions();
790071d4279SBram Moolenaar #endif
791071d4279SBram Moolenaar 
7921dc28783SBram Moolenaar 	if (PythonIO_Init_io())
793071d4279SBram Moolenaar 	    goto fail;
794071d4279SBram Moolenaar 
795071d4279SBram Moolenaar 	if (PythonMod_Init())
796071d4279SBram Moolenaar 	    goto fail;
797071d4279SBram Moolenaar 
798db913953SBram Moolenaar 	globals = PyModule_GetDict(PyImport_AddModule("__main__"));
799db913953SBram Moolenaar 
8009774ecc8SBram Moolenaar 	/* Remove the element from sys.path that was added because of our
8019774ecc8SBram Moolenaar 	 * argv[0] value in PythonMod_Init().  Previously we used an empty
80284a05accSBram Moolenaar 	 * string, but depending on the OS we then get an empty entry or
8039774ecc8SBram Moolenaar 	 * the current directory in sys.path. */
8049774ecc8SBram Moolenaar 	PyRun_SimpleString("import sys; sys.path = filter(lambda x: x != '/must>not&exist', sys.path)");
8059774ecc8SBram Moolenaar 
80676d711c3SBram Moolenaar 	/* lock is created and acquired in PyEval_InitThreads() and thread
80776d711c3SBram Moolenaar 	 * state is created in Py_Initialize()
80876d711c3SBram Moolenaar 	 * there _PyGILState_NoteThreadState() also sets gilcounter to 1
80976d711c3SBram Moolenaar 	 * (python must have threads enabled!)
81076d711c3SBram Moolenaar 	 * so the following does both: unlock GIL and save thread state in TLS
81176d711c3SBram Moolenaar 	 * without deleting thread state
81276d711c3SBram Moolenaar 	 */
81303db85b3SBram Moolenaar #ifndef PY_CAN_RECURSE
81403db85b3SBram Moolenaar 	saved_python_thread =
81503db85b3SBram Moolenaar #endif
81676d711c3SBram Moolenaar 	    PyEval_SaveThread();
817071d4279SBram Moolenaar 
818071d4279SBram Moolenaar 	initialised = 1;
819071d4279SBram Moolenaar     }
820071d4279SBram Moolenaar 
821071d4279SBram Moolenaar     return 0;
822071d4279SBram Moolenaar 
823071d4279SBram Moolenaar fail:
824071d4279SBram Moolenaar     /* We call PythonIO_Flush() here to print any Python errors.
825071d4279SBram Moolenaar      * This is OK, as it is possible to call this function even
8261dc28783SBram Moolenaar      * if PythonIO_Init_io() has not completed successfully (it will
827071d4279SBram Moolenaar      * not do anything in this case).
828071d4279SBram Moolenaar      */
829071d4279SBram Moolenaar     PythonIO_Flush();
830071d4279SBram Moolenaar     return -1;
831071d4279SBram Moolenaar }
832071d4279SBram Moolenaar 
833071d4279SBram Moolenaar /*
834071d4279SBram Moolenaar  * External interface
835071d4279SBram Moolenaar  */
836071d4279SBram Moolenaar     static void
837b52f4c02SBram Moolenaar DoPyCommand(const char *cmd, rangeinitializer init_range, runner run, void *arg)
838071d4279SBram Moolenaar {
8399ba0eb85SBram Moolenaar #ifndef PY_CAN_RECURSE
840071d4279SBram Moolenaar     static int		recursive = 0;
841071d4279SBram Moolenaar #endif
842071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
843071d4279SBram Moolenaar     GrafPtr		oldPort;
844071d4279SBram Moolenaar #endif
845071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
846071d4279SBram Moolenaar     char		*saved_locale;
847071d4279SBram Moolenaar #endif
84871700b89SBram Moolenaar #ifdef PY_CAN_RECURSE
84971700b89SBram Moolenaar     PyGILState_STATE	pygilstate;
85071700b89SBram Moolenaar #endif
851071d4279SBram Moolenaar 
852071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE
853071d4279SBram Moolenaar     if (recursive)
854071d4279SBram Moolenaar     {
855071d4279SBram Moolenaar 	EMSG(_("E659: Cannot invoke Python recursively"));
856071d4279SBram Moolenaar 	return;
857071d4279SBram Moolenaar     }
858071d4279SBram Moolenaar     ++recursive;
859071d4279SBram Moolenaar #endif
860071d4279SBram Moolenaar 
861071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
862071d4279SBram Moolenaar     GetPort(&oldPort);
863071d4279SBram Moolenaar     /* Check if the Python library is available */
864071d4279SBram Moolenaar     if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress)
865071d4279SBram Moolenaar 	goto theend;
866071d4279SBram Moolenaar #endif
867071d4279SBram Moolenaar     if (Python_Init())
868071d4279SBram Moolenaar 	goto theend;
869071d4279SBram Moolenaar 
870b52f4c02SBram Moolenaar     init_range(arg);
871b52f4c02SBram Moolenaar 
872071d4279SBram Moolenaar     Python_Release_Vim();	    /* leave vim */
873071d4279SBram Moolenaar 
874071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
875071d4279SBram Moolenaar     /* Python only works properly when the LC_NUMERIC locale is "C". */
876071d4279SBram Moolenaar     saved_locale = setlocale(LC_NUMERIC, NULL);
877071d4279SBram Moolenaar     if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0)
878071d4279SBram Moolenaar 	saved_locale = NULL;
879071d4279SBram Moolenaar     else
880071d4279SBram Moolenaar     {
881071d4279SBram Moolenaar 	/* Need to make a copy, value may change when setting new locale. */
882071d4279SBram Moolenaar 	saved_locale = (char *)vim_strsave((char_u *)saved_locale);
883071d4279SBram Moolenaar 	(void)setlocale(LC_NUMERIC, "C");
884071d4279SBram Moolenaar     }
885071d4279SBram Moolenaar #endif
886071d4279SBram Moolenaar 
88771700b89SBram Moolenaar #ifdef PY_CAN_RECURSE
88871700b89SBram Moolenaar     pygilstate = PyGILState_Ensure();
88971700b89SBram Moolenaar #else
890071d4279SBram Moolenaar     Python_RestoreThread();	    /* enter python */
89171700b89SBram Moolenaar #endif
892071d4279SBram Moolenaar 
893*2a0f3d3fSBram Moolenaar     run((char *) cmd, arg
894*2a0f3d3fSBram Moolenaar #ifdef PY_CAN_RECURSE
895*2a0f3d3fSBram Moolenaar 	    , &pygilstate
896*2a0f3d3fSBram Moolenaar #endif
897*2a0f3d3fSBram Moolenaar 	    );
898071d4279SBram Moolenaar 
89971700b89SBram Moolenaar #ifdef PY_CAN_RECURSE
90071700b89SBram Moolenaar     PyGILState_Release(pygilstate);
90171700b89SBram Moolenaar #else
902071d4279SBram Moolenaar     Python_SaveThread();	    /* leave python */
90371700b89SBram Moolenaar #endif
904071d4279SBram Moolenaar 
905071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
906071d4279SBram Moolenaar     if (saved_locale != NULL)
907071d4279SBram Moolenaar     {
908071d4279SBram Moolenaar 	(void)setlocale(LC_NUMERIC, saved_locale);
909071d4279SBram Moolenaar 	vim_free(saved_locale);
910071d4279SBram Moolenaar     }
911071d4279SBram Moolenaar #endif
912071d4279SBram Moolenaar 
913071d4279SBram Moolenaar     Python_Lock_Vim();		    /* enter vim */
914071d4279SBram Moolenaar     PythonIO_Flush();
915071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
916071d4279SBram Moolenaar     SetPort(oldPort);
917071d4279SBram Moolenaar #endif
918071d4279SBram Moolenaar 
919071d4279SBram Moolenaar theend:
920071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE
921071d4279SBram Moolenaar     --recursive;
922071d4279SBram Moolenaar #endif
923db913953SBram Moolenaar     return;
924071d4279SBram Moolenaar }
925071d4279SBram Moolenaar 
926071d4279SBram Moolenaar /*
927071d4279SBram Moolenaar  * ":python"
928071d4279SBram Moolenaar  */
929071d4279SBram Moolenaar     void
930071d4279SBram Moolenaar ex_python(exarg_T *eap)
931071d4279SBram Moolenaar {
932071d4279SBram Moolenaar     char_u *script;
933071d4279SBram Moolenaar 
934071d4279SBram Moolenaar     script = script_get(eap, eap->arg);
935071d4279SBram Moolenaar     if (!eap->skip)
936071d4279SBram Moolenaar     {
937b52f4c02SBram Moolenaar 	DoPyCommand(script == NULL ? (char *) eap->arg : (char *) script,
938b52f4c02SBram Moolenaar 		(rangeinitializer) init_range_cmd,
939b52f4c02SBram Moolenaar 		(runner) run_cmd,
940b52f4c02SBram Moolenaar 		(void *) eap);
941071d4279SBram Moolenaar     }
942071d4279SBram Moolenaar     vim_free(script);
943071d4279SBram Moolenaar }
944071d4279SBram Moolenaar 
945071d4279SBram Moolenaar #define BUFFER_SIZE 1024
946071d4279SBram Moolenaar 
947071d4279SBram Moolenaar /*
948071d4279SBram Moolenaar  * ":pyfile"
949071d4279SBram Moolenaar  */
950071d4279SBram Moolenaar     void
951071d4279SBram Moolenaar ex_pyfile(exarg_T *eap)
952071d4279SBram Moolenaar {
953071d4279SBram Moolenaar     static char buffer[BUFFER_SIZE];
954071d4279SBram Moolenaar     const char *file = (char *)eap->arg;
955071d4279SBram Moolenaar     char *p;
956071d4279SBram Moolenaar 
957071d4279SBram Moolenaar     /* Have to do it like this. PyRun_SimpleFile requires you to pass a
958071d4279SBram Moolenaar      * stdio file pointer, but Vim and the Python DLL are compiled with
959071d4279SBram Moolenaar      * different options under Windows, meaning that stdio pointers aren't
960071d4279SBram Moolenaar      * compatible between the two. Yuk.
961071d4279SBram Moolenaar      *
962071d4279SBram Moolenaar      * Put the string "execfile('file')" into buffer. But, we need to
963071d4279SBram Moolenaar      * escape any backslashes or single quotes in the file name, so that
964071d4279SBram Moolenaar      * Python won't mangle the file name.
965071d4279SBram Moolenaar      */
966071d4279SBram Moolenaar     strcpy(buffer, "execfile('");
967071d4279SBram Moolenaar     p = buffer + 10; /* size of "execfile('" */
968071d4279SBram Moolenaar 
969071d4279SBram Moolenaar     while (*file && p < buffer + (BUFFER_SIZE - 3))
970071d4279SBram Moolenaar     {
971071d4279SBram Moolenaar 	if (*file == '\\' || *file == '\'')
972071d4279SBram Moolenaar 	    *p++ = '\\';
973071d4279SBram Moolenaar 	*p++ = *file++;
974071d4279SBram Moolenaar     }
975071d4279SBram Moolenaar 
976071d4279SBram Moolenaar     /* If we didn't finish the file name, we hit a buffer overflow */
977071d4279SBram Moolenaar     if (*file != '\0')
978071d4279SBram Moolenaar 	return;
979071d4279SBram Moolenaar 
980071d4279SBram Moolenaar     /* Put in the terminating "')" and a null */
981071d4279SBram Moolenaar     *p++ = '\'';
982071d4279SBram Moolenaar     *p++ = ')';
983071d4279SBram Moolenaar     *p++ = '\0';
984071d4279SBram Moolenaar 
985071d4279SBram Moolenaar     /* Execute the file */
986b52f4c02SBram Moolenaar     DoPyCommand(buffer,
987b52f4c02SBram Moolenaar 	    (rangeinitializer) init_range_cmd,
988b52f4c02SBram Moolenaar 	    (runner) run_cmd,
989b52f4c02SBram Moolenaar 	    (void *) eap);
990071d4279SBram Moolenaar }
991071d4279SBram Moolenaar 
992d620aa9bSBram Moolenaar     void
993d620aa9bSBram Moolenaar ex_pydo(exarg_T *eap)
994d620aa9bSBram Moolenaar {
995b52f4c02SBram Moolenaar     DoPyCommand((char *)eap->arg,
996b52f4c02SBram Moolenaar 	    (rangeinitializer) init_range_cmd,
997b52f4c02SBram Moolenaar 	    (runner)run_do,
998b52f4c02SBram Moolenaar 	    (void *)eap);
999d620aa9bSBram Moolenaar }
1000d620aa9bSBram Moolenaar 
1001071d4279SBram Moolenaar /******************************************************
1002071d4279SBram Moolenaar  * 2. Python output stream: writes output via [e]msg().
1003071d4279SBram Moolenaar  */
1004071d4279SBram Moolenaar 
1005071d4279SBram Moolenaar /* Implementation functions
1006071d4279SBram Moolenaar  */
1007071d4279SBram Moolenaar 
1008071d4279SBram Moolenaar     static PyObject *
1009071d4279SBram Moolenaar OutputGetattr(PyObject *self, char *name)
1010071d4279SBram Moolenaar {
1011071d4279SBram Moolenaar     if (strcmp(name, "softspace") == 0)
1012071d4279SBram Moolenaar 	return PyInt_FromLong(((OutputObject *)(self))->softspace);
1013071d4279SBram Moolenaar 
1014071d4279SBram Moolenaar     return Py_FindMethod(OutputMethods, self, name);
1015071d4279SBram Moolenaar }
1016071d4279SBram Moolenaar 
1017071d4279SBram Moolenaar /******************************************************
1018071d4279SBram Moolenaar  * 3. Implementation of the Vim module for Python
1019071d4279SBram Moolenaar  */
1020071d4279SBram Moolenaar 
1021071d4279SBram Moolenaar /* Window type - Implementation functions
1022071d4279SBram Moolenaar  * --------------------------------------
1023071d4279SBram Moolenaar  */
1024071d4279SBram Moolenaar 
1025071d4279SBram Moolenaar #define WindowType_Check(obj) ((obj)->ob_type == &WindowType)
1026071d4279SBram Moolenaar 
1027071d4279SBram Moolenaar /* Buffer type - Implementation functions
1028071d4279SBram Moolenaar  * --------------------------------------
1029071d4279SBram Moolenaar  */
1030071d4279SBram Moolenaar 
1031071d4279SBram Moolenaar #define BufferType_Check(obj) ((obj)->ob_type == &BufferType)
1032071d4279SBram Moolenaar 
10332c45e945SBram Moolenaar static PyInt BufferAssItem(PyObject *, PyInt, PyObject *);
10342c45e945SBram Moolenaar static PyInt BufferAssSlice(PyObject *, PyInt, PyInt, PyObject *);
1035071d4279SBram Moolenaar 
1036071d4279SBram Moolenaar /* Line range type - Implementation functions
1037071d4279SBram Moolenaar  * --------------------------------------
1038071d4279SBram Moolenaar  */
1039071d4279SBram Moolenaar 
1040071d4279SBram Moolenaar #define RangeType_Check(obj) ((obj)->ob_type == &RangeType)
1041071d4279SBram Moolenaar 
10422c45e945SBram Moolenaar static PyInt RangeAssItem(PyObject *, PyInt, PyObject *);
10432c45e945SBram Moolenaar static PyInt RangeAssSlice(PyObject *, PyInt, PyInt, PyObject *);
1044071d4279SBram Moolenaar 
1045071d4279SBram Moolenaar /* Current objects type - Implementation functions
1046071d4279SBram Moolenaar  * -----------------------------------------------
1047071d4279SBram Moolenaar  */
1048071d4279SBram Moolenaar 
1049071d4279SBram Moolenaar static PySequenceMethods BufferAsSeq = {
10502c45e945SBram Moolenaar     (PyInquiry)		BufferLength,	    /* sq_length,    len(x)   */
105177fceb89SBram Moolenaar     (binaryfunc)	0,		    /* BufferConcat, sq_concat, x+y */
105277fceb89SBram Moolenaar     (PyIntArgFunc)	0,		    /* BufferRepeat, sq_repeat, x*n */
10532c45e945SBram Moolenaar     (PyIntArgFunc)	BufferItem,	    /* sq_item,      x[i]     */
10542c45e945SBram Moolenaar     (PyIntIntArgFunc)	BufferSlice,	    /* sq_slice,     x[i:j]   */
10552c45e945SBram Moolenaar     (PyIntObjArgProc)	BufferAssItem,	    /* sq_ass_item,  x[i]=v   */
10562c45e945SBram Moolenaar     (PyIntIntObjArgProc) BufferAssSlice,    /* sq_ass_slice, x[i:j]=v */
10574d1da49cSBram Moolenaar     (objobjproc)	0,
10584d1da49cSBram Moolenaar #if PY_MAJOR_VERSION >= 2
10594d1da49cSBram Moolenaar     (binaryfunc)	0,
1060071d4279SBram Moolenaar     0,
10614d1da49cSBram Moolenaar #endif
1062071d4279SBram Moolenaar };
1063071d4279SBram Moolenaar 
1064071d4279SBram Moolenaar /* Buffer object - Implementation
1065071d4279SBram Moolenaar  */
1066071d4279SBram Moolenaar 
1067071d4279SBram Moolenaar     static PyObject *
1068071d4279SBram Moolenaar BufferGetattr(PyObject *self, char *name)
1069071d4279SBram Moolenaar {
10704d1da49cSBram Moolenaar     PyObject *r;
1071071d4279SBram Moolenaar 
10724d1da49cSBram Moolenaar     if (CheckBuffer((BufferObject *)(self)))
1073071d4279SBram Moolenaar 	return NULL;
1074071d4279SBram Moolenaar 
10754d1da49cSBram Moolenaar     r = BufferAttr((BufferObject *)(self), name);
10764d1da49cSBram Moolenaar     if (r || PyErr_Occurred())
10774d1da49cSBram Moolenaar 	return r;
1078071d4279SBram Moolenaar     else
1079071d4279SBram Moolenaar 	return Py_FindMethod(BufferMethods, self, name);
1080071d4279SBram Moolenaar }
1081071d4279SBram Moolenaar 
1082071d4279SBram Moolenaar /******************/
1083071d4279SBram Moolenaar 
10842c45e945SBram Moolenaar     static PyInt
10852c45e945SBram Moolenaar BufferAssItem(PyObject *self, PyInt n, PyObject *val)
1086071d4279SBram Moolenaar {
10878f1723deSBram Moolenaar     return RBAsItem((BufferObject *)(self), n, val, 1, -1, NULL);
1088071d4279SBram Moolenaar }
1089071d4279SBram Moolenaar 
10902c45e945SBram Moolenaar     static PyInt
10912c45e945SBram Moolenaar BufferAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
1092071d4279SBram Moolenaar {
10938f1723deSBram Moolenaar     return RBAsSlice((BufferObject *)(self), lo, hi, val, 1, -1, NULL);
1094071d4279SBram Moolenaar }
1095071d4279SBram Moolenaar 
1096071d4279SBram Moolenaar static PySequenceMethods RangeAsSeq = {
10972c45e945SBram Moolenaar     (PyInquiry)		RangeLength,	      /* sq_length,    len(x)   */
1098071d4279SBram Moolenaar     (binaryfunc)	0, /* RangeConcat, */ /* sq_concat,    x+y      */
10992c45e945SBram Moolenaar     (PyIntArgFunc)	0, /* RangeRepeat, */ /* sq_repeat,    x*n      */
11002c45e945SBram Moolenaar     (PyIntArgFunc)	RangeItem,	      /* sq_item,      x[i]     */
11012c45e945SBram Moolenaar     (PyIntIntArgFunc)	RangeSlice,	      /* sq_slice,     x[i:j]   */
11022c45e945SBram Moolenaar     (PyIntObjArgProc)	RangeAssItem,	      /* sq_ass_item,  x[i]=v   */
11032c45e945SBram Moolenaar     (PyIntIntObjArgProc) RangeAssSlice,	      /* sq_ass_slice, x[i:j]=v */
11044d1da49cSBram Moolenaar     (objobjproc)	0,
11054d1da49cSBram Moolenaar #if PY_MAJOR_VERSION >= 2
11064d1da49cSBram Moolenaar     (binaryfunc)	0,
11074d1da49cSBram Moolenaar     0,
11084d1da49cSBram Moolenaar #endif
1109071d4279SBram Moolenaar };
1110071d4279SBram Moolenaar 
1111071d4279SBram Moolenaar /* Line range object - Implementation
1112071d4279SBram Moolenaar  */
1113071d4279SBram Moolenaar 
1114071d4279SBram Moolenaar     static PyObject *
1115071d4279SBram Moolenaar RangeGetattr(PyObject *self, char *name)
1116071d4279SBram Moolenaar {
1117071d4279SBram Moolenaar     if (strcmp(name, "start") == 0)
1118e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->start - 1);
1119071d4279SBram Moolenaar     else if (strcmp(name, "end") == 0)
1120e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->end - 1);
1121071d4279SBram Moolenaar     else
1122071d4279SBram Moolenaar 	return Py_FindMethod(RangeMethods, self, name);
1123071d4279SBram Moolenaar }
1124071d4279SBram Moolenaar 
1125071d4279SBram Moolenaar /****************/
1126071d4279SBram Moolenaar 
11272c45e945SBram Moolenaar     static PyInt
11282c45e945SBram Moolenaar RangeAssItem(PyObject *self, PyInt n, PyObject *val)
1129071d4279SBram Moolenaar {
1130ca8a4dfeSBram Moolenaar     return RBAsItem(((RangeObject *)(self))->buf, n, val,
1131071d4279SBram Moolenaar 		     ((RangeObject *)(self))->start,
1132071d4279SBram Moolenaar 		     ((RangeObject *)(self))->end,
1133071d4279SBram Moolenaar 		     &((RangeObject *)(self))->end);
1134071d4279SBram Moolenaar }
1135071d4279SBram Moolenaar 
11362c45e945SBram Moolenaar     static PyInt
11372c45e945SBram Moolenaar RangeAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
1138071d4279SBram Moolenaar {
113919e60943SBram Moolenaar     return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val,
1140071d4279SBram Moolenaar 		      ((RangeObject *)(self))->start,
1141071d4279SBram Moolenaar 		      ((RangeObject *)(self))->end,
1142071d4279SBram Moolenaar 		      &((RangeObject *)(self))->end);
1143071d4279SBram Moolenaar }
1144071d4279SBram Moolenaar 
11455e538ecdSBram Moolenaar /* TabPage object - Implementation
11465e538ecdSBram Moolenaar  */
11475e538ecdSBram Moolenaar 
11485e538ecdSBram Moolenaar     static PyObject *
11495e538ecdSBram Moolenaar TabPageGetattr(PyObject *self, char *name)
11505e538ecdSBram Moolenaar {
11515e538ecdSBram Moolenaar     PyObject *r;
11525e538ecdSBram Moolenaar 
11535e538ecdSBram Moolenaar     if (CheckTabPage((TabPageObject *)(self)))
11545e538ecdSBram Moolenaar 	return NULL;
11555e538ecdSBram Moolenaar 
11565e538ecdSBram Moolenaar     r = TabPageAttr((TabPageObject *)(self), name);
11575e538ecdSBram Moolenaar     if (r || PyErr_Occurred())
11585e538ecdSBram Moolenaar 	return r;
11595e538ecdSBram Moolenaar     else
11605e538ecdSBram Moolenaar 	return Py_FindMethod(TabPageMethods, self, name);
11615e538ecdSBram Moolenaar }
11625e538ecdSBram Moolenaar 
1163071d4279SBram Moolenaar /* Window object - Implementation
1164071d4279SBram Moolenaar  */
1165071d4279SBram Moolenaar 
1166071d4279SBram Moolenaar     static PyObject *
1167071d4279SBram Moolenaar WindowGetattr(PyObject *self, char *name)
1168071d4279SBram Moolenaar {
11694d1da49cSBram Moolenaar     PyObject *r;
1170071d4279SBram Moolenaar 
11714d1da49cSBram Moolenaar     if (CheckWindow((WindowObject *)(self)))
1172071d4279SBram Moolenaar 	return NULL;
1173071d4279SBram Moolenaar 
11744d1da49cSBram Moolenaar     r = WindowAttr((WindowObject *)(self), name);
11754d1da49cSBram Moolenaar     if (r || PyErr_Occurred())
11764d1da49cSBram Moolenaar 	return r;
1177071d4279SBram Moolenaar     else
1178071d4279SBram Moolenaar 	return Py_FindMethod(WindowMethods, self, name);
1179071d4279SBram Moolenaar }
1180071d4279SBram Moolenaar 
11815e538ecdSBram Moolenaar /* Tab page list object - Definitions
11825e538ecdSBram Moolenaar  */
11835e538ecdSBram Moolenaar 
11845e538ecdSBram Moolenaar static PySequenceMethods TabListAsSeq = {
11855e538ecdSBram Moolenaar     (PyInquiry)		TabListLength,	    /* sq_length,    len(x)   */
11865e538ecdSBram Moolenaar     (binaryfunc)	0,		    /* sq_concat,    x+y      */
11875e538ecdSBram Moolenaar     (PyIntArgFunc)	0,		    /* sq_repeat,    x*n      */
11885e538ecdSBram Moolenaar     (PyIntArgFunc)	TabListItem,	    /* sq_item,      x[i]     */
11895e538ecdSBram Moolenaar     (PyIntIntArgFunc)	0,		    /* sq_slice,     x[i:j]   */
11905e538ecdSBram Moolenaar     (PyIntObjArgProc)	0,		    /* sq_ass_item,  x[i]=v   */
11915e538ecdSBram Moolenaar     (PyIntIntObjArgProc) 0,		    /* sq_ass_slice, x[i:j]=v */
11925e538ecdSBram Moolenaar     (objobjproc)	0,
11935e538ecdSBram Moolenaar #if PY_MAJOR_VERSION >= 2
11945e538ecdSBram Moolenaar     (binaryfunc)	0,
11955e538ecdSBram Moolenaar     0,
11965e538ecdSBram Moolenaar #endif
11975e538ecdSBram Moolenaar };
11985e538ecdSBram Moolenaar 
1199071d4279SBram Moolenaar /* Window list object - Definitions
1200071d4279SBram Moolenaar  */
1201071d4279SBram Moolenaar 
1202071d4279SBram Moolenaar static PySequenceMethods WinListAsSeq = {
12032c45e945SBram Moolenaar     (PyInquiry)		WinListLength,	    /* sq_length,    len(x)   */
1204071d4279SBram Moolenaar     (binaryfunc)	0,		    /* sq_concat,    x+y      */
12052c45e945SBram Moolenaar     (PyIntArgFunc)	0,		    /* sq_repeat,    x*n      */
12062c45e945SBram Moolenaar     (PyIntArgFunc)	WinListItem,	    /* sq_item,      x[i]     */
12072c45e945SBram Moolenaar     (PyIntIntArgFunc)	0,		    /* sq_slice,     x[i:j]   */
12082c45e945SBram Moolenaar     (PyIntObjArgProc)	0,		    /* sq_ass_item,  x[i]=v   */
12092c45e945SBram Moolenaar     (PyIntIntObjArgProc) 0,		    /* sq_ass_slice, x[i:j]=v */
12104d1da49cSBram Moolenaar     (objobjproc)	0,
12114d1da49cSBram Moolenaar #if PY_MAJOR_VERSION >= 2
12124d1da49cSBram Moolenaar     (binaryfunc)	0,
12134d1da49cSBram Moolenaar     0,
12144d1da49cSBram Moolenaar #endif
1215071d4279SBram Moolenaar };
1216071d4279SBram Moolenaar 
1217071d4279SBram Moolenaar /* External interface
1218071d4279SBram Moolenaar  */
1219071d4279SBram Moolenaar 
1220071d4279SBram Moolenaar     void
1221071d4279SBram Moolenaar python_buffer_free(buf_T *buf)
1222071d4279SBram Moolenaar {
1223971db467SBram Moolenaar     if (BUF_PYTHON_REF(buf) != NULL)
1224071d4279SBram Moolenaar     {
1225971db467SBram Moolenaar 	BufferObject *bp = BUF_PYTHON_REF(buf);
1226071d4279SBram Moolenaar 	bp->buf = INVALID_BUFFER_VALUE;
1227971db467SBram Moolenaar 	BUF_PYTHON_REF(buf) = NULL;
1228071d4279SBram Moolenaar     }
1229071d4279SBram Moolenaar }
1230071d4279SBram Moolenaar 
1231071d4279SBram Moolenaar #if defined(FEAT_WINDOWS) || defined(PROTO)
1232071d4279SBram Moolenaar     void
1233071d4279SBram Moolenaar python_window_free(win_T *win)
1234071d4279SBram Moolenaar {
1235971db467SBram Moolenaar     if (WIN_PYTHON_REF(win) != NULL)
1236071d4279SBram Moolenaar     {
1237971db467SBram Moolenaar 	WindowObject *wp = WIN_PYTHON_REF(win);
1238071d4279SBram Moolenaar 	wp->win = INVALID_WINDOW_VALUE;
1239971db467SBram Moolenaar 	WIN_PYTHON_REF(win) = NULL;
1240071d4279SBram Moolenaar     }
1241071d4279SBram Moolenaar }
12425e538ecdSBram Moolenaar 
12435e538ecdSBram Moolenaar     void
12445e538ecdSBram Moolenaar python_tabpage_free(tabpage_T *tab)
12455e538ecdSBram Moolenaar {
12465e538ecdSBram Moolenaar     if (TAB_PYTHON_REF(tab) != NULL)
12475e538ecdSBram Moolenaar     {
12485e538ecdSBram Moolenaar 	TabPageObject *tp = TAB_PYTHON_REF(tab);
12495e538ecdSBram Moolenaar 	tp->tab = INVALID_TABPAGE_VALUE;
12505e538ecdSBram Moolenaar 	TAB_PYTHON_REF(tab) = NULL;
12515e538ecdSBram Moolenaar     }
12525e538ecdSBram Moolenaar }
1253071d4279SBram Moolenaar #endif
1254071d4279SBram Moolenaar 
12551dc28783SBram Moolenaar     static int
12561dc28783SBram Moolenaar add_object(PyObject *dict, const char *name, PyObject *object)
1257071d4279SBram Moolenaar {
12581dc28783SBram Moolenaar     if (PyDict_SetItemString(dict, (char *) name, object))
12591dc28783SBram Moolenaar 	return -1;
12601dc28783SBram Moolenaar     Py_DECREF(object);
12611dc28783SBram Moolenaar     return 0;
12621dc28783SBram Moolenaar }
12635e538ecdSBram Moolenaar 
1264071d4279SBram Moolenaar     static int
1265071d4279SBram Moolenaar PythonMod_Init(void)
1266071d4279SBram Moolenaar {
1267071d4279SBram Moolenaar     PyObject *mod;
1268071d4279SBram Moolenaar     PyObject *dict;
12691dc28783SBram Moolenaar 
12709774ecc8SBram Moolenaar     /* The special value is removed from sys.path in Python_Init(). */
12719774ecc8SBram Moolenaar     static char *(argv[2]) = {"/must>not&exist/foo", NULL};
1272071d4279SBram Moolenaar 
12731dc28783SBram Moolenaar     if (init_types())
12741dc28783SBram Moolenaar 	return -1;
1275071d4279SBram Moolenaar 
1276071d4279SBram Moolenaar     /* Set sys.argv[] to avoid a crash in warn(). */
1277071d4279SBram Moolenaar     PySys_SetArgv(1, argv);
1278071d4279SBram Moolenaar 
1279e7cb9cf6SBram Moolenaar     mod = Py_InitModule4("vim", VimMethods, (char *)NULL, (PyObject *)NULL, PYTHON_API_VERSION);
1280071d4279SBram Moolenaar     dict = PyModule_GetDict(mod);
1281071d4279SBram Moolenaar 
12821dc28783SBram Moolenaar     return populate_module(dict, add_object);
1283071d4279SBram Moolenaar }
1284071d4279SBram Moolenaar 
1285071d4279SBram Moolenaar /*************************************************************************
1286071d4279SBram Moolenaar  * 4. Utility functions for handling the interface between Vim and Python.
1287071d4279SBram Moolenaar  */
1288071d4279SBram Moolenaar 
1289071d4279SBram Moolenaar /* Convert a Vim line into a Python string.
1290071d4279SBram Moolenaar  * All internal newlines are replaced by null characters.
1291071d4279SBram Moolenaar  *
1292071d4279SBram Moolenaar  * On errors, the Python exception data is set, and NULL is returned.
1293071d4279SBram Moolenaar  */
1294071d4279SBram Moolenaar     static PyObject *
1295071d4279SBram Moolenaar LineToString(const char *str)
1296071d4279SBram Moolenaar {
1297071d4279SBram Moolenaar     PyObject *result;
12982c45e945SBram Moolenaar     PyInt len = strlen(str);
1299071d4279SBram Moolenaar     char *p;
1300071d4279SBram Moolenaar 
1301071d4279SBram Moolenaar     /* Allocate an Python string object, with uninitialised contents. We
1302071d4279SBram Moolenaar      * must do it this way, so that we can modify the string in place
1303071d4279SBram Moolenaar      * later. See the Python source, Objects/stringobject.c for details.
1304071d4279SBram Moolenaar      */
1305071d4279SBram Moolenaar     result = PyString_FromStringAndSize(NULL, len);
1306071d4279SBram Moolenaar     if (result == NULL)
1307071d4279SBram Moolenaar 	return NULL;
1308071d4279SBram Moolenaar 
1309071d4279SBram Moolenaar     p = PyString_AsString(result);
1310071d4279SBram Moolenaar 
1311071d4279SBram Moolenaar     while (*str)
1312071d4279SBram Moolenaar     {
1313071d4279SBram Moolenaar 	if (*str == '\n')
1314071d4279SBram Moolenaar 	    *p = '\0';
1315071d4279SBram Moolenaar 	else
1316071d4279SBram Moolenaar 	    *p = *str;
1317071d4279SBram Moolenaar 
1318071d4279SBram Moolenaar 	++p;
1319071d4279SBram Moolenaar 	++str;
1320071d4279SBram Moolenaar     }
1321071d4279SBram Moolenaar 
1322071d4279SBram Moolenaar     return result;
1323071d4279SBram Moolenaar }
1324071d4279SBram Moolenaar 
1325db913953SBram Moolenaar     static PyObject *
1326db913953SBram Moolenaar DictionaryGetattr(PyObject *self, char *name)
1327db913953SBram Moolenaar {
132866b7985eSBram Moolenaar     DictionaryObject	*this = ((DictionaryObject *) (self));
132966b7985eSBram Moolenaar 
133066b7985eSBram Moolenaar     if (strcmp(name, "locked") == 0)
133166b7985eSBram Moolenaar 	return PyInt_FromLong(this->dict->dv_lock);
133266b7985eSBram Moolenaar     else if (strcmp(name, "scope") == 0)
133366b7985eSBram Moolenaar 	return PyInt_FromLong(this->dict->dv_scope);
133466b7985eSBram Moolenaar 
1335db913953SBram Moolenaar     return Py_FindMethod(DictionaryMethods, self, name);
1336db913953SBram Moolenaar }
1337db913953SBram Moolenaar 
1338db913953SBram Moolenaar static PySequenceMethods ListAsSeq = {
1339db913953SBram Moolenaar     (PyInquiry)			ListLength,
1340db913953SBram Moolenaar     (binaryfunc)		0,
1341db913953SBram Moolenaar     (PyIntArgFunc)		0,
1342db913953SBram Moolenaar     (PyIntArgFunc)		ListItem,
1343db913953SBram Moolenaar     (PyIntIntArgFunc)		ListSlice,
1344db913953SBram Moolenaar     (PyIntObjArgProc)		ListAssItem,
1345db913953SBram Moolenaar     (PyIntIntObjArgProc)	ListAssSlice,
1346db913953SBram Moolenaar     (objobjproc)		0,
1347db913953SBram Moolenaar #if PY_MAJOR_VERSION >= 2
1348db913953SBram Moolenaar     (binaryfunc)		ListConcatInPlace,
1349db913953SBram Moolenaar     0,
1350db913953SBram Moolenaar #endif
1351db913953SBram Moolenaar };
1352db913953SBram Moolenaar 
1353db913953SBram Moolenaar     static PyObject *
1354db913953SBram Moolenaar ListGetattr(PyObject *self, char *name)
1355db913953SBram Moolenaar {
135666b7985eSBram Moolenaar     if (strcmp(name, "locked") == 0)
135766b7985eSBram Moolenaar 	return PyInt_FromLong(((ListObject *)(self))->list->lv_lock);
135866b7985eSBram Moolenaar 
1359db913953SBram Moolenaar     return Py_FindMethod(ListMethods, self, name);
1360db913953SBram Moolenaar }
1361db913953SBram Moolenaar 
1362db913953SBram Moolenaar     static PyObject *
1363db913953SBram Moolenaar FunctionGetattr(PyObject *self, char *name)
1364db913953SBram Moolenaar {
1365db913953SBram Moolenaar     FunctionObject	*this = (FunctionObject *)(self);
1366db913953SBram Moolenaar 
1367db913953SBram Moolenaar     if (strcmp(name, "name") == 0)
1368db913953SBram Moolenaar 	return PyString_FromString((char *)(this->name));
1369db913953SBram Moolenaar     else
1370db913953SBram Moolenaar 	return Py_FindMethod(FunctionMethods, self, name);
1371db913953SBram Moolenaar }
1372db913953SBram Moolenaar 
1373db913953SBram Moolenaar     void
1374db913953SBram Moolenaar do_pyeval (char_u *str, typval_T *rettv)
1375db913953SBram Moolenaar {
1376b52f4c02SBram Moolenaar     DoPyCommand((char *) str,
1377b52f4c02SBram Moolenaar 	    (rangeinitializer) init_range_eval,
1378b52f4c02SBram Moolenaar 	    (runner) run_eval,
1379b52f4c02SBram Moolenaar 	    (void *) rettv);
1380db913953SBram Moolenaar     switch(rettv->v_type)
1381db913953SBram Moolenaar     {
1382db913953SBram Moolenaar 	case VAR_DICT: ++rettv->vval.v_dict->dv_refcount; break;
1383db913953SBram Moolenaar 	case VAR_LIST: ++rettv->vval.v_list->lv_refcount; break;
1384db913953SBram Moolenaar 	case VAR_FUNC: func_ref(rettv->vval.v_string);    break;
138577fceb89SBram Moolenaar 	case VAR_UNKNOWN:
138677fceb89SBram Moolenaar 	    rettv->v_type = VAR_NUMBER;
138777fceb89SBram Moolenaar 	    rettv->vval.v_number = 0;
138877fceb89SBram Moolenaar 	    break;
1389db913953SBram Moolenaar     }
1390db913953SBram Moolenaar }
1391071d4279SBram Moolenaar 
1392071d4279SBram Moolenaar /* Don't generate a prototype for the next function, it generates an error on
1393071d4279SBram Moolenaar  * newer Python versions. */
1394071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ && !defined(PROTO)
1395071d4279SBram Moolenaar 
1396071d4279SBram Moolenaar     char *
1397071d4279SBram Moolenaar Py_GetProgramName(void)
1398071d4279SBram Moolenaar {
1399071d4279SBram Moolenaar     return "vim";
1400071d4279SBram Moolenaar }
1401071d4279SBram Moolenaar #endif /* Python 1.4 */
1402170bf1aeSBram Moolenaar 
1403db913953SBram Moolenaar     void
1404db913953SBram Moolenaar set_ref_in_python (int copyID)
1405db913953SBram Moolenaar {
1406db913953SBram Moolenaar     set_ref_in_py(copyID);
1407db913953SBram Moolenaar }
1408