diff --git a/comtypes/tools/codegenerator/codegenerator.py b/comtypes/tools/codegenerator/codegenerator.py index 37ef4691..5d207ea0 100644 --- a/comtypes/tools/codegenerator/codegenerator.py +++ b/comtypes/tools/codegenerator/codegenerator.py @@ -196,7 +196,6 @@ def generate_wrapper_code( print("from typing import TYPE_CHECKING", file=output) print(file=output) print("if TYPE_CHECKING:", file=output) - print(" from typing import Any, Tuple", file=output) print(" from comtypes import hints", file=output) print(file=output) print(file=output) diff --git a/comtypes/tools/codegenerator/typeannotator.py b/comtypes/tools/codegenerator/typeannotator.py index 7893e764..d788601d 100644 --- a/comtypes/tools/codegenerator/typeannotator.py +++ b/comtypes/tools/codegenerator/typeannotator.py @@ -232,14 +232,14 @@ def getvalue(self, name: str) -> str: has_optional = False for _, argname, default in self.inarg_specs: if keyword.iskeyword(argname): - inargs = ["*args: Any", "**kwargs: Any"] + inargs = ["*args: hints.Any", "**kwargs: hints.Any"] break if default is None: if has_optional: # probably propput or propputref # HACK: Something that goes into this conditional branch # should be a special callback. - inargs.append("**kwargs: Any") + inargs.append("**kwargs: hints.Any") break inargs.append(f"{argname}: hints.Incomplete") else: @@ -251,7 +251,7 @@ def getvalue(self, name: str) -> str: elif len(outargs) == 1: out = outargs[0] else: - out = "Tuple[" + ", ".join(outargs) + "]" + out = "hints.Tuple[" + ", ".join(outargs) + "]" in_ = ("self, " + ", ".join(inargs)) if inargs else "self" return f"def {name}({in_}) -> {out}: ..." @@ -275,14 +275,14 @@ def getvalue(self, name: str) -> str: has_optional = False for _, argname, default in self.inarg_specs: if keyword.iskeyword(argname): - inargs = ["*args: Any", "**kwargs: Any"] + inargs = ["*args: hints.Any", "**kwargs: hints.Any"] break if default is None: if has_optional: # probably propput or propputref # HACK: Something that goes into this conditional branch # should be a special callback. - inargs.append("**kwargs: Any") + inargs.append("**kwargs: hints.Any") break inargs.append(f"{argname}: hints.Incomplete") else: