Adds way of loading context back to python

Signed-off-by: ncordon <nacho.cordon.castillo@gmail.com>

Author: ncordon

bblfsh/__init__.py

@@ -3,3 +3,4 @@
 from bblfsh.tree_order import TreeOrder
 from bblfsh.aliases import *
 from bblfsh.roles import role_id, role_name
+from bblfsh.context import context

bblfsh/context.py

@@ -0,0 +1,19 @@
+from bblfsh.aliases import ParseResponse
+from bblfsh.pyuast import decode, iterator, uast
+from bblfsh.tree_order import TreeOrder
+
+# Python context
+class Context:
+    def __init__(self, root: dict) -> None:
+        self.ctx = uast()
+        self.root = root
+
+    def filter(self, query: str) -> dict:
+        return self.ctx.filter(query, self.root)
+
+    def iterate(self, order: int) -> iterator:
+        TreeOrder.check_order(order)
+        return iterator(self.root, order)
+
+def context(root: dict) -> Context:
+    return Context(root)

bblfsh/pyuast.cc

@@ -1066,7 +1066,6 @@ static PyObject *PythonContext_new(PyObject *self, PyObject *args) {
     // TODO: optionally accept root object
     if (!PyArg_ParseTuple(args, "")) return nullptr;
 
-
     PythonContext *pyU = PyObject_New(PythonContext, &PythonContextType);
     if (!pyU) return nullptr;