[people/ms/libloc.git] / src / python / network.c

/*
	libloc - A library to determine the location of someone on the Internet

	Copyright (C) 2017 IPFire Development Team <info@ipfire.org>

	This library is free software; you can redistribute it and/or
	modify it under the terms of the GNU Lesser General Public
	License as published by the Free Software Foundation; either
	version 2.1 of the License, or (at your option) any later version.

	This library is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
	Lesser General Public License for more details.
*/

#include <Python.h>

#include <errno.h>

#include <loc/libloc.h>
#include <loc/network.h>

#include "locationmodule.h"
#include "network.h"

static PyObject* Network_new(PyTypeObject* type, PyObject* args, PyObject* kwds) {
	NetworkObject* self = (NetworkObject*)type->tp_alloc(type, 0);

	return (PyObject*)self;
}

static void Network_dealloc(NetworkObject* self) {
	if (self->network)
		loc_network_unref(self->network);

	Py_TYPE(self)->tp_free((PyObject* )self);
}

static int Network_init(NetworkObject* self, PyObject* args, PyObject* kwargs) {
	const char* network = NULL;

	if (!PyArg_ParseTuple(args, "s", &network))
		return -1;

	// Load the Network
	int r = loc_network_new_from_string(loc_ctx, &self->network, network);
	if (r) {
		PyErr_Format(PyExc_ValueError, "Invalid network: %s", network);
		return -1;
	}

	return 0;
}

static PyObject* Network_repr(NetworkObject* self) {
	char* network = loc_network_str(self->network);

	PyObject* obj = PyUnicode_FromFormat("<location.Network %s>", network);
	free(network);

	return obj;
}

static PyObject* Network_get_country_code(NetworkObject* self) {
	const char* country_code = loc_network_get_country_code(self->network);

	return PyUnicode_FromString(country_code);
}

static int Network_set_country_code(NetworkObject* self, PyObject* value) {
	const char* country_code = PyUnicode_AsUTF8(value);

	int r = loc_network_set_country_code(self->network, country_code);
	if (r) {
		if (r == -EINVAL)
			PyErr_SetString(PyExc_ValueError, "Invalid country code");

		return -1;
	}

	return 0;
}

static struct PyGetSetDef Network_getsetters[] = {
	{
		"country_code",
		(getter)Network_get_country_code,
		(setter)Network_set_country_code,
		NULL,
		NULL,
	},
	{ NULL },
};

PyTypeObject NetworkType = {
	PyVarObject_HEAD_INIT(NULL, 0)
	tp_name:                "location.Network",
	tp_basicsize:           sizeof(NetworkObject),
	tp_flags:               Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
	tp_new:                 Network_new,
	tp_dealloc:             (destructor)Network_dealloc,
	tp_init:                (initproc)Network_init,
	tp_doc:                 "Network object",
	tp_getset:              Network_getsetters,
	tp_repr:                (reprfunc)Network_repr,
};
Commit	Line	Data
	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
	27	static 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
	33	static 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
	40	static 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
	56	static 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
	65	static 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
	71	static 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
	85	static 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
	96	PyTypeObject 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	};