Add two methods to Item class

This commit is contained in:
Edward Betts 2021-06-15 00:08:24 +02:00
parent 503401ff57
commit d407224f35

View file

@ -58,6 +58,13 @@ class Item(Base):
if d_list: if d_list:
return d_list[0]['value'] return d_list[0]['value']
def get_aliases(self, lang='en'):
if lang not in self.aliases:
if 'en' not in self.aliases:
return []
lang = 'en'
return [a['value'] for a in self.aliases[lang]]
def get_part_of_names(self): def get_part_of_names(self):
if not self.claims: if not self.claims:
return set() return set()
@ -128,6 +135,9 @@ class Item(Base):
return dict(d) or None return dict(d) or None
def is_street(self):
return any(v and v["id"] == 'Q79007' for v in self.get_claim("P31"))
# class Claim(Base): # class Claim(Base):
# __tablename__ = "claim" # __tablename__ = "claim"