Python code to find wikidata entities to translate. Here we search the api for entit...

[covid19.git] / translations / src / wikidata_api_calls.py

diff --git a/translations/src/wikidata_api_calls.py b/translations/src/wikidata_api_calls.py
new file mode 100644 (file)
index 0000000..022a9d4
--- /dev/null
+++ b/translations/src/wikidata_api_calls.py
@@ -0,0 +1,29 @@
+# File defines functions for making api calls to find translations and transliterations for key terms.
+
+import mwapi
+import sys
+sys.path.append("..")
+from defaults import user_agent
+
+def get_wikidata_api():
+    session = mwapi.Session(host="https://wikidata.org/w/api.php", user_agent=user_agent)
+    return session
+
+def search_wikidata(session, term, *args, **kwargs):
+    search_results = session.get(action='query',
+                                 list='search',
+                                 srsearch=term,
+#                                 srqiprofile='popular_inclinks_pv',
+                                 srlimit='max',
+                                 srnamespace=0,
+                                 *args,
+                                 **kwargs)
+
+
+    query = search_results.get('query', None)
+    results = query.get('search', None)
+
+    if results is None:
+        raise mwapi.session.APIError(f"No results for query: {term}")
+
+    return results