From 608c84e0b5c2140be099e75547b894a81ada6fa0 Mon Sep 17 00:00:00 2001 From: Josh Ferrell Date: Tue, 10 Jan 2023 17:26:05 -0500 Subject: Fix generation of incorrect python type hints for enums --- python/generator.cpp | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'python/generator.cpp') diff --git a/python/generator.cpp b/python/generator.cpp index 0db85535..a26334e5 100644 --- a/python/generator.cpp +++ b/python/generator.cpp @@ -181,7 +181,7 @@ void OutputSwizzledType(FILE* out, Type* type) string name = type->GetNamedTypeReference()->GetName().GetString(); if (name.size() > 2 && name.substr(0, 2) == "BN") name = name.substr(2); - fprintf(out, "%sEnum", name.c_str()); + fprintf(out, "%s", name.c_str()); } else { @@ -253,8 +253,10 @@ int main(int argc, char* argv[]) FILE* enums = fopen(argv[3], "w"); fprintf(out, "import ctypes, os\n\n"); - fprintf(out, "from typing import Optional, AnyStr"); - fprintf(enums, "import enum"); + fprintf(out, "from typing import Optional, AnyStr\n"); + fprintf(out, "from .enums import *"); + + fprintf(enums, "import enum\n"); fprintf(out, "# Load core module\n"); fprintf(out, "import platform\n"); -- cgit v1.3.1