X-Git-Url: https://code.communitydata.science/covid19.git/blobdiff_plain/09d171608f699eafe6528325938a892937a65302..98b07b8098611287eaa775b09622d1f3514303c8:/keywords/src/wikidata_api_calls.py

diff --git a/keywords/src/wikidata_api_calls.py b/keywords/src/wikidata_api_calls.py
new file mode 100644
index 0000000..212ed4f
--- /dev/null
+++ b/keywords/src/wikidata_api_calls.py
@@ -0,0 +1,35 @@
+# File defines functions for making api calls to find translations and transliterations for key terms.
+import mwapi
+import requests
+import sys
+import time
+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
+
+def run_sparql_query(q):
+    results = requests.get("https://query.wikidata.org/bigdata/namespace/wdq/sparql",params={"format":"json","query":q})
+    time.sleep(2)
+    return results
+