Robocrys client update

materialsproject · May 11, 2021 · c8a2f93 · c8a2f93
1 parent ee0d297
commit c8a2f93
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/src/mp_api/routes/robocrys/client.py b/src/mp_api/routes/robocrys/client.py
@@ -5,4 +5,25 @@
 class RobocrysRester(BaseRester):
 
     suffix = "robocrys"
-    document_model = RobocrysDoc
+    document_model = RobocrysDoc  # type: ignore
+
+    def search_robocrys_text(self, keywords: list[str]):
+        """
+        Search text generated from Robocrystallographer.
+
+        Arguments:
+            keywords (List[str]): List of search keywords
+
+        Returns:
+            robocrys_docs (List[RobocrysDoc]): List of robocrystallographer documents
+        """
+
+        keyword_string = ",".join(keywords)
+
+        results = self._query_resource(
+            criteria={"keywords": keyword_string},
+            suburl="text_search",
+            use_document_model=True,
+        )
+
+        return results
diff --git a/src/mp_api/routes/robocrys/client.pyi b/src/mp_api/routes/robocrys/client.pyi
@@ -3,6 +3,7 @@ from mp_api.routes.robocrys.models import RobocrysDoc
 
 
 class RobocrysRester:
+
     def query_by_task_id(
         self, task_id, fields: Optional[List[str]] = None, monty_decode: bool = True
     ) -> RobocrysDoc: