xref: /vim-8.2.3635/src/if_python.c (revision 774267bb)
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
227*774267bbSBram Moolenaar # define _PyObject_GC_New dll__PyObject_GC_New
228*774267bbSBram Moolenaar # define PyObject_GC_Del dll_PyObject_GC_Del
229*774267bbSBram 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 *);
337*774267bbSBram Moolenaar static PyObject*(*dll__PyObject_GC_New)(PyTypeObject *);
338*774267bbSBram Moolenaar static void(*dll_PyObject_GC_Del)(void *);
339*774267bbSBram 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},
483*774267bbSBram Moolenaar     {"_PyObject_GC_New", (PYTHON_PROC*)&dll__PyObject_GC_New},
484*774267bbSBram Moolenaar     {"PyObject_GC_Del", (PYTHON_PROC*)&dll_PyObject_GC_Del},
485*774267bbSBram 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*774267bbSBram Moolenaar #define DESTRUCTOR_FINISH(self) Py_TYPE(self)->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 
659170bf1aeSBram Moolenaar /*
660170bf1aeSBram Moolenaar  * Include the code shared with if_python3.c
661170bf1aeSBram Moolenaar  */
662170bf1aeSBram Moolenaar #include "if_py_both.h"
663170bf1aeSBram Moolenaar 
664170bf1aeSBram Moolenaar 
665071d4279SBram Moolenaar /******************************************************
666071d4279SBram Moolenaar  * Internal function prototypes.
667071d4279SBram Moolenaar  */
668071d4279SBram Moolenaar 
669071d4279SBram Moolenaar static int PythonMod_Init(void);
670071d4279SBram Moolenaar 
671071d4279SBram Moolenaar 
672071d4279SBram Moolenaar /******************************************************
673071d4279SBram Moolenaar  * 1. Python interpreter main program.
674071d4279SBram Moolenaar  */
675071d4279SBram Moolenaar 
676071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */
677071d4279SBram Moolenaar typedef PyObject PyThreadState;
6789ba0eb85SBram Moolenaar #endif
679071d4279SBram Moolenaar 
68071700b89SBram Moolenaar #ifndef PY_CAN_RECURSE
681071d4279SBram Moolenaar static PyThreadState *saved_python_thread = NULL;
682071d4279SBram Moolenaar 
683071d4279SBram Moolenaar /*
684071d4279SBram Moolenaar  * Suspend a thread of the Python interpreter, other threads are allowed to
685071d4279SBram Moolenaar  * run.
686071d4279SBram Moolenaar  */
687293ee4d4SBram Moolenaar     static void
688293ee4d4SBram Moolenaar Python_SaveThread(void)
689071d4279SBram Moolenaar {
690071d4279SBram Moolenaar     saved_python_thread = PyEval_SaveThread();
691071d4279SBram Moolenaar }
692071d4279SBram Moolenaar 
693071d4279SBram Moolenaar /*
694071d4279SBram Moolenaar  * Restore a thread of the Python interpreter, waits for other threads to
695071d4279SBram Moolenaar  * block.
696071d4279SBram Moolenaar  */
697293ee4d4SBram Moolenaar     static void
698293ee4d4SBram Moolenaar Python_RestoreThread(void)
699071d4279SBram Moolenaar {
700071d4279SBram Moolenaar     PyEval_RestoreThread(saved_python_thread);
701071d4279SBram Moolenaar     saved_python_thread = NULL;
7029ba0eb85SBram Moolenaar }
70371700b89SBram Moolenaar #endif
704071d4279SBram Moolenaar 
705071d4279SBram Moolenaar     void
706071d4279SBram Moolenaar python_end()
707071d4279SBram Moolenaar {
708a5792f58SBram Moolenaar     static int recurse = 0;
709a5792f58SBram Moolenaar 
710a5792f58SBram Moolenaar     /* If a crash occurs while doing this, don't try again. */
711a5792f58SBram Moolenaar     if (recurse != 0)
712a5792f58SBram Moolenaar 	return;
713a5792f58SBram Moolenaar 
714a5792f58SBram Moolenaar     ++recurse;
715a5792f58SBram Moolenaar 
716071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
7170e21a3f6SBram Moolenaar     if (hinstPython && Py_IsInitialized())
7189ba0eb85SBram Moolenaar     {
71971700b89SBram Moolenaar # ifdef PY_CAN_RECURSE
72071700b89SBram Moolenaar 	PyGILState_Ensure();
72171700b89SBram Moolenaar # else
7229ba0eb85SBram Moolenaar 	Python_RestoreThread();	    /* enter python */
72371700b89SBram Moolenaar # endif
7240e21a3f6SBram Moolenaar 	Py_Finalize();
7259ba0eb85SBram Moolenaar     }
726071d4279SBram Moolenaar     end_dynamic_python();
7270e21a3f6SBram Moolenaar #else
7280e21a3f6SBram Moolenaar     if (Py_IsInitialized())
7299ba0eb85SBram Moolenaar     {
73071700b89SBram Moolenaar # ifdef PY_CAN_RECURSE
73171700b89SBram Moolenaar 	PyGILState_Ensure();
73271700b89SBram Moolenaar # else
7339ba0eb85SBram Moolenaar 	Python_RestoreThread();	    /* enter python */
73471700b89SBram Moolenaar # endif
7350e21a3f6SBram Moolenaar 	Py_Finalize();
7369ba0eb85SBram Moolenaar     }
737071d4279SBram Moolenaar #endif
738a5792f58SBram Moolenaar 
739a5792f58SBram Moolenaar     --recurse;
740071d4279SBram Moolenaar }
741071d4279SBram Moolenaar 
7424c3a326cSBram Moolenaar #if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
7434c3a326cSBram Moolenaar     int
7444c3a326cSBram Moolenaar python_loaded()
7454c3a326cSBram Moolenaar {
7464c3a326cSBram Moolenaar     return (hinstPython != 0);
7474c3a326cSBram Moolenaar }
7484c3a326cSBram Moolenaar #endif
7494c3a326cSBram Moolenaar 
750071d4279SBram Moolenaar     static int
751071d4279SBram Moolenaar Python_Init(void)
752071d4279SBram Moolenaar {
753071d4279SBram Moolenaar     if (!initialised)
754071d4279SBram Moolenaar     {
755071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
756071d4279SBram Moolenaar 	if (!python_enabled(TRUE))
757071d4279SBram Moolenaar 	{
758071d4279SBram Moolenaar 	    EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded."));
759071d4279SBram Moolenaar 	    goto fail;
760071d4279SBram Moolenaar 	}
761071d4279SBram Moolenaar #endif
762071d4279SBram Moolenaar 
763644d37b8SBram Moolenaar #ifdef PYTHON_HOME
764644d37b8SBram Moolenaar 	Py_SetPythonHome(PYTHON_HOME);
765644d37b8SBram Moolenaar #endif
766644d37b8SBram Moolenaar 
767170bf1aeSBram Moolenaar 	init_structs();
768170bf1aeSBram Moolenaar 
769071d4279SBram Moolenaar #if !defined(MACOS) || defined(MACOS_X_UNIX)
770071d4279SBram Moolenaar 	Py_Initialize();
771071d4279SBram Moolenaar #else
772071d4279SBram Moolenaar 	PyMac_Initialize();
773071d4279SBram Moolenaar #endif
77402366255SBram Moolenaar 	/* Initialise threads, and below save the state using
77576d711c3SBram Moolenaar 	 * PyEval_SaveThread.  Without the call to PyEval_SaveThread, thread
77602366255SBram Moolenaar 	 * specific state (such as the system trace hook), will be lost
77702366255SBram Moolenaar 	 * between invocations of Python code. */
778071d4279SBram Moolenaar 	PyEval_InitThreads();
779071d4279SBram Moolenaar #ifdef DYNAMIC_PYTHON
780071d4279SBram Moolenaar 	get_exceptions();
781071d4279SBram Moolenaar #endif
782071d4279SBram Moolenaar 
7831dc28783SBram Moolenaar 	if (PythonIO_Init_io())
784071d4279SBram Moolenaar 	    goto fail;
785071d4279SBram Moolenaar 
786071d4279SBram Moolenaar 	if (PythonMod_Init())
787071d4279SBram Moolenaar 	    goto fail;
788071d4279SBram Moolenaar 
789db913953SBram Moolenaar 	globals = PyModule_GetDict(PyImport_AddModule("__main__"));
790db913953SBram Moolenaar 
7919774ecc8SBram Moolenaar 	/* Remove the element from sys.path that was added because of our
7929774ecc8SBram Moolenaar 	 * argv[0] value in PythonMod_Init().  Previously we used an empty
79384a05accSBram Moolenaar 	 * string, but depending on the OS we then get an empty entry or
7949774ecc8SBram Moolenaar 	 * the current directory in sys.path. */
7959774ecc8SBram Moolenaar 	PyRun_SimpleString("import sys; sys.path = filter(lambda x: x != '/must>not&exist', sys.path)");
7969774ecc8SBram Moolenaar 
79776d711c3SBram Moolenaar 	/* lock is created and acquired in PyEval_InitThreads() and thread
79876d711c3SBram Moolenaar 	 * state is created in Py_Initialize()
79976d711c3SBram Moolenaar 	 * there _PyGILState_NoteThreadState() also sets gilcounter to 1
80076d711c3SBram Moolenaar 	 * (python must have threads enabled!)
80176d711c3SBram Moolenaar 	 * so the following does both: unlock GIL and save thread state in TLS
80276d711c3SBram Moolenaar 	 * without deleting thread state
80376d711c3SBram Moolenaar 	 */
80403db85b3SBram Moolenaar #ifndef PY_CAN_RECURSE
80503db85b3SBram Moolenaar 	saved_python_thread =
80603db85b3SBram Moolenaar #endif
80776d711c3SBram Moolenaar 	    PyEval_SaveThread();
808071d4279SBram Moolenaar 
809071d4279SBram Moolenaar 	initialised = 1;
810071d4279SBram Moolenaar     }
811071d4279SBram Moolenaar 
812071d4279SBram Moolenaar     return 0;
813071d4279SBram Moolenaar 
814071d4279SBram Moolenaar fail:
815071d4279SBram Moolenaar     /* We call PythonIO_Flush() here to print any Python errors.
816071d4279SBram Moolenaar      * This is OK, as it is possible to call this function even
8171dc28783SBram Moolenaar      * if PythonIO_Init_io() has not completed successfully (it will
818071d4279SBram Moolenaar      * not do anything in this case).
819071d4279SBram Moolenaar      */
820071d4279SBram Moolenaar     PythonIO_Flush();
821071d4279SBram Moolenaar     return -1;
822071d4279SBram Moolenaar }
823071d4279SBram Moolenaar 
824071d4279SBram Moolenaar /*
825071d4279SBram Moolenaar  * External interface
826071d4279SBram Moolenaar  */
827071d4279SBram Moolenaar     static void
828b52f4c02SBram Moolenaar DoPyCommand(const char *cmd, rangeinitializer init_range, runner run, void *arg)
829071d4279SBram Moolenaar {
8309ba0eb85SBram Moolenaar #ifndef PY_CAN_RECURSE
831071d4279SBram Moolenaar     static int		recursive = 0;
832071d4279SBram Moolenaar #endif
833071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
834071d4279SBram Moolenaar     GrafPtr		oldPort;
835071d4279SBram Moolenaar #endif
836071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
837071d4279SBram Moolenaar     char		*saved_locale;
838071d4279SBram Moolenaar #endif
83971700b89SBram Moolenaar #ifdef PY_CAN_RECURSE
84071700b89SBram Moolenaar     PyGILState_STATE	pygilstate;
84171700b89SBram Moolenaar #endif
842071d4279SBram Moolenaar 
843071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE
844071d4279SBram Moolenaar     if (recursive)
845071d4279SBram Moolenaar     {
846071d4279SBram Moolenaar 	EMSG(_("E659: Cannot invoke Python recursively"));
847071d4279SBram Moolenaar 	return;
848071d4279SBram Moolenaar     }
849071d4279SBram Moolenaar     ++recursive;
850071d4279SBram Moolenaar #endif
851071d4279SBram Moolenaar 
852071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
853071d4279SBram Moolenaar     GetPort(&oldPort);
854071d4279SBram Moolenaar     /* Check if the Python library is available */
855071d4279SBram Moolenaar     if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress)
856071d4279SBram Moolenaar 	goto theend;
857071d4279SBram Moolenaar #endif
858071d4279SBram Moolenaar     if (Python_Init())
859071d4279SBram Moolenaar 	goto theend;
860071d4279SBram Moolenaar 
861b52f4c02SBram Moolenaar     init_range(arg);
862b52f4c02SBram Moolenaar 
863071d4279SBram Moolenaar     Python_Release_Vim();	    /* leave vim */
864071d4279SBram Moolenaar 
865071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
866071d4279SBram Moolenaar     /* Python only works properly when the LC_NUMERIC locale is "C". */
867071d4279SBram Moolenaar     saved_locale = setlocale(LC_NUMERIC, NULL);
868071d4279SBram Moolenaar     if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0)
869071d4279SBram Moolenaar 	saved_locale = NULL;
870071d4279SBram Moolenaar     else
871071d4279SBram Moolenaar     {
872071d4279SBram Moolenaar 	/* Need to make a copy, value may change when setting new locale. */
873071d4279SBram Moolenaar 	saved_locale = (char *)vim_strsave((char_u *)saved_locale);
874071d4279SBram Moolenaar 	(void)setlocale(LC_NUMERIC, "C");
875071d4279SBram Moolenaar     }
876071d4279SBram Moolenaar #endif
877071d4279SBram Moolenaar 
87871700b89SBram Moolenaar #ifdef PY_CAN_RECURSE
87971700b89SBram Moolenaar     pygilstate = PyGILState_Ensure();
88071700b89SBram Moolenaar #else
881071d4279SBram Moolenaar     Python_RestoreThread();	    /* enter python */
88271700b89SBram Moolenaar #endif
883071d4279SBram Moolenaar 
884b52f4c02SBram Moolenaar     run((char *) cmd, arg, &pygilstate);
885071d4279SBram Moolenaar 
88671700b89SBram Moolenaar #ifdef PY_CAN_RECURSE
88771700b89SBram Moolenaar     PyGILState_Release(pygilstate);
88871700b89SBram Moolenaar #else
889071d4279SBram Moolenaar     Python_SaveThread();	    /* leave python */
89071700b89SBram Moolenaar #endif
891071d4279SBram Moolenaar 
892071d4279SBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
893071d4279SBram Moolenaar     if (saved_locale != NULL)
894071d4279SBram Moolenaar     {
895071d4279SBram Moolenaar 	(void)setlocale(LC_NUMERIC, saved_locale);
896071d4279SBram Moolenaar 	vim_free(saved_locale);
897071d4279SBram Moolenaar     }
898071d4279SBram Moolenaar #endif
899071d4279SBram Moolenaar 
900071d4279SBram Moolenaar     Python_Lock_Vim();		    /* enter vim */
901071d4279SBram Moolenaar     PythonIO_Flush();
902071d4279SBram Moolenaar #if defined(MACOS) && !defined(MACOS_X_UNIX)
903071d4279SBram Moolenaar     SetPort(oldPort);
904071d4279SBram Moolenaar #endif
905071d4279SBram Moolenaar 
906071d4279SBram Moolenaar theend:
907071d4279SBram Moolenaar #ifndef PY_CAN_RECURSE
908071d4279SBram Moolenaar     --recursive;
909071d4279SBram Moolenaar #endif
910db913953SBram Moolenaar     return;
911071d4279SBram Moolenaar }
912071d4279SBram Moolenaar 
913071d4279SBram Moolenaar /*
914071d4279SBram Moolenaar  * ":python"
915071d4279SBram Moolenaar  */
916071d4279SBram Moolenaar     void
917071d4279SBram Moolenaar ex_python(exarg_T *eap)
918071d4279SBram Moolenaar {
919071d4279SBram Moolenaar     char_u *script;
920071d4279SBram Moolenaar 
921071d4279SBram Moolenaar     script = script_get(eap, eap->arg);
922071d4279SBram Moolenaar     if (!eap->skip)
923071d4279SBram Moolenaar     {
924b52f4c02SBram Moolenaar 	DoPyCommand(script == NULL ? (char *) eap->arg : (char *) script,
925b52f4c02SBram Moolenaar 		(rangeinitializer) init_range_cmd,
926b52f4c02SBram Moolenaar 		(runner) run_cmd,
927b52f4c02SBram Moolenaar 		(void *) eap);
928071d4279SBram Moolenaar     }
929071d4279SBram Moolenaar     vim_free(script);
930071d4279SBram Moolenaar }
931071d4279SBram Moolenaar 
932071d4279SBram Moolenaar #define BUFFER_SIZE 1024
933071d4279SBram Moolenaar 
934071d4279SBram Moolenaar /*
935071d4279SBram Moolenaar  * ":pyfile"
936071d4279SBram Moolenaar  */
937071d4279SBram Moolenaar     void
938071d4279SBram Moolenaar ex_pyfile(exarg_T *eap)
939071d4279SBram Moolenaar {
940071d4279SBram Moolenaar     static char buffer[BUFFER_SIZE];
941071d4279SBram Moolenaar     const char *file = (char *)eap->arg;
942071d4279SBram Moolenaar     char *p;
943071d4279SBram Moolenaar 
944071d4279SBram Moolenaar     /* Have to do it like this. PyRun_SimpleFile requires you to pass a
945071d4279SBram Moolenaar      * stdio file pointer, but Vim and the Python DLL are compiled with
946071d4279SBram Moolenaar      * different options under Windows, meaning that stdio pointers aren't
947071d4279SBram Moolenaar      * compatible between the two. Yuk.
948071d4279SBram Moolenaar      *
949071d4279SBram Moolenaar      * Put the string "execfile('file')" into buffer. But, we need to
950071d4279SBram Moolenaar      * escape any backslashes or single quotes in the file name, so that
951071d4279SBram Moolenaar      * Python won't mangle the file name.
952071d4279SBram Moolenaar      */
953071d4279SBram Moolenaar     strcpy(buffer, "execfile('");
954071d4279SBram Moolenaar     p = buffer + 10; /* size of "execfile('" */
955071d4279SBram Moolenaar 
956071d4279SBram Moolenaar     while (*file && p < buffer + (BUFFER_SIZE - 3))
957071d4279SBram Moolenaar     {
958071d4279SBram Moolenaar 	if (*file == '\\' || *file == '\'')
959071d4279SBram Moolenaar 	    *p++ = '\\';
960071d4279SBram Moolenaar 	*p++ = *file++;
961071d4279SBram Moolenaar     }
962071d4279SBram Moolenaar 
963071d4279SBram Moolenaar     /* If we didn't finish the file name, we hit a buffer overflow */
964071d4279SBram Moolenaar     if (*file != '\0')
965071d4279SBram Moolenaar 	return;
966071d4279SBram Moolenaar 
967071d4279SBram Moolenaar     /* Put in the terminating "')" and a null */
968071d4279SBram Moolenaar     *p++ = '\'';
969071d4279SBram Moolenaar     *p++ = ')';
970071d4279SBram Moolenaar     *p++ = '\0';
971071d4279SBram Moolenaar 
972071d4279SBram Moolenaar     /* Execute the file */
973b52f4c02SBram Moolenaar     DoPyCommand(buffer,
974b52f4c02SBram Moolenaar 	    (rangeinitializer) init_range_cmd,
975b52f4c02SBram Moolenaar 	    (runner) run_cmd,
976b52f4c02SBram Moolenaar 	    (void *) eap);
977071d4279SBram Moolenaar }
978071d4279SBram Moolenaar 
979d620aa9bSBram Moolenaar     void
980d620aa9bSBram Moolenaar ex_pydo(exarg_T *eap)
981d620aa9bSBram Moolenaar {
982b52f4c02SBram Moolenaar     DoPyCommand((char *)eap->arg,
983b52f4c02SBram Moolenaar 	    (rangeinitializer) init_range_cmd,
984b52f4c02SBram Moolenaar 	    (runner)run_do,
985b52f4c02SBram Moolenaar 	    (void *)eap);
986d620aa9bSBram Moolenaar }
987d620aa9bSBram Moolenaar 
988071d4279SBram Moolenaar /******************************************************
989071d4279SBram Moolenaar  * 2. Python output stream: writes output via [e]msg().
990071d4279SBram Moolenaar  */
991071d4279SBram Moolenaar 
992071d4279SBram Moolenaar /* Implementation functions
993071d4279SBram Moolenaar  */
994071d4279SBram Moolenaar 
995071d4279SBram Moolenaar     static PyObject *
996071d4279SBram Moolenaar OutputGetattr(PyObject *self, char *name)
997071d4279SBram Moolenaar {
998071d4279SBram Moolenaar     if (strcmp(name, "softspace") == 0)
999071d4279SBram Moolenaar 	return PyInt_FromLong(((OutputObject *)(self))->softspace);
1000071d4279SBram Moolenaar 
1001071d4279SBram Moolenaar     return Py_FindMethod(OutputMethods, self, name);
1002071d4279SBram Moolenaar }
1003071d4279SBram Moolenaar 
1004071d4279SBram Moolenaar /******************************************************
1005071d4279SBram Moolenaar  * 3. Implementation of the Vim module for Python
1006071d4279SBram Moolenaar  */
1007071d4279SBram Moolenaar 
1008071d4279SBram Moolenaar /* Window type - Implementation functions
1009071d4279SBram Moolenaar  * --------------------------------------
1010071d4279SBram Moolenaar  */
1011071d4279SBram Moolenaar 
1012071d4279SBram Moolenaar #define WindowType_Check(obj) ((obj)->ob_type == &WindowType)
1013071d4279SBram Moolenaar 
1014071d4279SBram Moolenaar /* Buffer type - Implementation functions
1015071d4279SBram Moolenaar  * --------------------------------------
1016071d4279SBram Moolenaar  */
1017071d4279SBram Moolenaar 
1018071d4279SBram Moolenaar #define BufferType_Check(obj) ((obj)->ob_type == &BufferType)
1019071d4279SBram Moolenaar 
10202c45e945SBram Moolenaar static PyInt BufferAssItem(PyObject *, PyInt, PyObject *);
10212c45e945SBram Moolenaar static PyInt BufferAssSlice(PyObject *, PyInt, PyInt, PyObject *);
1022071d4279SBram Moolenaar 
1023071d4279SBram Moolenaar /* Line range type - Implementation functions
1024071d4279SBram Moolenaar  * --------------------------------------
1025071d4279SBram Moolenaar  */
1026071d4279SBram Moolenaar 
1027071d4279SBram Moolenaar #define RangeType_Check(obj) ((obj)->ob_type == &RangeType)
1028071d4279SBram Moolenaar 
10292c45e945SBram Moolenaar static PyInt RangeAssItem(PyObject *, PyInt, PyObject *);
10302c45e945SBram Moolenaar static PyInt RangeAssSlice(PyObject *, PyInt, PyInt, PyObject *);
1031071d4279SBram Moolenaar 
1032071d4279SBram Moolenaar /* Current objects type - Implementation functions
1033071d4279SBram Moolenaar  * -----------------------------------------------
1034071d4279SBram Moolenaar  */
1035071d4279SBram Moolenaar 
1036071d4279SBram Moolenaar static PySequenceMethods BufferAsSeq = {
10372c45e945SBram Moolenaar     (PyInquiry)		BufferLength,	    /* sq_length,    len(x)   */
103877fceb89SBram Moolenaar     (binaryfunc)	0,		    /* BufferConcat, sq_concat, x+y */
103977fceb89SBram Moolenaar     (PyIntArgFunc)	0,		    /* BufferRepeat, sq_repeat, x*n */
10402c45e945SBram Moolenaar     (PyIntArgFunc)	BufferItem,	    /* sq_item,      x[i]     */
10412c45e945SBram Moolenaar     (PyIntIntArgFunc)	BufferSlice,	    /* sq_slice,     x[i:j]   */
10422c45e945SBram Moolenaar     (PyIntObjArgProc)	BufferAssItem,	    /* sq_ass_item,  x[i]=v   */
10432c45e945SBram Moolenaar     (PyIntIntObjArgProc) BufferAssSlice,    /* sq_ass_slice, x[i:j]=v */
10444d1da49cSBram Moolenaar     (objobjproc)	0,
10454d1da49cSBram Moolenaar #if PY_MAJOR_VERSION >= 2
10464d1da49cSBram Moolenaar     (binaryfunc)	0,
1047071d4279SBram Moolenaar     0,
10484d1da49cSBram Moolenaar #endif
1049071d4279SBram Moolenaar };
1050071d4279SBram Moolenaar 
1051071d4279SBram Moolenaar /* Buffer object - Implementation
1052071d4279SBram Moolenaar  */
1053071d4279SBram Moolenaar 
1054071d4279SBram Moolenaar     static PyObject *
1055071d4279SBram Moolenaar BufferGetattr(PyObject *self, char *name)
1056071d4279SBram Moolenaar {
10574d1da49cSBram Moolenaar     PyObject *r;
1058071d4279SBram Moolenaar 
10594d1da49cSBram Moolenaar     if (CheckBuffer((BufferObject *)(self)))
1060071d4279SBram Moolenaar 	return NULL;
1061071d4279SBram Moolenaar 
10624d1da49cSBram Moolenaar     r = BufferAttr((BufferObject *)(self), name);
10634d1da49cSBram Moolenaar     if (r || PyErr_Occurred())
10644d1da49cSBram Moolenaar 	return r;
1065071d4279SBram Moolenaar     else
1066071d4279SBram Moolenaar 	return Py_FindMethod(BufferMethods, self, name);
1067071d4279SBram Moolenaar }
1068071d4279SBram Moolenaar 
1069071d4279SBram Moolenaar /******************/
1070071d4279SBram Moolenaar 
10712c45e945SBram Moolenaar     static PyInt
10722c45e945SBram Moolenaar BufferAssItem(PyObject *self, PyInt n, PyObject *val)
1073071d4279SBram Moolenaar {
10748f1723deSBram Moolenaar     return RBAsItem((BufferObject *)(self), n, val, 1, -1, NULL);
1075071d4279SBram Moolenaar }
1076071d4279SBram Moolenaar 
10772c45e945SBram Moolenaar     static PyInt
10782c45e945SBram Moolenaar BufferAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
1079071d4279SBram Moolenaar {
10808f1723deSBram Moolenaar     return RBAsSlice((BufferObject *)(self), lo, hi, val, 1, -1, NULL);
1081071d4279SBram Moolenaar }
1082071d4279SBram Moolenaar 
1083071d4279SBram Moolenaar static PySequenceMethods RangeAsSeq = {
10842c45e945SBram Moolenaar     (PyInquiry)		RangeLength,	      /* sq_length,    len(x)   */
1085071d4279SBram Moolenaar     (binaryfunc)	0, /* RangeConcat, */ /* sq_concat,    x+y      */
10862c45e945SBram Moolenaar     (PyIntArgFunc)	0, /* RangeRepeat, */ /* sq_repeat,    x*n      */
10872c45e945SBram Moolenaar     (PyIntArgFunc)	RangeItem,	      /* sq_item,      x[i]     */
10882c45e945SBram Moolenaar     (PyIntIntArgFunc)	RangeSlice,	      /* sq_slice,     x[i:j]   */
10892c45e945SBram Moolenaar     (PyIntObjArgProc)	RangeAssItem,	      /* sq_ass_item,  x[i]=v   */
10902c45e945SBram Moolenaar     (PyIntIntObjArgProc) RangeAssSlice,	      /* sq_ass_slice, x[i:j]=v */
10914d1da49cSBram Moolenaar     (objobjproc)	0,
10924d1da49cSBram Moolenaar #if PY_MAJOR_VERSION >= 2
10934d1da49cSBram Moolenaar     (binaryfunc)	0,
10944d1da49cSBram Moolenaar     0,
10954d1da49cSBram Moolenaar #endif
1096071d4279SBram Moolenaar };
1097071d4279SBram Moolenaar 
1098071d4279SBram Moolenaar /* Line range object - Implementation
1099071d4279SBram Moolenaar  */
1100071d4279SBram Moolenaar 
1101071d4279SBram Moolenaar     static PyObject *
1102071d4279SBram Moolenaar RangeGetattr(PyObject *self, char *name)
1103071d4279SBram Moolenaar {
1104071d4279SBram Moolenaar     if (strcmp(name, "start") == 0)
1105e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->start - 1);
1106071d4279SBram Moolenaar     else if (strcmp(name, "end") == 0)
1107e7cb9cf6SBram Moolenaar 	return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->end - 1);
1108071d4279SBram Moolenaar     else
1109071d4279SBram Moolenaar 	return Py_FindMethod(RangeMethods, self, name);
1110071d4279SBram Moolenaar }
1111071d4279SBram Moolenaar 
1112071d4279SBram Moolenaar /****************/
1113071d4279SBram Moolenaar 
11142c45e945SBram Moolenaar     static PyInt
11152c45e945SBram Moolenaar RangeAssItem(PyObject *self, PyInt n, PyObject *val)
1116071d4279SBram Moolenaar {
1117ca8a4dfeSBram Moolenaar     return RBAsItem(((RangeObject *)(self))->buf, n, val,
1118071d4279SBram Moolenaar 		     ((RangeObject *)(self))->start,
1119071d4279SBram Moolenaar 		     ((RangeObject *)(self))->end,
1120071d4279SBram Moolenaar 		     &((RangeObject *)(self))->end);
1121071d4279SBram Moolenaar }
1122071d4279SBram Moolenaar 
11232c45e945SBram Moolenaar     static PyInt
11242c45e945SBram Moolenaar RangeAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
1125071d4279SBram Moolenaar {
112619e60943SBram Moolenaar     return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val,
1127071d4279SBram Moolenaar 		      ((RangeObject *)(self))->start,
1128071d4279SBram Moolenaar 		      ((RangeObject *)(self))->end,
1129071d4279SBram Moolenaar 		      &((RangeObject *)(self))->end);
1130071d4279SBram Moolenaar }
1131071d4279SBram Moolenaar 
11325e538ecdSBram Moolenaar /* TabPage object - Implementation
11335e538ecdSBram Moolenaar  */
11345e538ecdSBram Moolenaar 
11355e538ecdSBram Moolenaar     static PyObject *
11365e538ecdSBram Moolenaar TabPageGetattr(PyObject *self, char *name)
11375e538ecdSBram Moolenaar {
11385e538ecdSBram Moolenaar     PyObject *r;
11395e538ecdSBram Moolenaar 
11405e538ecdSBram Moolenaar     if (CheckTabPage((TabPageObject *)(self)))
11415e538ecdSBram Moolenaar 	return NULL;
11425e538ecdSBram Moolenaar 
11435e538ecdSBram Moolenaar     r = TabPageAttr((TabPageObject *)(self), name);
11445e538ecdSBram Moolenaar     if (r || PyErr_Occurred())
11455e538ecdSBram Moolenaar 	return r;
11465e538ecdSBram Moolenaar     else
11475e538ecdSBram Moolenaar 	return Py_FindMethod(TabPageMethods, self, name);
11485e538ecdSBram Moolenaar }
11495e538ecdSBram Moolenaar 
1150071d4279SBram Moolenaar /* Window object - Implementation
1151071d4279SBram Moolenaar  */
1152071d4279SBram Moolenaar 
1153071d4279SBram Moolenaar     static PyObject *
1154071d4279SBram Moolenaar WindowGetattr(PyObject *self, char *name)
1155071d4279SBram Moolenaar {
11564d1da49cSBram Moolenaar     PyObject *r;
1157071d4279SBram Moolenaar 
11584d1da49cSBram Moolenaar     if (CheckWindow((WindowObject *)(self)))
1159071d4279SBram Moolenaar 	return NULL;
1160071d4279SBram Moolenaar 
11614d1da49cSBram Moolenaar     r = WindowAttr((WindowObject *)(self), name);
11624d1da49cSBram Moolenaar     if (r || PyErr_Occurred())
11634d1da49cSBram Moolenaar 	return r;
1164071d4279SBram Moolenaar     else
1165071d4279SBram Moolenaar 	return Py_FindMethod(WindowMethods, self, name);
1166071d4279SBram Moolenaar }
1167071d4279SBram Moolenaar 
11685e538ecdSBram Moolenaar /* Tab page list object - Definitions
11695e538ecdSBram Moolenaar  */
11705e538ecdSBram Moolenaar 
11715e538ecdSBram Moolenaar static PySequenceMethods TabListAsSeq = {
11725e538ecdSBram Moolenaar     (PyInquiry)		TabListLength,	    /* sq_length,    len(x)   */
11735e538ecdSBram Moolenaar     (binaryfunc)	0,		    /* sq_concat,    x+y      */
11745e538ecdSBram Moolenaar     (PyIntArgFunc)	0,		    /* sq_repeat,    x*n      */
11755e538ecdSBram Moolenaar     (PyIntArgFunc)	TabListItem,	    /* sq_item,      x[i]     */
11765e538ecdSBram Moolenaar     (PyIntIntArgFunc)	0,		    /* sq_slice,     x[i:j]   */
11775e538ecdSBram Moolenaar     (PyIntObjArgProc)	0,		    /* sq_ass_item,  x[i]=v   */
11785e538ecdSBram Moolenaar     (PyIntIntObjArgProc) 0,		    /* sq_ass_slice, x[i:j]=v */
11795e538ecdSBram Moolenaar     (objobjproc)	0,
11805e538ecdSBram Moolenaar #if PY_MAJOR_VERSION >= 2
11815e538ecdSBram Moolenaar     (binaryfunc)	0,
11825e538ecdSBram Moolenaar     0,
11835e538ecdSBram Moolenaar #endif
11845e538ecdSBram Moolenaar };
11855e538ecdSBram Moolenaar 
1186071d4279SBram Moolenaar /* Window list object - Definitions
1187071d4279SBram Moolenaar  */
1188071d4279SBram Moolenaar 
1189071d4279SBram Moolenaar static PySequenceMethods WinListAsSeq = {
11902c45e945SBram Moolenaar     (PyInquiry)		WinListLength,	    /* sq_length,    len(x)   */
1191071d4279SBram Moolenaar     (binaryfunc)	0,		    /* sq_concat,    x+y      */
11922c45e945SBram Moolenaar     (PyIntArgFunc)	0,		    /* sq_repeat,    x*n      */
11932c45e945SBram Moolenaar     (PyIntArgFunc)	WinListItem,	    /* sq_item,      x[i]     */
11942c45e945SBram Moolenaar     (PyIntIntArgFunc)	0,		    /* sq_slice,     x[i:j]   */
11952c45e945SBram Moolenaar     (PyIntObjArgProc)	0,		    /* sq_ass_item,  x[i]=v   */
11962c45e945SBram Moolenaar     (PyIntIntObjArgProc) 0,		    /* sq_ass_slice, x[i:j]=v */
11974d1da49cSBram Moolenaar     (objobjproc)	0,
11984d1da49cSBram Moolenaar #if PY_MAJOR_VERSION >= 2
11994d1da49cSBram Moolenaar     (binaryfunc)	0,
12004d1da49cSBram Moolenaar     0,
12014d1da49cSBram Moolenaar #endif
1202071d4279SBram Moolenaar };
1203071d4279SBram Moolenaar 
1204071d4279SBram Moolenaar /* External interface
1205071d4279SBram Moolenaar  */
1206071d4279SBram Moolenaar 
1207071d4279SBram Moolenaar     void
1208071d4279SBram Moolenaar python_buffer_free(buf_T *buf)
1209071d4279SBram Moolenaar {
1210971db467SBram Moolenaar     if (BUF_PYTHON_REF(buf) != NULL)
1211071d4279SBram Moolenaar     {
1212971db467SBram Moolenaar 	BufferObject *bp = BUF_PYTHON_REF(buf);
1213071d4279SBram Moolenaar 	bp->buf = INVALID_BUFFER_VALUE;
1214971db467SBram Moolenaar 	BUF_PYTHON_REF(buf) = NULL;
1215071d4279SBram Moolenaar     }
1216071d4279SBram Moolenaar }
1217071d4279SBram Moolenaar 
1218071d4279SBram Moolenaar #if defined(FEAT_WINDOWS) || defined(PROTO)
1219071d4279SBram Moolenaar     void
1220071d4279SBram Moolenaar python_window_free(win_T *win)
1221071d4279SBram Moolenaar {
1222971db467SBram Moolenaar     if (WIN_PYTHON_REF(win) != NULL)
1223071d4279SBram Moolenaar     {
1224971db467SBram Moolenaar 	WindowObject *wp = WIN_PYTHON_REF(win);
1225071d4279SBram Moolenaar 	wp->win = INVALID_WINDOW_VALUE;
1226971db467SBram Moolenaar 	WIN_PYTHON_REF(win) = NULL;
1227071d4279SBram Moolenaar     }
1228071d4279SBram Moolenaar }
12295e538ecdSBram Moolenaar 
12305e538ecdSBram Moolenaar     void
12315e538ecdSBram Moolenaar python_tabpage_free(tabpage_T *tab)
12325e538ecdSBram Moolenaar {
12335e538ecdSBram Moolenaar     if (TAB_PYTHON_REF(tab) != NULL)
12345e538ecdSBram Moolenaar     {
12355e538ecdSBram Moolenaar 	TabPageObject *tp = TAB_PYTHON_REF(tab);
12365e538ecdSBram Moolenaar 	tp->tab = INVALID_TABPAGE_VALUE;
12375e538ecdSBram Moolenaar 	TAB_PYTHON_REF(tab) = NULL;
12385e538ecdSBram Moolenaar     }
12395e538ecdSBram Moolenaar }
1240071d4279SBram Moolenaar #endif
1241071d4279SBram Moolenaar 
12421dc28783SBram Moolenaar     static int
12431dc28783SBram Moolenaar add_object(PyObject *dict, const char *name, PyObject *object)
1244071d4279SBram Moolenaar {
12451dc28783SBram Moolenaar     if (PyDict_SetItemString(dict, (char *) name, object))
12461dc28783SBram Moolenaar 	return -1;
12471dc28783SBram Moolenaar     Py_DECREF(object);
12481dc28783SBram Moolenaar     return 0;
12491dc28783SBram Moolenaar }
12505e538ecdSBram Moolenaar 
1251071d4279SBram Moolenaar     static int
1252071d4279SBram Moolenaar PythonMod_Init(void)
1253071d4279SBram Moolenaar {
1254071d4279SBram Moolenaar     PyObject *mod;
1255071d4279SBram Moolenaar     PyObject *dict;
12561dc28783SBram Moolenaar 
12579774ecc8SBram Moolenaar     /* The special value is removed from sys.path in Python_Init(). */
12589774ecc8SBram Moolenaar     static char *(argv[2]) = {"/must>not&exist/foo", NULL};
1259071d4279SBram Moolenaar 
12601dc28783SBram Moolenaar     if (init_types())
12611dc28783SBram Moolenaar 	return -1;
1262071d4279SBram Moolenaar 
1263071d4279SBram Moolenaar     /* Set sys.argv[] to avoid a crash in warn(). */
1264071d4279SBram Moolenaar     PySys_SetArgv(1, argv);
1265071d4279SBram Moolenaar 
1266e7cb9cf6SBram Moolenaar     mod = Py_InitModule4("vim", VimMethods, (char *)NULL, (PyObject *)NULL, PYTHON_API_VERSION);
1267071d4279SBram Moolenaar     dict = PyModule_GetDict(mod);
1268071d4279SBram Moolenaar 
12691dc28783SBram Moolenaar     return populate_module(dict, add_object);
1270071d4279SBram Moolenaar }
1271071d4279SBram Moolenaar 
1272071d4279SBram Moolenaar /*************************************************************************
1273071d4279SBram Moolenaar  * 4. Utility functions for handling the interface between Vim and Python.
1274071d4279SBram Moolenaar  */
1275071d4279SBram Moolenaar 
1276071d4279SBram Moolenaar /* Convert a Vim line into a Python string.
1277071d4279SBram Moolenaar  * All internal newlines are replaced by null characters.
1278071d4279SBram Moolenaar  *
1279071d4279SBram Moolenaar  * On errors, the Python exception data is set, and NULL is returned.
1280071d4279SBram Moolenaar  */
1281071d4279SBram Moolenaar     static PyObject *
1282071d4279SBram Moolenaar LineToString(const char *str)
1283071d4279SBram Moolenaar {
1284071d4279SBram Moolenaar     PyObject *result;
12852c45e945SBram Moolenaar     PyInt len = strlen(str);
1286071d4279SBram Moolenaar     char *p;
1287071d4279SBram Moolenaar 
1288071d4279SBram Moolenaar     /* Allocate an Python string object, with uninitialised contents. We
1289071d4279SBram Moolenaar      * must do it this way, so that we can modify the string in place
1290071d4279SBram Moolenaar      * later. See the Python source, Objects/stringobject.c for details.
1291071d4279SBram Moolenaar      */
1292071d4279SBram Moolenaar     result = PyString_FromStringAndSize(NULL, len);
1293071d4279SBram Moolenaar     if (result == NULL)
1294071d4279SBram Moolenaar 	return NULL;
1295071d4279SBram Moolenaar 
1296071d4279SBram Moolenaar     p = PyString_AsString(result);
1297071d4279SBram Moolenaar 
1298071d4279SBram Moolenaar     while (*str)
1299071d4279SBram Moolenaar     {
1300071d4279SBram Moolenaar 	if (*str == '\n')
1301071d4279SBram Moolenaar 	    *p = '\0';
1302071d4279SBram Moolenaar 	else
1303071d4279SBram Moolenaar 	    *p = *str;
1304071d4279SBram Moolenaar 
1305071d4279SBram Moolenaar 	++p;
1306071d4279SBram Moolenaar 	++str;
1307071d4279SBram Moolenaar     }
1308071d4279SBram Moolenaar 
1309071d4279SBram Moolenaar     return result;
1310071d4279SBram Moolenaar }
1311071d4279SBram Moolenaar 
1312db913953SBram Moolenaar     static PyObject *
1313db913953SBram Moolenaar DictionaryGetattr(PyObject *self, char *name)
1314db913953SBram Moolenaar {
131566b7985eSBram Moolenaar     DictionaryObject	*this = ((DictionaryObject *) (self));
131666b7985eSBram Moolenaar 
131766b7985eSBram Moolenaar     if (strcmp(name, "locked") == 0)
131866b7985eSBram Moolenaar 	return PyInt_FromLong(this->dict->dv_lock);
131966b7985eSBram Moolenaar     else if (strcmp(name, "scope") == 0)
132066b7985eSBram Moolenaar 	return PyInt_FromLong(this->dict->dv_scope);
132166b7985eSBram Moolenaar 
1322db913953SBram Moolenaar     return Py_FindMethod(DictionaryMethods, self, name);
1323db913953SBram Moolenaar }
1324db913953SBram Moolenaar 
1325db913953SBram Moolenaar static PySequenceMethods ListAsSeq = {
1326db913953SBram Moolenaar     (PyInquiry)			ListLength,
1327db913953SBram Moolenaar     (binaryfunc)		0,
1328db913953SBram Moolenaar     (PyIntArgFunc)		0,
1329db913953SBram Moolenaar     (PyIntArgFunc)		ListItem,
1330db913953SBram Moolenaar     (PyIntIntArgFunc)		ListSlice,
1331db913953SBram Moolenaar     (PyIntObjArgProc)		ListAssItem,
1332db913953SBram Moolenaar     (PyIntIntObjArgProc)	ListAssSlice,
1333db913953SBram Moolenaar     (objobjproc)		0,
1334db913953SBram Moolenaar #if PY_MAJOR_VERSION >= 2
1335db913953SBram Moolenaar     (binaryfunc)		ListConcatInPlace,
1336db913953SBram Moolenaar     0,
1337db913953SBram Moolenaar #endif
1338db913953SBram Moolenaar };
1339db913953SBram Moolenaar 
1340db913953SBram Moolenaar     static PyObject *
1341db913953SBram Moolenaar ListGetattr(PyObject *self, char *name)
1342db913953SBram Moolenaar {
134366b7985eSBram Moolenaar     if (strcmp(name, "locked") == 0)
134466b7985eSBram Moolenaar 	return PyInt_FromLong(((ListObject *)(self))->list->lv_lock);
134566b7985eSBram Moolenaar 
1346db913953SBram Moolenaar     return Py_FindMethod(ListMethods, self, name);
1347db913953SBram Moolenaar }
1348db913953SBram Moolenaar 
1349db913953SBram Moolenaar     static PyObject *
1350db913953SBram Moolenaar FunctionGetattr(PyObject *self, char *name)
1351db913953SBram Moolenaar {
1352db913953SBram Moolenaar     FunctionObject	*this = (FunctionObject *)(self);
1353db913953SBram Moolenaar 
1354db913953SBram Moolenaar     if (strcmp(name, "name") == 0)
1355db913953SBram Moolenaar 	return PyString_FromString((char *)(this->name));
1356db913953SBram Moolenaar     else
1357db913953SBram Moolenaar 	return Py_FindMethod(FunctionMethods, self, name);
1358db913953SBram Moolenaar }
1359db913953SBram Moolenaar 
1360db913953SBram Moolenaar     void
1361db913953SBram Moolenaar do_pyeval (char_u *str, typval_T *rettv)
1362db913953SBram Moolenaar {
1363b52f4c02SBram Moolenaar     DoPyCommand((char *) str,
1364b52f4c02SBram Moolenaar 	    (rangeinitializer) init_range_eval,
1365b52f4c02SBram Moolenaar 	    (runner) run_eval,
1366b52f4c02SBram Moolenaar 	    (void *) rettv);
1367db913953SBram Moolenaar     switch(rettv->v_type)
1368db913953SBram Moolenaar     {
1369db913953SBram Moolenaar 	case VAR_DICT: ++rettv->vval.v_dict->dv_refcount; break;
1370db913953SBram Moolenaar 	case VAR_LIST: ++rettv->vval.v_list->lv_refcount; break;
1371db913953SBram Moolenaar 	case VAR_FUNC: func_ref(rettv->vval.v_string);    break;
137277fceb89SBram Moolenaar 	case VAR_UNKNOWN:
137377fceb89SBram Moolenaar 	    rettv->v_type = VAR_NUMBER;
137477fceb89SBram Moolenaar 	    rettv->vval.v_number = 0;
137577fceb89SBram Moolenaar 	    break;
1376db913953SBram Moolenaar     }
1377db913953SBram Moolenaar }
1378071d4279SBram Moolenaar 
1379071d4279SBram Moolenaar /* Don't generate a prototype for the next function, it generates an error on
1380071d4279SBram Moolenaar  * newer Python versions. */
1381071d4279SBram Moolenaar #if PYTHON_API_VERSION < 1007 /* Python 1.4 */ && !defined(PROTO)
1382071d4279SBram Moolenaar 
1383071d4279SBram Moolenaar     char *
1384071d4279SBram Moolenaar Py_GetProgramName(void)
1385071d4279SBram Moolenaar {
1386071d4279SBram Moolenaar     return "vim";
1387071d4279SBram Moolenaar }
1388071d4279SBram Moolenaar #endif /* Python 1.4 */
1389170bf1aeSBram Moolenaar 
1390db913953SBram Moolenaar     void
1391db913953SBram Moolenaar set_ref_in_python (int copyID)
1392db913953SBram Moolenaar {
1393db913953SBram Moolenaar     set_ref_in_py(copyID);
1394db913953SBram Moolenaar }
1395