python: Show better repr for AS classes
[people/ms/libloc.git] / src / python / as.c
CommitLineData
fadebc89
MT
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>
fadebc89 21
1da9cd39 22#include "locationmodule.h"
fadebc89
MT
23#include "as.h"
24
86ca7ef7
MT
25PyObject* 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
fadebc89 34static PyObject* AS_new(PyTypeObject* type, PyObject* args, PyObject* kwds) {
fadebc89 35 ASObject* self = (ASObject*)type->tp_alloc(type, 0);
fadebc89
MT
36
37 return (PyObject*)self;
38}
39
40static void AS_dealloc(ASObject* self) {
41 if (self->as)
42 loc_as_unref(self->as);
43
fadebc89
MT
44 Py_TYPE(self)->tp_free((PyObject* )self);
45}
46
47static 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
5e6100a3 54 int r = loc_as_new(loc_ctx, NULL, &self->as, number);
fadebc89
MT
55 if (r)
56 return -1;
57
58 return 0;
59}
60
2e07b5a7
MT
61static 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
fadebc89
MT
71static PyObject* AS_get_number(ASObject* self) {
72 uint32_t number = loc_as_get_number(self->as);
73
74 return PyLong_FromLong(number);
75}
76
77static PyObject* AS_get_name(ASObject* self) {
78 const char* name = loc_as_get_name(self->as);
79
80 return PyUnicode_FromString(name);
81}
82
83static struct PyGetSetDef AS_getsetters[] = {
84 {
85 "name",
86 (getter)AS_get_name,
87 NULL,
88 NULL,
89 NULL,
90 },
91 {
92 "number",
93 (getter)AS_get_number,
94 NULL,
95 NULL,
96 NULL,
97 },
98 { NULL },
99};
100
101PyTypeObject ASType = {
102 PyVarObject_HEAD_INIT(NULL, 0)
103 tp_name: "location.AS",
104 tp_basicsize: sizeof(ASObject),
105 tp_flags: Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
106 tp_new: AS_new,
107 tp_dealloc: (destructor)AS_dealloc,
108 tp_init: (initproc)AS_init,
109 tp_doc: "AS object",
110 tp_getset: AS_getsetters,
2e07b5a7 111 tp_repr: (reprfunc)AS_repr,
fadebc89 112};