Adding restapi.

jarvis/__init__.py

@@ -11,4 +11,3 @@ def test(*args):
 
     path = os.path.join(os.path.split(__file__)[0], "tests")
     pytest.main(args=[path] + list(args))
-

jarvis/db/restapi.py

@@ -1,4 +1,5 @@
 """Module to access or upload data in JARVIS-API."""
+
 import requests
 import pprint
 import os
@@ -9,6 +10,25 @@
 import json
 
 
+def jarvisdft_optimade(
+    prefix="https://jarvis.nist.gov/optimade/jarvisdft/v1/structures/?filter=",
+    query="elements HAS  ALL C,Si",
+):
+    url = prefix + query
+    vals = []
+    while True:
+        if url is not None:
+            # print('url', url)
+            response = requests.get(url)
+            data = response.json()["data"]
+            for i in data:
+                vals.append(i)
+            url = response.json()["links"]["next"]
+        else:
+            break
+    return vals
+
+
 class Api(object):
     """Class for handling functions for JARVIS-API."""
 

jarvis/tests/testfiles/db/test_restapi.py

@@ -0,0 +1,8 @@
+from jarvis.db.restapi import jarvisdft_optimade
+
+
+def test_optimade():
+    x = jarvisdft_optimade()
+    print(x)
+    print(len(x))
+    assert len(x) > 1