Move another function into a dedicated module

This commit is contained in:
Edward Betts 2019-10-10 09:42:16 +01:00
parent d030f745de
commit bd6d13bc36
2 changed files with 31 additions and 26 deletions

28
depicts/wikidata_edit.py Normal file
View file

@ -0,0 +1,28 @@
from depicts import mediawiki
from depicts.model import DepictsItem
def create_depicts_item(item_id):
qid = f'Q{item_id}'
entity = mediawiki.get_entity(qid)
if 'en' in entity['labels']:
label = entity['labels']['en']['value']
else:
label = None
if 'en' in entity['descriptions']:
description = entity['descriptions']['en']['value']
else:
description = None
if 'en' in entity['aliases']:
alt_labels = {alt['value'] for alt in entity['aliases']['en']}
else:
alt_labels = set()
return DepictsItem(item_id=item_id,
label=label,
description=description,
alt_labels=alt_labels,
count=0)