python: Implement comparison of ASes
[people/ms/libloc.git] / 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, NULL, &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 PyObject* AS_richcompare(ASObject* self, ASObject* other, int op) {
84         int r = loc_as_cmp(self->as, other->as);
85
86         switch (op) {
87                 case Py_EQ:
88                         if (r == 0)
89                                 Py_RETURN_TRUE;
90
91                         Py_RETURN_FALSE;
92
93                 case Py_LT:
94                         if (r < 0)
95                                 Py_RETURN_TRUE;
96
97                         Py_RETURN_FALSE;
98
99                 default:
100                         break;
101         }
102
103         Py_RETURN_NOTIMPLEMENTED;
104 }
105
106 static struct PyGetSetDef AS_getsetters[] = {
107         {
108                 "name",
109                 (getter)AS_get_name,
110                 NULL,
111                 NULL,
112                 NULL,
113         },
114         {
115                 "number",
116                 (getter)AS_get_number,
117                 NULL,
118                 NULL,
119                 NULL,
120         },
121         { NULL },
122 };
123
124 PyTypeObject ASType = {
125         PyVarObject_HEAD_INIT(NULL, 0)
126         tp_name:                "location.AS",
127         tp_basicsize:           sizeof(ASObject),
128         tp_flags:               Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
129         tp_new:                 AS_new,
130         tp_dealloc:             (destructor)AS_dealloc,
131         tp_init:                (initproc)AS_init,
132         tp_doc:                 "AS object",
133         tp_getset:              AS_getsetters,
134         tp_repr:                (reprfunc)AS_repr,
135         tp_richcompare:         (richcmpfunc)AS_richcompare,
136 };