python: Expose version of the module

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/python/location-query.in b/src/python/location-query.in
index 260f4bd88abe0102629d98006c0f338377551c7e..e750ea4bb15cacfe36b3aa9498c6ea3825aefcbe 100644 (file)
--- a/src/python/location-query.in
+++ b/src/python/location-query.in
@@ -42,6 +42,10 @@ class CLI(object):
                parser.add_argument("--debug", action="store_true",
                        help=_("Enable debug output"))
 
+               # version
+               parser.add_argument("--version", action="version",
+                       version="%%(prog)s %s" % location.__version__)
+
                # lookup an IP address
                lookup = subparsers.add_parser("lookup",
                        help=_("Lookup one or multiple IP addresses"),

diff --git a/src/python/locationmodule.c b/src/python/locationmodule.c
index 40b0c35ffb032d4e2cf1a917437580bdec17659e..dcb5a47560d174565c1e01c609d395087af23b69 100644 (file)
--- a/src/python/locationmodule.c
+++ b/src/python/locationmodule.c
@@ -106,5 +106,11 @@ PyMODINIT_FUNC PyInit_location(void) {
        Py_INCREF(&WriterType);
        PyModule_AddObject(m, "Writer", (PyObject *)&WriterType);
 
+       // Add constants
+       PyObject* d = PyModule_GetDict(m);
+
+       // Version
+       PyDict_SetItemString(d, "__version__", PyUnicode_FromString(VERSION));
+
        return m;
 }