diff options
| -rw-r--r-- | bindings/python/clang/cindex.py | 18 | ||||
| -rw-r--r-- | bindings/python/tests/cindex/test_cursor.py | 15 |
2 files changed, 33 insertions, 0 deletions
diff --git a/bindings/python/clang/cindex.py b/bindings/python/clang/cindex.py index 4c819c26fe..8629213207 100644 --- a/bindings/python/clang/cindex.py +++ b/bindings/python/clang/cindex.py @@ -995,6 +995,19 @@ class Cursor(Structure): return self._underlying_type @property + def enum_type(self): + """Return the integer type of an enum declaration. + + Returns a type corresponding to an integer. If the cursor is not for an + enum, this raises. + """ + if not hasattr(self, '_enum_type'): + assert self.kind == CursorKind.ENUM_DECL + self._enum_type = Cursor_enum_type(self) + + return self._enum_type + + @property def hash(self): """Returns a hash of the cursor as an int.""" if not hasattr(self, '_hash'): @@ -1818,6 +1831,11 @@ Cursor_underlying_type.argtypes = [Cursor] Cursor_underlying_type.restype = Type Cursor_underlying_type.errcheck = Type.from_result +Cursor_enum_type = lib.clang_getEnumDeclIntegerType +Cursor_enum_type.argtypes = [Cursor] +Cursor_enum_type.restype = Type +Cursor_enum_type.errcheck = Type.from_result + Cursor_visit_callback = CFUNCTYPE(c_int, Cursor, Cursor, py_object) Cursor_visit = lib.clang_visitChildren Cursor_visit.argtypes = [Cursor, Cursor_visit_callback, py_object] diff --git a/bindings/python/tests/cindex/test_cursor.py b/bindings/python/tests/cindex/test_cursor.py index d86830334d..71ee0c5763 100644 --- a/bindings/python/tests/cindex/test_cursor.py +++ b/bindings/python/tests/cindex/test_cursor.py @@ -77,3 +77,18 @@ def test_underlying_type(): assert typedef.kind.is_declaration() underlying = typedef.underlying_typedef_type assert underlying.kind == TypeKind.INT + +def test_enum_type(): + source = 'enum TEST { FOO=1, BAR=2 };' + index = Index.create() + tu = index.parse('test.c', unsaved_files=[('test.c', source)]) + assert tu is not None + + for cursor in tu.cursor.get_children(): + if cursor.spelling == 'TEST': + enum = cursor + break + + assert enum.kind == CursorKind.ENUM_DECL + enum_type = enum.enum_type + assert enum_type.kind == TypeKind.UINT |