X-Git-Url: https://code.communitydata.science/covid19.git/blobdiff_plain/308d462e767920ef541f8ccef2942e87eb854be8..36167295ecfe297780be2ec42ef1213e26d014e0:/transliterations/src/wikidata_api_calls.py

diff --git a/transliterations/src/wikidata_api_calls.py b/transliterations/src/wikidata_api_calls.py
new file mode 100644
index 0000000..f2fac88
--- /dev/null
+++ b/transliterations/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={"query":q,"format":"json"})
+    time.sleep(2)
+    return results
+