Type hints and docstrings.

This commit is contained in:
Edward Betts 2023-05-14 20:29:24 +00:00
parent 0ed3d3c761
commit 384eb859b8

View file

@ -304,7 +304,8 @@ WHERE tags ? 'wikidata'
return tagged
def get_items(item_ids):
def get_items(item_ids: list[int]) -> list[model.Item]:
"""Get a Wikidata items with the given item IDs."""
items = []
for item_id in item_ids:
item = model.Item.query.get(item_id)
@ -317,8 +318,8 @@ def get_items(item_ids):
return items
def get_item_tags(item):
isa_list = [v["numeric-id"] for v in item.get_isa()]
def get_item_tags(item: model.Item) -> dict[str, list[str]]:
isa_list: list[int] = [v["numeric-id"] for v in item.get_isa()]
isa_items = [(isa, []) for isa in get_items(isa_list)]
osm_list = defaultdict(list)