def test_new_handle(self): ffi = FFI(backend=self.Backend()) o = [2, 3, 4] p = ffi.new_handle(o) assert ffi.typeof(p) == ffi.typeof("void *") assert ffi.from_handle(p) is o assert ffi.from_handle(ffi.cast("char *", p)) is o py.test.raises(RuntimeError, ffi.from_handle, ffi.NULL)
def test_new_handle(self): ffi = FFI(backend=self.Backend()) o = [2, 3, 4] p = ffi.new_handle(o) assert ffi.typeof(p) == ffi.typeof("void *") assert ffi.from_handle(p) is o assert ffi.from_handle(ffi.cast("char *", p)) is o py.test.raises(RuntimeError, ffi.from_handle, ffi.NULL)
class LibGraphqlParser: def __init__(self): self._ffi = FFI() self._ffi.cdef(CDEFS_LIBGRAPHQL) # TODO do a conf of this ? try: self._lib = self._ffi.dlopen("/usr/local/lib/libgraphqlparser.so") except OSError: self._lib = self._ffi.dlopen("/usr/local/lib/libgraphqlparser.dylib") self._lib_callbacks = self._ffi.new( "struct GraphQLAstVisitorCallbacks *" ) self._errors = self._ffi.new("char **") self._callbacks = [] self._interested_by = {} self._default_visitor_cls = Visitor self._creates_callbacks() def _create_visitor_element(self, libgraphql_type, element): try: return _LIBGRAPHQL_TYPE_TO_CLASS[libgraphql_type]( self._lib, self._ffi, element) except KeyError: pass return _VisitorElement(self._lib, self._ffi, libgraphql_type, element) def _callback_enter(self, libgraphql_type, element, udata): context = self._ffi.from_handle(udata) context.update( Visitor.IN, self._create_visitor_element(libgraphql_type, element) ) return context.continue_child def _callback_exit(self, libgraphql_type, element, udata): context = self._ffi.from_handle(udata) if context.continue_child: context.update( Visitor.OUT, self._create_visitor_element(libgraphql_type, element) ) return None def _set_callback(self, proto, func, attr): c_func = self._ffi.callback(proto)(func) # Keep the callbacks alive in this list # to keep the underlying cdata alive. # because we do it with reflexion and # not with decoration self._callbacks.append(c_func) setattr(self._lib_callbacks, attr, c_func) def _set_exit_callback(self, typee): self._set_callback( "void(struct GraphQLAst%s *, void *)" % typee[0], partial(self._callback_exit, typee[0]), "end_visit_%s" % typee[1] ) def _set_enter_callback(self, typee): self._set_callback( "int(struct GraphQLAst%s *, void *)" % typee[0], partial(self._callback_enter, typee[0]), "visit_%s" % typee[1] ) def _creates_callbacks(self): for typee in TYPES_LIBGRAPHQL: if typee[0] not in ["Name"]: self._set_enter_callback(typee) self._set_exit_callback(typee) def _parse(self, query): if isinstance(query, str): # TODO don't replace here. query = query.replace("\n", " ").encode("UTF-8") c_query = self._ffi.new("char[]", query) parsed_data = _ParsedData( self._lib.graphql_parse_string(c_query, self._errors), self._lib.graphql_node_free ) if self._errors[0] != self._ffi.NULL: # TODO specialize Exception here e = Exception( self._ffi.string(self._errors[0]).decode('UTF-8', 'replace') ) self._lib.graphql_error_free(self._errors[0]) raise e return parsed_data def parse_and_visit(self, query, ctx=None): if not ctx: ctx = self._default_visitor_cls() with self._parse(query) as parsed: self._lib.graphql_node_visit( parsed, self._lib_callbacks, self._ffi.new_handle(ctx) ) def parse_and_jsonify(self, query): with self._parse(query) as parsed: return self._ffi.string(self._lib.graphql_ast_to_json(parsed))