python: Implement Network class
[people/ms/libloc.git] / src / python / network.c
CommitLineData
39967361
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 <errno.h>
20
21#include <loc/libloc.h>
22#include <loc/network.h>
23
24#include "locationmodule.h"
25#include "network.h"
26
27static PyObject* Network_new(PyTypeObject* type, PyObject* args, PyObject* kwds) {
28 NetworkObject* self = (NetworkObject*)type->tp_alloc(type, 0);
29
30 return (PyObject*)self;
31}
32
33static void Network_dealloc(NetworkObject* self) {
34 if (self->network)
35 loc_network_unref(self->network);
36
37 Py_TYPE(self)->tp_free((PyObject* )self);
38}
39
40static int Network_init(NetworkObject* self, PyObject* args, PyObject* kwargs) {
41 const char* network = NULL;
42
43 if (!PyArg_ParseTuple(args, "s", &network))
44 return -1;
45
46 // Load the Network
47 int r = loc_network_new_from_string(loc_ctx, &self->network, network);
48 if (r) {
49 PyErr_Format(PyExc_ValueError, "Invalid network: %s", network);
50 return -1;
51 }
52
53 return 0;
54}
55
56static PyObject* Network_repr(NetworkObject* self) {
57 char* network = loc_network_str(self->network);
58
59 PyObject* obj = PyUnicode_FromFormat("<location.Network %s>", network);
60 free(network);
61
62 return obj;
63}
64
65static PyObject* Network_get_country_code(NetworkObject* self) {
66 const char* country_code = loc_network_get_country_code(self->network);
67
68 return PyUnicode_FromString(country_code);
69}
70
71static int Network_set_country_code(NetworkObject* self, PyObject* value) {
72 const char* country_code = PyUnicode_AsUTF8(value);
73
74 int r = loc_network_set_country_code(self->network, country_code);
75 if (r) {
76 if (r == -EINVAL)
77 PyErr_SetString(PyExc_ValueError, "Invalid country code");
78
79 return -1;
80 }
81
82 return 0;
83}
84
85static struct PyGetSetDef Network_getsetters[] = {
86 {
87 "country_code",
88 (getter)Network_get_country_code,
89 (setter)Network_set_country_code,
90 NULL,
91 NULL,
92 },
93 { NULL },
94};
95
96PyTypeObject NetworkType = {
97 PyVarObject_HEAD_INIT(NULL, 0)
98 tp_name: "location.Network",
99 tp_basicsize: sizeof(NetworkObject),
100 tp_flags: Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
101 tp_new: Network_new,
102 tp_dealloc: (destructor)Network_dealloc,
103 tp_init: (initproc)Network_init,
104 tp_doc: "Network object",
105 tp_getset: Network_getsetters,
106 tp_repr: (reprfunc)Network_repr,
107};