Add props parameter to wbgetentities

LeMyst · Feb 14, 2022 · ef59bfb · ef59bfb
1 parent 0d281b5
commit ef59bfb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/wikibaseintegrator/entities/baseentity.py b/wikibaseintegrator/entities/baseentity.py
@@ -79,8 +79,7 @@ def from_json(self, json_data: Dict[str, Any]) -> BaseEntity:
 
         return self
 
-    # noinspection PyMethodMayBeStatic
-    def _get(self, entity_id: str, login: _Login = None, allow_anonymous: bool = True, is_bot: bool = None, **kwargs: Any) -> Dict:  # pylint: disable=no-self-use
+    def _get(self, entity_id: str, login: _Login = None, allow_anonymous: bool = True, is_bot: bool = None, props: List = None, **kwargs: Any) -> Dict:
         """
         Retrieve an entity in json representation from the Wikibase instance
 
@@ -98,6 +97,9 @@ def _get(self, entity_id: str, login: _Login = None, allow_anonymous: bool = Tru
             'format': 'json'
         }
 
+        if props:
+            params.update({'props': '|'.join(props)})
+
         login = login or self.api.login
         is_bot = is_bot if is_bot is not None else self.api.is_bot
 

diff --git a/wikibaseintegrator/wbi_fastrun.py b/wikibaseintegrator/wbi_fastrun.py
@@ -618,7 +618,7 @@ def _process_lang(result: List) -> defaultdict[str, set]:
     def get_prop_datatype(self, prop_nr: str) -> Optional[str]:  # pylint: disable=no-self-use
         from wikibaseintegrator import WikibaseIntegrator
         wbi = WikibaseIntegrator()
-        property = wbi.property.get(prop_nr)
+        property = wbi.property.get(prop_nr, props=['datatype'])
         return property.datatype
 
     def clear(self) -> None: