1 files changed, 18 insertions, 9 deletions
diff --git a/sources/shiboken6/ApiExtractor/enumtypeentry.h b/sources/shiboken6/ApiExtractor/enumtypeentry.h
index b67e5ad84..3360d7db5 100644
--- a/sources/shiboken6/ApiExtractor/enumtypeentry.h
+++ b/sources/shiboken6/ApiExtractor/enumtypeentry.h
@@ -4,33 +4,42 @@
 #ifndef ENUMTYPEENTRY_H
 #define ENUMTYPEENTRY_H
 
-#include "typesystem.h"
+#include "configurabletypeentry.h"
+#include "typesystem_enums.h"
 
 class EnumTypeEntryPrivate;
-class EnumValueTypeEntry;
-class FlagsTypeEntry;
 
-class EnumTypeEntry : public TypeEntry
+// EnumTypeEntry is configurable for global enums only
+class EnumTypeEntry : public ConfigurableTypeEntry
 {
 public:
     explicit EnumTypeEntry(const QString &entryName,
                            const QVersionNumber &vr,
-                           const TypeEntry *parent);
+                           const TypeEntryCPtr &parent);
+
+    TypeSystem::PythonEnumType pythonEnumType() const;
+    void setPythonEnumType(TypeSystem::PythonEnumType t);
 
     QString targetLangQualifier() const;
 
     QString qualifier() const;
 
-    const EnumValueTypeEntry *nullValue() const;
-    void setNullValue(const EnumValueTypeEntry *n);
+    EnumValueTypeEntryCPtr nullValue() const;
+    void setNullValue(const EnumValueTypeEntryCPtr &n);
+
+    void setFlags(const FlagsTypeEntryPtr &flags);
+    FlagsTypeEntryPtr flags() const;
 
-    void setFlags(FlagsTypeEntry *flags);
-    FlagsTypeEntry *flags() const;
+    QString cppType() const;
+    void setCppType(const QString &t);
 
     bool isEnumValueRejected(const QString &name) const;
     void addEnumValueRejection(const QString &name);
     QStringList enumValueRejections() const;
 
+    QString docFile() const;
+    void setDocFile(const QString &df);
+
     TypeEntry *clone() const override;
 #ifndef QT_NO_DEBUG_STREAM
     void formatDebug(QDebug &d) const override;