Adjust import style

This commit is contained in:
Edward Betts 2023-11-09 07:39:39 +01:00
parent e7004a08f4
commit d814ecf3a4

View file

@ -1,6 +1,6 @@
"""Class to represent artwork.""" """Class to represent artwork."""
from . import mediawiki from .mediawiki import get_entity, get_entity_with_cache
from .type import Claims, Entity, Sitelinks from .type import Claims, Entity, Sitelinks
@ -12,7 +12,7 @@ class Artwork:
def __init__(self, qid: str) -> None: def __init__(self, qid: str) -> None:
"""Init.""" """Init."""
entity = mediawiki.get_entity_with_cache(qid) entity = get_entity_with_cache(qid)
assert entity assert entity
self.entity = entity self.entity = entity
self.item_id = int(qid[1:]) self.item_id = int(qid[1:])
@ -48,7 +48,7 @@ class Artwork:
self.artist_entities = [] self.artist_entities = []
for artist in self.artists_claim: for artist in self.artists_claim:
artist_entity = mediawiki.get_entity(artist["id"]) artist_entity = get_entity(artist["id"])
assert artist_entity assert artist_entity
self.artist_entities.append(artist_entity) self.artist_entities.append(artist_entity)