more mypy fixes

This commit is contained in:
svlandeg 2023-07-26 17:17:48 +02:00
parent fa6eceb5a7
commit b2a7c94787

View File

@ -44,22 +44,26 @@ def find_function(func_name: str, registry_name: str) -> Tuple[str, int]:
registry_desc = None registry_desc = None
try: try:
registry_desc = registry.find(registry_name, func_name) registry_desc = registry.find(registry_name, func_name)
except RegistryError as e: except RegistryError as e:
msg.fail( msg.fail(
f"Couldn't find registered function: '{func_name}' in registry '{registry_name}'", f"Couldn't find registered function: '{func_name}' in registry '{registry_name}'",
) )
msg.fail(f"{e}", exits=1) msg.fail(f"{e}", exits=1)
assert registry_desc is not None assert registry_desc is not None
registry_path = None
line_no = None
if registry_desc["file"]: if registry_desc["file"]:
registry_path = registry_desc["file"] registry_path = registry_desc["file"]
line_no = registry_desc["line_no"] line_no = registry_desc["line_no"]
else:
if not registry_path or not line_no:
msg.fail( msg.fail(
f"Couldn't find path to registered function: '{func_name}' in registry '{registry_name}'", f"Couldn't find path to registered function: '{func_name}' in registry '{registry_name}'",
exits=1, exits=1,
) )
assert registry_path is not None
assert line_no is not None
msg.good(f"Found registered function at {registry_path}:{line_no}") msg.good(f"Found registered function at {registry_path}:{line_no}")
return registry_path, line_no return str(registry_path), int(line_no)