always quote types in generated hooks

This commit is contained in:
Damien Elmes 2021-02-01 08:36:33 +10:00
parent 1741ce1ed8
commit 25a1a2c89c

View File

@ -36,8 +36,7 @@ class Hook:
types = []
for arg in self.args or []:
(name, type) = arg.split(":")
if "." in type:
type = '"' + type.strip() + '"'
type = '"' + type.strip() + '"'
types.append(type)
types_str = ", ".join(types)
return f"Callable[[{types_str}], {self.return_type or 'None'}]"