Age         Owner                    Branch data    TLA  Line data    Source code

                                  1                 :                : #include "postgres.h"
                                  2                 :                : 
                                  3                 :                : #include "fmgr.h"
                                  4                 :                : #include "ltree/ltree.h"
                                  5                 :                : #include "plpython.h"
                                  6                 :                : 
 3276 peter_e@gmx.net             7                 :CBC           1 : PG_MODULE_MAGIC;
                                  8                 :                : 
                                  9                 :                : /* Linkage to functions in plpython module */
                                 10                 :                : typedef PyObject *(*PLyUnicode_FromStringAndSize_t) (const char *s, Py_ssize_t size);
                                 11                 :                : static PLyUnicode_FromStringAndSize_t PLyUnicode_FromStringAndSize_p;
                                 12                 :                : 
                                 13                 :                : 
                                 14                 :                : /*
                                 15                 :                :  * Module initialize function: fetch function pointers for cross-module calls.
                                 16                 :                :  */
                                 17                 :                : void
 2749 tgl@sss.pgh.pa.us          18                 :              1 : _PG_init(void)
                                 19                 :                : {
                                 20                 :                :     /* Asserts verify that typedefs above match original declarations */
                                 21                 :                :     AssertVariableIsOfType(&PLyUnicode_FromStringAndSize, PLyUnicode_FromStringAndSize_t);
                                 22                 :              1 :     PLyUnicode_FromStringAndSize_p = (PLyUnicode_FromStringAndSize_t)
                                 23                 :              1 :         load_external_function("$libdir/" PLPYTHON_LIBNAME, "PLyUnicode_FromStringAndSize",
                                 24                 :                :                                true, NULL);
                                 25                 :              1 : }
                                 26                 :                : 
                                 27                 :                : 
                                 28                 :                : /* These defines must be after the module init function */
                                 29                 :                : #define PLyUnicode_FromStringAndSize PLyUnicode_FromStringAndSize_p
                                 30                 :                : 
                                 31                 :                : 
 3276 peter_e@gmx.net            32                 :              2 : PG_FUNCTION_INFO_V1(ltree_to_plpython);
                                 33                 :                : 
                                 34                 :                : Datum
                                 35                 :              2 : ltree_to_plpython(PG_FUNCTION_ARGS)
                                 36                 :                : {
 2400 tgl@sss.pgh.pa.us          37                 :              2 :     ltree      *in = PG_GETARG_LTREE_P(0);
                                 38                 :                :     int         i;
                                 39                 :                :     PyObject   *list;
                                 40                 :                :     ltree_level *curlevel;
                                 41                 :                : 
 3276 peter_e@gmx.net            42                 :              2 :     list = PyList_New(in->numlevel);
 2357                            43         [ -  + ]:              2 :     if (!list)
 2357 peter_e@gmx.net            44         [ #  # ]:UBC           0 :         ereport(ERROR,
                                 45                 :                :                 (errcode(ERRCODE_OUT_OF_MEMORY),
                                 46                 :                :                  errmsg("out of memory")));
                                 47                 :                : 
 3276 peter_e@gmx.net            48                 :CBC           2 :     curlevel = LTREE_FIRST(in);
                                 49         [ +  + ]:              8 :     for (i = 0; i < in->numlevel; i++)
                                 50                 :                :     {
  769 andres@anarazel.de         51                 :              6 :         PyList_SetItem(list, i, PLyUnicode_FromStringAndSize(curlevel->name, curlevel->len));
 3276 peter_e@gmx.net            52                 :              6 :         curlevel = LEVEL_NEXT(curlevel);
                                 53                 :                :     }
                                 54                 :                : 
                                 55         [ -  + ]:              2 :     PG_FREE_IF_COPY(in, 0);
                                 56                 :                : 
                                 57                 :              2 :     return PointerGetDatum(list);
                                 58                 :                : }