Added nodetrie submodule. Updated setup.py for nodetrie C library ext…

…ension building. Updated cython templates test
InfluxGraph · Mar 16, 2017 · 9900035 · 9900035
1 parent b20c336
commit 9900035
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 7 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -0,0 +1,3 @@
+[submodule "nodetrie"]
+	path = nodetrie
+	url = https://github.com/NodeTrie/NodeTrie_Py.git
diff --git a/nodetrie b/nodetrie
diff --git a/setup.py b/setup.py
@@ -20,14 +20,28 @@
 
 ext = 'pyx' if USING_CYTHON else 'c'
 
-extensions = [Extension("influxgraph.ext.classes.tree",
-                        ["influxgraph/ext/classes/tree.%s" % (ext,)]),
-              Extension("influxgraph.ext.templates",
-                        ["influxgraph/ext/templates.%s" % (ext,)]),]
+extensions = [Extension("influxgraph.ext.templates",
+                        ["influxgraph/ext/templates.%s" % (ext,)],
+                        extra_compile_args=["-O3"],
+                        ),
+              Extension("influxgraph.ext.nodetrie",
+                        ["nodetrie/nodetrie/nodetrie.c",
+                         "nodetrie/nodetrie_c/src/node.c",],
+                         depends=["nodetrie/nodetrie_c/src/node.h"],
+                         include_dirs=["nodetrie/nodetrie_c/src"],
+                         extra_compile_args=["-std=c99", "-O3"],
+                         ),
+             ]
 
 if USING_CYTHON:
-    extensions = cythonize(extensions)
-
+    extensions = cythonize(
+        extensions,
+        compiler_directives={'embedsignature': True,
+                             'optimize.use_switch': True,
+                             'boundscheck': False,
+                             'wraparound': False,
+                             })
+
 ext_errors = (CCompilerError, DistutilsExecError, DistutilsPlatformError)
 
 class BuildFailed(Exception):

diff --git a/tests/test_c_templates.py b/tests/test_c_templates.py
@@ -34,7 +34,7 @@ def setUp(self):
     def test_parse_series_missing_fields(self):
         templates = parse_influxdb_graphite_templates(
             ['measurement.field*'])
-        paths = ['series1,id=1', u'series2,id=1']
+        paths = [u'series1,id=1', u'series2,id=1']
         fields = {}
         series = parse_series(paths, fields, templates)
         self.assertTrue(len(series.children) == 0)