src/python/as.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
  19 #include <loc/libloc.h>
  20 #include <loc/as.h>
  21
  22 #include "locationmodule.h"
  23 #include "as.h"
  24
  25 PyObject* new_as(PyTypeObject* type, struct loc_as* as) {
  26         ASObject* self = (ASObject*)type->tp_alloc(type, 0);
  27         if (self) {
  28                 self->as = loc_as_ref(as);
  29         }
  30
  31         return (PyObject*)self;
  32 }
  33
  34 static PyObject* AS_new(PyTypeObject* type, PyObject* args, PyObject* kwds) {
  35         ASObject* self = (ASObject*)type->tp_alloc(type, 0);
  36
  37         return (PyObject*)self;
  38 }
  39
  40 static void AS_dealloc(ASObject* self) {
  41         if (self->as)
  42                 loc_as_unref(self->as);
  43
  44         Py_TYPE(self)->tp_free((PyObject* )self);
  45 }
  46
  47 static int AS_init(ASObject* self, PyObject* args, PyObject* kwargs) {
  48         uint32_t number = 0;
  49
  50         if (!PyArg_ParseTuple(args, "i", &number))
  51                 return -1;
  52
  53         // Create the AS object
  54         int r = loc_as_new(loc_ctx, &self->as, number);
  55         if (r)
  56                 return -1;
  57
  58         return 0;
  59 }
  60
  61 static PyObject* AS_repr(ASObject* self) {
  62         uint32_t number = loc_as_get_number(self->as);
  63         const char* name = loc_as_get_name(self->as);
  64
  65         if (name)
  66                 return PyUnicode_FromFormat("<AS %d (%s)>", number, name);
  67
  68         return PyUnicode_FromFormat("<AS %d>", number);
  69 }
  70
  71 static PyObject* AS_get_number(ASObject* self) {
  72         uint32_t number = loc_as_get_number(self->as);
  73
  74         return PyLong_FromLong(number);
  75 }
  76
  77 static PyObject* AS_get_name(ASObject* self) {
  78         const char* name = loc_as_get_name(self->as);
  79
  80         return PyUnicode_FromString(name);
  81 }
  82
  83 static int AS_set_name(ASObject* self, PyObject* value) {
  84         const char* name = PyUnicode_AsUTF8(value);
  85
  86         int r = loc_as_set_name(self->as, name);
  87         if (r) {
  88                 PyErr_Format(PyExc_ValueError, "Could not set name: %s", name);
  89                 return r;
  90         }
  91
  92         return 0;
  93 }
  94
  95 static PyObject* AS_richcompare(ASObject* self, ASObject* other, int op) {
  96         int r = loc_as_cmp(self->as, other->as);
  97
  98         switch (op) {
  99                 case Py_EQ:
 100                         if (r == 0)
 101                                 Py_RETURN_TRUE;
 102
 103                         Py_RETURN_FALSE;
 104
 105                 case Py_LT:
 106                         if (r < 0)
 107                                 Py_RETURN_TRUE;
 108
 109                         Py_RETURN_FALSE;
 110
 111                 default:
 112                         break;
 113         }
 114
 115         Py_RETURN_NOTIMPLEMENTED;
 116 }
 117
 118 static struct PyGetSetDef AS_getsetters[] = {
 119         {
 120                 "name",
 121                 (getter)AS_get_name,
 122                 (setter)AS_set_name,
 123                 NULL,
 124                 NULL,
 125         },
 126         {
 127                 "number",
 128                 (getter)AS_get_number,
 129                 NULL,
 130                 NULL,
 131                 NULL,
 132         },
 133         { NULL },
 134 };
 135
 136 PyTypeObject ASType = {
 137         PyVarObject_HEAD_INIT(NULL, 0)
 138         tp_name:                "location.AS",
 139         tp_basicsize:           sizeof(ASObject),
 140         tp_flags:               Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
 141         tp_new:                 AS_new,
 142         tp_dealloc:             (destructor)AS_dealloc,
 143         tp_init:                (initproc)AS_init,
 144         tp_doc:                 "AS object",
 145         tp_getset:              AS_getsetters,
 146         tp_repr:                (reprfunc)AS_repr,
 147         tp_richcompare:         (richcmpfunc)AS_richcompare,
 148 };