src/python/locationmodule.c

   1 /*
   2         libloc - A library to determine the location of someone on the Internet
   3
   4         Copyright (C) 2017 IPFire Development Team <info@ipfire.org>
   5
   6         This library is free software; you can redistribute it and/or
   7         modify it under the terms of the GNU Lesser General Public
   8         License as published by the Free Software Foundation; either
   9         version 2.1 of the License, or (at your option) any later version.
  10
  11         This library is distributed in the hope that it will be useful,
  12         but WITHOUT ANY WARRANTY; without even the implied warranty of
  13         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14         Lesser General Public License for more details.
  15 */
  16
  17 #include <Python.h>
  18 #include <syslog.h>
  19
  20 #include "locationmodule.h"
  21 #include "as.h"
  22 #include "database.h"
  23 #include "network.h"
  24 #include "writer.h"
  25
  26 PyMODINIT_FUNC PyInit_location(void);
  27
  28 static void location_free(void) {
  29         // Release context
  30         if (loc_ctx)
  31                 loc_unref(loc_ctx);
  32 }
  33
  34 static PyObject* set_log_level(PyObject* m, PyObject* args) {
  35         int priority = LOG_INFO;
  36
  37         if (!PyArg_ParseTuple(args, "i", &priority))
  38                 return NULL;
  39
  40         loc_set_log_priority(loc_ctx, priority);
  41
  42         Py_RETURN_NONE;
  43 }
  44
  45 static PyMethodDef location_module_methods[] = {
  46         {
  47                 "set_log_level",
  48                 (PyCFunction)set_log_level,
  49                 METH_VARARGS,
  50                 NULL,
  51         },
  52         { NULL },
  53 };
  54
  55 static struct PyModuleDef location_module = {
  56         .m_base = PyModuleDef_HEAD_INIT,
  57         .m_name = "location",
  58         .m_size = -1,
  59         .m_doc = "Python module for libloc",
  60         .m_methods = location_module_methods,
  61         .m_free = (freefunc)location_free,
  62 };
  63
  64 PyMODINIT_FUNC PyInit_location(void) {
  65         // Initialise loc context
  66         int r = loc_new(&loc_ctx);
  67         if (r)
  68                 return NULL;
  69
  70         PyObject* m = PyModule_Create(&location_module);
  71         if (!m)
  72                 return NULL;
  73
  74         // AS
  75         if (PyType_Ready(&ASType) < 0)
  76                 return NULL;
  77
  78         Py_INCREF(&ASType);
  79         PyModule_AddObject(m, "AS", (PyObject *)&ASType);
  80
  81         // Database
  82         if (PyType_Ready(&DatabaseType) < 0)
  83                 return NULL;
  84
  85         Py_INCREF(&DatabaseType);
  86         PyModule_AddObject(m, "Database", (PyObject *)&DatabaseType);
  87
  88         // Database Enumerator
  89         if (PyType_Ready(&DatabaseEnumeratorType) < 0)
  90                 return NULL;
  91
  92         Py_INCREF(&DatabaseEnumeratorType);
  93         //PyModule_AddObject(m, "DatabaseEnumerator", (PyObject *)&DatabaseEnumeratorType);
  94
  95         // Network
  96         if (PyType_Ready(&NetworkType) < 0)
  97                 return NULL;
  98
  99         Py_INCREF(&NetworkType);
 100         PyModule_AddObject(m, "Network", (PyObject *)&NetworkType);
 101
 102         // Writer
 103         if (PyType_Ready(&WriterType) < 0)
 104                 return NULL;
 105
 106         Py_INCREF(&WriterType);
 107         PyModule_AddObject(m, "Writer", (PyObject *)&WriterType);
 108
 109         // Add constants
 110         PyObject* d = PyModule_GetDict(m);
 111
 112         // Version
 113         PyDict_SetItemString(d, "__version__", PyUnicode_FromString(VERSION));
 114
 115         return m;
 116 }