[people/ms/libloc.git] / src / python / as.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 <loc/libloc.h>
#include <loc/as.h>

#include "locationmodule.h"
#include "as.h"

PyObject* new_as(PyTypeObject* type, struct loc_as* as) {
	ASObject* self = (ASObject*)type->tp_alloc(type, 0);
	if (self) {
		self->as = loc_as_ref(as);
	}

	return (PyObject*)self;
}

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

	return (PyObject*)self;
}

static void AS_dealloc(ASObject* self) {
	if (self->as)
		loc_as_unref(self->as);

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

static int AS_init(ASObject* self, PyObject* args, PyObject* kwargs) {
	uint32_t number = 0;

	if (!PyArg_ParseTuple(args, "i", &number))
		return -1;

	// Create the AS object
	int r = loc_as_new(loc_ctx, NULL, &self->as, number);
	if (r)
		return -1;

	return 0;
}

static PyObject* AS_repr(ASObject* self) {
	uint32_t number = loc_as_get_number(self->as);
	const char* name = loc_as_get_name(self->as);

	if (name)
		return PyUnicode_FromFormat("<AS %d (%s)>", number, name);

	return PyUnicode_FromFormat("<AS %d>", number);
}

static PyObject* AS_get_number(ASObject* self) {
	uint32_t number = loc_as_get_number(self->as);

	return PyLong_FromLong(number);
}

static PyObject* AS_get_name(ASObject* self) {
	const char* name = loc_as_get_name(self->as);

	return PyUnicode_FromString(name);
}

static struct PyGetSetDef AS_getsetters[] = {
	{
		"name",
		(getter)AS_get_name,
		NULL,
		NULL,
		NULL,
	},
	{
		"number",
		(getter)AS_get_number,
		NULL,
		NULL,
		NULL,
	},
	{ NULL },
};

PyTypeObject ASType = {
	PyVarObject_HEAD_INIT(NULL, 0)
	tp_name:                "location.AS",
	tp_basicsize:           sizeof(ASObject),
	tp_flags:               Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
	tp_new:                 AS_new,
	tp_dealloc:             (destructor)AS_dealloc,
	tp_init:                (initproc)AS_init,
	tp_doc:                 "AS object",
	tp_getset:              AS_getsetters,
	tp_repr:                (reprfunc)AS_repr,
};
Commit	Line	Data
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"
fadebc89 MT	24
86ca7ef7 MT	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
fadebc89	34	static 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
	40	static 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
	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
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	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
fadebc89 MT	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 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
	101	PyTypeObject 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	};