import re
from typing import Any, Optional, Union
from wikibaseintegrator.datatypes.basedatatype import BaseDataType
[docs]
class Property(BaseDataType):
"""
Implements the Wikibase data type 'property'
"""
DTYPE = 'wikibase-property'
sparql_query = '''
SELECT * WHERE {{
?item_id <{wb_url}/prop/{pid}> ?s .
?s <{wb_url}/prop/statement/{pid}> <{wb_url}/entity/P{value}> .
}}
'''
[docs]
def __init__(self, value: Optional[Union[str, int]] = None, **kwargs: Any):
"""
Constructor, calls the superclass BaseDataType
:param value: The property number to serve as a value
:type value: str with a 'P' prefix, followed by several digits or only the digits without the 'P' prefix
"""
super().__init__(**kwargs)
self.set_value(value=value)
[docs]
def set_value(self, value: Optional[Union[str, int]] = None):
assert isinstance(value, (str, int)) or value is None, f"Expected str or int, found {type(value)} ({value})"
if value:
if isinstance(value, str):
pattern = re.compile(r'^(?:[a-zA-Z]+:|.+\/entity\/)?P?([0-9]+)$')
matches = pattern.match(value)
if not matches:
raise ValueError(f"Invalid property ID ({value}), format must be 'P[0-9]+'")
value = int(matches.group(1))
self.mainsnak.datavalue = {
'value': {
'entity-type': 'property',
'numeric-id': value,
'id': f'P{value}'
},
'type': 'wikibase-entityid'
}
[docs]
def get_sparql_value(self) -> str:
return self.mainsnak.datavalue['value']['id']